File size: 434 Bytes
0575976 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | import requests
OLLAMA_URL = ""
def query_llm(prompt: str, model: str = "") -> str:
payload = {
"model": model,
"prompt": prompt,
"stream": False
}
try:
response = requests.post(OLLAMA_URL, json=payload, timeout=60)
response.raise_for_status()
data = response.json()
return data.get("response", "").strip()
except Exception as e:
return f"[Error] {e}"
|