24 lines
564 B
Python
24 lines
564 B
Python
import os
|
|
import requests
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
|
|
API_KEY = os.getenv("OPENROUTER_API_KEY")
|
|
MODEL = os.getenv("OPENROUTER_MODEL")
|
|
|
|
def ask_llm(messages):
|
|
r = requests.post(
|
|
"https://openrouter.ai/api/v1/chat/completions",
|
|
headers={
|
|
"Authorization": f"Bearer {API_KEY}",
|
|
"Content-Type": "application/json",
|
|
},
|
|
json={
|
|
"model": MODEL,
|
|
"messages": messages,
|
|
"temperature": 0.3
|
|
}
|
|
)
|
|
|
|
return r.json()["choices"][0]["message"]["content"] |