Update app.py

app.py

@@ -2,11 +2,16 @@ import os
 import json
 import torch
 import logging
+import numpy
 import gradio as gr
+import torch.serialization
 from importlib.resources import files
 from txagent import TxAgent
 from tooluniverse import ToolUniverse
 
+# Patch PyTorch to allow loading old numpy pickles
+torch.serialization.add_safe_globals([numpy.core.multiarray._reconstruct])
+
 logging.basicConfig(
     level=logging.INFO,
     format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
@@ -48,6 +53,9 @@ def patch_embedding_loading():
             if not os.path.exists(CONFIG["embedding_filename"]):
                 return False
             self.tool_desc_embedding = safe_load_embeddings(CONFIG["embedding_filename"])
+            if self.tool_desc_embedding is None:
+                logger.error("Tool embedding file could not be loaded.")
+                return False
 
             tools = tooluniverse.get_all_tools() if hasattr(tooluniverse, "get_all_tools") else getattr(tooluniverse, "tools", [])
             if len(tools) != len(self.tool_desc_embedding):
@@ -75,6 +83,11 @@ def create_agent():
     patch_embedding_loading()
     prepare_tool_files()
     try:
+        tu = ToolUniverse()
+        tools = tu.get_all_tools() if hasattr(tu, "get_all_tools") else getattr(tu, "tools", [])
+        available_tool_names = [t["name"] for t in tools]
+        additional_default_tools = [t for t in ["DirectResponse", "RequireClarification"] if t in available_tool_names]
+
         agent = TxAgent(
             CONFIG["model_name"],
             CONFIG["rag_model_name"],
@@ -83,7 +96,7 @@ def create_agent():
             enable_checker=True,
             step_rag_num=10,
             seed=42,
-            additional_default_tools=["DirectResponse", "RequireClarification"]
+            additional_default_tools=additional_default_tools
         )
         agent.init_model()
         return agent
@@ -91,7 +104,6 @@ def create_agent():
         logger.error(f"Agent initialization failed: {e}")
         raise
 
-# ✅ FIXED: Proper message formatting
 def respond(msg, chat_history, temperature, max_new_tokens, max_tokens, multi_agent, conversation, max_round):
     if not isinstance(msg, str) or len(msg.strip()) <= 10:
         return chat_history + [{"role": "assistant", "content": "Hi, I am TxAgent. Please provide a valid message longer than 10 characters."}]
@@ -147,4 +159,4 @@ def main():
     demo.launch(share=False)
 
 if __name__ == "__main__":
-    main()
+    main()