From ff6740eb79a4f9668bd63b8a6376edced94f8985 Mon Sep 17 00:00:00 2001 From: Bryan Qiu Date: Wed, 18 Dec 2024 12:05:41 -0800 Subject: [PATCH] wip Signed-off-by: Bryan Qiu --- .../langchain/src/databricks_langchain/chat_agent_langchain.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/integrations/langchain/src/databricks_langchain/chat_agent_langchain.py b/integrations/langchain/src/databricks_langchain/chat_agent_langchain.py index 0bd64de..250ad8d 100644 --- a/integrations/langchain/src/databricks_langchain/chat_agent_langchain.py +++ b/integrations/langchain/src/databricks_langchain/chat_agent_langchain.py @@ -15,4 +15,4 @@ def predict_stream( self, messages: list[ChatAgentMessage], params: Optional[ChatAgentParams] = None ): for event in self.agent.stream({"messages": self._convert_messages_to_dict(messages)}): - yield ChatAgentResponse(**event).to_dict() + yield ChatAgentResponse(**event)