Update app.py

app.py

@@ -43,7 +43,7 @@ CONFIG = {
 def generate_tool_embeddings(agent):
     tu = ToolUniverse(tool_files=CONFIG["tool_files"])
     tu.load_tools()
-    embedding_tensor = agent.rag_model.generate_tool_desc_embedding(tu)
+    embedding_tensor = agent.rag_model.load_tool_desc_embedding(tu)
     if embedding_tensor is not None:
         torch.save(embedding_tensor, CONFIG["embedding_filename"])
         logger.info(f"Saved new embedding tensor to {CONFIG['embedding_filename']}")
@@ -88,7 +88,7 @@ def respond(msg, chat_history, temperature, max_new_tokens, max_tokens, multi_ag
 
     message = msg.strip()
     chat_history.append({"role": "user", "content": message})
-    formatted_history = [(m["role"], m["content"]) for m in chat_history if "role" in m and "content" in m]
+    formatted_history = chat_history  # format as list of dicts for run_gradio_chat
 
     try:
         response_generator = agent.run_gradio_chat(
@@ -101,18 +101,20 @@ def respond(msg, chat_history, temperature, max_new_tokens, max_tokens, multi_ag
             conversation=conversation,
             max_round=max_round,
             seed=42,
-            call_agent_level=None,
+            call_agent_level=0,
             sub_agent_task=None
         )
 
         collected = ""
         for chunk in response_generator:
-            if isinstance(chunk, dict) and "content" in chunk:
+            if isinstance(chunk, list):
+                for msg in chunk:
+                    if isinstance(msg, dict) and "content" in msg:
+                        collected += msg["content"]
+            elif isinstance(chunk, dict) and "content" in chunk:
                 collected += chunk["content"]
             elif isinstance(chunk, str):
                 collected += chunk
-            elif chunk is not None:
-                collected += str(chunk)
 
         chat_history.append({"role": "assistant", "content": collected or "⚠️ No content returned."})
 
@@ -143,7 +145,7 @@ def main():
     global agent
     agent = create_agent()
     demo = create_demo(agent)
-    demo.launch(share=False)
+    demo.launch(share=True)
 
 if __name__ == "__main__":
     main()