Make new agent work with Argo

Author: shuds13

ensemble_agent/agent.py

@@ -125,7 +125,10 @@ async def _run_autonomous(agent, messages, initial_msg, config):
     print(f"\n{'=' * 60}")
     print("Agent completed")
     print(f"{'=' * 60}")
-    print(result["messages"][-1].content)
+    content = result["messages"][-1].content
+    if isinstance(content, list):
+        content = "".join(block.get("text", "") for block in content)
+    print(content)
 
 
 async def _run_interactive(agent, messages, initial_msg, config, has_generator):
@@ -142,6 +145,8 @@ async def _run_interactive(agent, messages, initial_msg, config, has_generator):
             result = await agent.ainvoke({"messages": messages})
             messages = result["messages"]
             response = messages[-1].content
+            if isinstance(response, list):
+                response = "".join(block.get("text", "") for block in response)
             if response:
                 print(f"\n{response}", flush=True)
         except Exception as e:

ensemble_agent/llm.py

@@ -13,7 +13,8 @@ def create_llm(model, temperature=0, base_url=None):
             sys.exit("Error: pip install langchain-anthropic required for Claude models")
         anthropic_base = os.environ.get("ANTHROPIC_BASE_URL")
         return ChatAnthropic(
-            model=model, temperature=temperature, base_url=anthropic_base or None
+            model=model, temperature=temperature, base_url=anthropic_base or None,
+            streaming=True,
         )
     from langchain_openai import ChatOpenAI