25 lines
636 B
Python
25 lines
636 B
Python
# ai.py
|
|
|
|
import requests
|
|
import os
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
AI_URL = os.getenv("OLLAMA_API_URL")
|
|
|
|
def get_ai_response(prompt):
|
|
payload = {
|
|
"model": "mistral:7b", # Adjust to match your model
|
|
"prompt": prompt,
|
|
"stream": False
|
|
}
|
|
|
|
try:
|
|
response = requests.post(AI_URL, json=payload)
|
|
if response.status_code == 200:
|
|
result = response.json()
|
|
return result.get("response", "[No response]")
|
|
else:
|
|
return f"[Error {response.status_code}] {response.text}"
|
|
except Exception as e:
|
|
return f"[Request failed] {str(e)}"
|