Merge branch 'master' into bagatur/3_12_ci

langchain-ai · Jan 27, 2024 · 554b424 · 554b424
2 parents b522d11 + 27665e3
commit 554b424
Show file tree

Hide file tree

Showing 10 changed files with 6 additions and 122 deletions.
diff --git a/.github/workflows/langchain_cli_release.yml b/.github/workflows/langchain_cli_release.yml
diff --git a/.github/workflows/langchain_community_release.yml b/.github/workflows/langchain_community_release.yml
diff --git a/.github/workflows/langchain_core_release.yml b/.github/workflows/langchain_core_release.yml
diff --git a/.github/workflows/langchain_experimental_release.yml b/.github/workflows/langchain_experimental_release.yml
diff --git a/.github/workflows/langchain_experimental_test_release.yml b/.github/workflows/langchain_experimental_test_release.yml
diff --git a/.github/workflows/langchain_openai_release.yml b/.github/workflows/langchain_openai_release.yml
diff --git a/.github/workflows/langchain_release.yml b/.github/workflows/langchain_release.yml
diff --git a/.github/workflows/langchain_test_release.yml b/.github/workflows/langchain_test_release.yml
diff --git a/.github/workflows/_release.yml → .github/workflows/release.yml b/.github/workflows/_release.yml → .github/workflows/release.yml
diff --git a/libs/community/langchain_community/chat_models/anthropic.py b/libs/community/langchain_community/chat_models/anthropic.py
@@ -142,9 +142,10 @@ def _stream(
         stream_resp = self.client.completions.create(**params, stream=True)
         for data in stream_resp:
             delta = data.completion
-            yield ChatGenerationChunk(message=AIMessageChunk(content=delta))
+            chunk = ChatGenerationChunk(message=AIMessageChunk(content=delta))
+            yield chunk
             if run_manager:
-                run_manager.on_llm_new_token(delta)
+                run_manager.on_llm_new_token(delta, chunk=chunk)
 
     async def _astream(
         self,
@@ -161,9 +162,10 @@ async def _astream(
         stream_resp = await self.async_client.completions.create(**params, stream=True)
         async for data in stream_resp:
             delta = data.completion
-            yield ChatGenerationChunk(message=AIMessageChunk(content=delta))
+            chunk = ChatGenerationChunk(message=AIMessageChunk(content=delta))
+            yield chunk
             if run_manager:
-                await run_manager.on_llm_new_token(delta)
+                await run_manager.on_llm_new_token(delta, chunk=chunk)
 
     def _generate(
         self,