anthropic[patch]: Handle non text llm tokens (langchain-ai#6179)

* anthropic[patch]: Handle non text llm tokens * check for input field
tylerdavis · Jul 23, 2024 · e576ade · e576ade
1 parent 87d92d9
commit e576ade
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/libs/langchain-anthropic/src/chat_models.ts b/libs/langchain-anthropic/src/chat_models.ts
@@ -598,9 +598,14 @@ function extractToolCallChunk(
 }
 
 function extractToken(chunk: AIMessageChunk): string | undefined {
-  return typeof chunk.content === "string" && chunk.content !== ""
-    ? chunk.content
-    : undefined;
+  if (typeof chunk.content === "string") {
+    return chunk.content;
+  } else if (Array.isArray(chunk.content) && "input" in chunk.content[0]) {
+    return typeof chunk.content[0].input === "string"
+      ? chunk.content[0].input
+      : JSON.stringify(chunk.content[0].input);
+  }
+  return undefined;
 }
 
 function extractToolUseContent(