wip

Signed-off-by: Bryan Qiu <[email protected]>
databricks · Dec 18, 2024 · ff6740e · ff6740e
1 parent 57bf496
commit ff6740e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)