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