| # 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() |