diff --git a/libs/partners/openai/langchain_openai/chat_models/base.py b/libs/partners/openai/langchain_openai/chat_models/base.py index d0bed6f921..b5b7199edb 100644 --- a/libs/partners/openai/langchain_openai/chat_models/base.py +++ b/libs/partners/openai/langchain_openai/chat_models/base.py @@ -424,9 +424,9 @@ class ChatOpenAI(BaseChatModel): chunk = ChatGenerationChunk( message=chunk, generation_info=generation_info or None ) - yield chunk if run_manager: run_manager.on_llm_new_token(chunk.text, chunk=chunk, logprobs=logprobs) + yield chunk def _generate( self, @@ -516,11 +516,11 @@ class ChatOpenAI(BaseChatModel): chunk = ChatGenerationChunk( message=chunk, generation_info=generation_info or None ) - yield chunk if run_manager: await run_manager.on_llm_new_token( token=chunk.text, chunk=chunk, logprobs=logprobs ) + yield chunk async def _agenerate( self,