File size: 418 Bytes
46b621e
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
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"]