fix

2026-02-03 15:55:44 +00:00 · 2024-02-23 12:09:38 -05:00
1 changed files with 2 additions and 2 deletions
--- a/libs/community/langchain_community/chat_models/openai.py
+++ b/libs/community/langchain_community/chat_models/openai.py
@@ -415,7 +415,7 @@ class ChatOpenAI(BaseChatModel):
                message=chunk, generation_info=generation_info
            )
            if run_manager:
-                run_manager.on_llm_new_token(chunk.text, chunk=cg_chunk)
+                run_manager.on_llm_new_token(cg_chunk.text, chunk=cg_chunk)
            yield cg_chunk

    def _generate(
@@ -507,7 +507,7 @@ class ChatOpenAI(BaseChatModel):
                message=chunk, generation_info=generation_info
            )
            if run_manager:
-                await run_manager.on_llm_new_token(token=chunk.text, chunk=cg_chunk)
+                await run_manager.on_llm_new_token(token=cg_chunk.text, chunk=cg_chunk)
            yield cg_chunk

    async def _agenerate(