Create app/agents/chat_agent.py

app/agents/chat_agent.py

@@ -0,0 +1,20 @@
+# app/agents/chat_agent.py
+
+import requests
+import os
+
+OLLAMA_URL = os.getenv("OLLAMA_URL")
+
+def run(task):
+    prompt = task["prompt"]
+
+    r = requests.post(
+        f"{OLLAMA_URL}/api/generate",
+        json={
+            "model": os.getenv("OLLAMA_MODEL"),
+            "prompt": prompt,
+            "stream": False
+        }
+    )
+
+    return r.json()