Navya-Sree's picture
Create https://ollama.ai/
46b621e verified
raw
history blame contribute delete
418 Bytes
# 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"]