From 91150d41504fe728170c352b0209b35e9bbb065b Mon Sep 17 00:00:00 2001
From: Elliot Kang <kkang2097@gmail.com>
Date: Sat, 30 Sep 2023 13:49:54 -0700
Subject: [PATCH] Updated Anthropic Stream Token
---
packages/core/src/callbacks/CallbackManager.ts | 9 ++++++++-
1 file changed, 8 insertions(+), 1 deletion(-)
diff --git a/packages/core/src/callbacks/CallbackManager.ts b/packages/core/src/callbacks/CallbackManager.ts
index ff7f924ba..fb7eff8c9 100644
--- a/packages/core/src/callbacks/CallbackManager.ts
+++ b/packages/core/src/callbacks/CallbackManager.ts
@@ -39,7 +39,14 @@ export interface DefaultStreamToken {
//OpenAI stream token schema is the default.
//Note: Anthropic and Replicate also use similar token schemas.
export type OpenAIStreamToken = DefaultStreamToken;
-export type AnthropicStreamToken = DefaultStreamToken;
+export type AnthropicStreamToken =
+{
+ completion: string,
+ stop_reason: string | undefined,
+ model: string,
+ stop: boolean | undefined,
+ log_id: string
+};
//
//Callback Responses
--
GitLab