Add custom bind_tools

agents.py

@@ -138,9 +138,6 @@ def arvix_search(query: str) -> str:
 with open("system_prompt.txt", "r", encoding="utf-8") as f:
     system_prompt = f.read()
 
-# System message
-sys_msg = SystemMessage(content=system_prompt)
-
 
 tools = [
     multiply,
@@ -176,10 +173,47 @@ def build_graph(provider: str = "rwkv"):
         title = "rwkv7-g1-0.1b-20250307-ctx4096"
         pth = hf_hub_download(repo_id="BlinkDL/rwkv7-g1", filename=f"{title}.pth")
         model_path = pth.replace(".pth", "")
-        llm = RWKV(
+        raw_llm = RWKV(
             model=model_path,
             strategy="cpu fp32", 
         )
+
+        class RWKVWithTools:
+            def __init__(self, llm, system_prompt: str):
+                self.llm = llm
+                self.system_prompt = system_prompt
+                self.tools = []
+
+            def bind_tools(self, tools):
+                self.tools = tools
+                return self
+
+            def invoke(self, messages):
+                # Build a tools spec block
+                specs = []
+                for t in self.tools:
+                    specs.append(f"- {t.name}({getattr(t, 'args_schema', {})}): {t.description}")
+
+                header = (
+                    f"{self.system_prompt}\n\n"
+                    "TOOLS AVAILABLE:\n"
+                    + "\n".join(specs)
+                    + "\n\n"
+                    "To call a tool, respond exactly with:\n"
+                    "`<tool_name>(arg1=…,arg2=…)` and nothing else.\n\n"
+                )
+
+                # Reconstruct conversation
+                convo = "\n".join(
+                    f"{'User:' if isinstance(m, HumanMessage) else 'Assistant:'} {m.content}"
+                    for m in messages
+                )
+
+                prompt = header + convo
+                # delegate to LangChain’s invoke()
+                return self.llm.invoke(prompt)
+
+        llm = RWKVWithTools(raw_llm, system_prompt=system_prompt)
         # --- END RWKV SETUP ---
     else:
         raise ValueError("Invalid provider. Choose 'google', 'groq' or 'huggingface'.")