Update app.py

app.py

@@ -7,6 +7,7 @@ from tooluniverse import ToolUniverse
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import warnings
 from typing import List, Dict, Any
+from importlib.resources import files
 
 # Suppress specific warnings
 warnings.filterwarnings("ignore", category=UserWarning)
@@ -17,10 +18,10 @@ CONFIG = {
     "rag_model_name": "mims-harvard/ToolRAG-T1-GTE-Qwen2-1.5B",
     "embedding_filename": "ToolRAG-T1-GTE-Qwen2-1.5Btool_embedding_47dc56b3e3ddeb31af4f19defdd538d984de1500368852a0fab80bc2e826c944.pt",
     "tool_files": {
-        "opentarget": "opentarget_tools.json",
-        "fda_drug_label": "fda_drug_labeling_tools.json",
-        "special_tools": "special_tools.json",
-        "monarch": "monarch_tools.json",
+        "opentarget": str(files('tooluniverse.data').joinpath('opentarget_tools.json')),
+        "fda_drug_label": str(files('tooluniverse.data').joinpath('fda_drug_labeling_tools.json')),
+        "special_tools": str(files('tooluniverse.data').joinpath('special_tools.json')),
+        "monarch": str(files('tooluniverse.data').joinpath('monarch_tools.json')),
         "new_tool": "./data/new_tool.json"
     }
 }
@@ -75,7 +76,12 @@ class TxAgentWrapper:
         try:
             logger.info("Loading models from Hugging Face Hub...")
             
-            # Initialize ToolUniverse first
+            # Verify tool files exist
+            for tool_name, tool_path in CONFIG["tool_files"].items():
+                if tool_name != "new_tool" and not os.path.exists(tool_path):
+                    raise FileNotFoundError(f"Tool file not found: {tool_path}")
+            
+            # Initialize ToolUniverse with verified paths
             self.tooluniverse = ToolUniverse(tool_files=CONFIG["tool_files"])
             if hasattr(self.tooluniverse, 'load_tools'):
                 self.tooluniverse.load_tools()
@@ -119,7 +125,7 @@ class TxAgentWrapper:
             
             # Format conversation
             conversation = [
-                {"role": "system", "content": "You are a helpful assistant that will solve problems through detailed, step-by-step reasoning."},
+                {"role": "system", "content": "You are a helpful assistant that will solve problems through detailed, step-by-step reasoning." + tools_prompt},
                 *self._format_history(history),
                 {"role": "user", "content": message}
             ]