Update app/agent/agent.py

app/agent/agent.py

@@ -1,13 +1,17 @@
 import requests
 import json
 from config import Config
-import tools
+from tools_runtime import execute_tool_call
+
 
 class Agent:
 
     def __init__(self):
         self.history = []
 
+    # ========================
+    # LLM CALL LAYER
+    # ========================
     def call_llm(self, messages):
         res = requests.post(
             Config.LLM_API_URL,
@@ -21,20 +25,88 @@ class Agent:
                 "temperature": 0.2
             }
         )
+
         return res.json()
 
+    # ========================
+    # MAIN EXECUTION LOOP
+    # ========================
     def run(self, user_input):
 
         messages = [
-            {"role": "system", "content": Config.SYSTEM_PROMPT},
+            {
+                "role": "system",
+                "content": Config.SYSTEM_PROMPT + """
+
+You are an autonomous coding agent.
+
+You can respond in TWO ways:
+
+1) NORMAL RESPONSE:
+Explain or answer normally.
+
+2) TOOL CALL (STRICT JSON ONLY):
+{
+  "tool": "write_file",
+  "args": {
+    "path": "test.txt",
+    "content": "hello"
+  }
+}
+
+Available tools:
+- run_shell(cmd)
+- write_file(path, content)
+- read_file(path)
+- list_files(path)
+- delete_file(path)
+
+If tool is required → respond ONLY JSON.
+If not → normal text.
+"""
+            },
             {"role": "user", "content": user_input}
         ]
 
         response = self.call_llm(messages)
 
-        return response
+        # ========================
+        # Extract LLM output
+        # ========================
+        content = response["choices"][0]["message"]["content"]
+
+        # ========================
+        # TOOL PARSING LAYER
+        # ========================
+        try:
+            tool_call = json.loads(content)
+
+            if isinstance(tool_call, dict) and "tool" in tool_call:
+                result = execute_tool_call(tool_call)
+
+                # store history
+                self.history.append({
+                    "input": user_input,
+                    "tool_call": tool_call,
+                    "result": result
+                })
+
+                return result
+
+        except json.JSONDecodeError:
+            pass
+
+        # ========================
+        # NORMAL RESPONSE
+        # ========================
+        return {
+            "response": content
+        }
 
 
+# ========================
+# DEBUG RUN
+# ========================
 if __name__ == "__main__":
     agent = Agent()
-    print(agent.run("Explain this project structure"))
+    print(agent.run("List files in current directory"))