Create Local Models with Ollama

Local Models with Ollama

@@ -0,0 +1,14 @@
+# Install Ollama first: https://ollama.ai/
+import requests
+
+class LocalCoderAgent:
+    def generate_code(self, prompt):
+        response = requests.post(
+            "http://localhost:11434/api/generate",
+            json={
+                "model": "codellama",
+                "prompt": f"Write Python code: {prompt}",
+                "stream": False
+            }
+        )
+        return response.json()["response"]