Create backend/agent_instance.py

backend/agent_instance.py

@@ -0,0 +1,27 @@
+# backend/agent_instance.py
+import os
+from txagent.txagent import TxAgent
+
+def init_agent():
+    model_cache_dir = os.path.expanduser("~/.cache/txagent_models")
+    os.environ["TRANSFORMERS_CACHE"] = model_cache_dir
+    os.environ["HF_HOME"] = model_cache_dir
+
+    model_name = "mims-harvard/TxAgent-T1-Llama-3.1-8B"
+    rag_model_name = "mims-harvard/ToolRAG-T1-GTE-Qwen2-1.5B"
+    tool_files_dict = {
+        "new_tool": os.path.abspath("data/new_tool.json")
+    }
+
+    agent = TxAgent(
+        model_name=model_name,
+        rag_model_name=rag_model_name,
+        tool_files_dict=tool_files_dict,
+        force_finish=True,
+        enable_checker=True,
+        step_rag_num=10,
+        seed=100,
+        additional_default_tools=[]
+    )
+    agent.init_model()
+    return agent