Create app/agent/agent.py

app/agent/agent.py

@@ -0,0 +1,40 @@
+import requests
+import json
+from config import Config
+import tools
+
+class Agent:
+
+    def __init__(self):
+        self.history = []
+
+    def call_llm(self, messages):
+        res = requests.post(
+            Config.LLM_API_URL,
+            headers={
+                "Authorization": f"Bearer {Config.LLM_API_KEY}",
+                "Content-Type": "application/json"
+            },
+            json={
+                "model": Config.MODEL,
+                "messages": messages,
+                "temperature": 0.2
+            }
+        )
+        return res.json()
+
+    def run(self, user_input):
+
+        messages = [
+            {"role": "system", "content": Config.SYSTEM_PROMPT},
+            {"role": "user", "content": user_input}
+        ]
+
+        response = self.call_llm(messages)
+
+        return response
+
+
+if __name__ == "__main__":
+    agent = Agent()
+    print(agent.run("Explain this project structure"))