101 lines
3.6 KiB
Python
101 lines
3.6 KiB
Python
|
|
import httpx
|
||
|
|
|
||
|
|
from app.providers.llm.base import LLMProvider
|
||
|
|
|
||
|
|
|
||
|
|
SYSTEM_PROMPT = """
|
||
|
|
You are a voice assistant for spoken conversations with older adults.
|
||
|
|
|
||
|
|
Speak naturally, clearly, and calmly.
|
||
|
|
Use short, simple sentences.
|
||
|
|
Prefer plain everyday language over technical wording.
|
||
|
|
Answer in the same language as the user, unless the user asks to switch languages.
|
||
|
|
|
||
|
|
Important response rules:
|
||
|
|
- Output plain text only.
|
||
|
|
- No markdown.
|
||
|
|
- No bullet points.
|
||
|
|
- No numbered lists.
|
||
|
|
- No tables.
|
||
|
|
- No code.
|
||
|
|
- No emojis.
|
||
|
|
- No URLs unless the user explicitly asks for one.
|
||
|
|
- Do not use asterisks, hashtags, or formatting symbols.
|
||
|
|
- Do not write headings.
|
||
|
|
- Do not use long disclaimers.
|
||
|
|
|
||
|
|
Voice style rules:
|
||
|
|
- Sound helpful, warm, and patient.
|
||
|
|
- Keep answers brief by default: 1 to 3 short sentences.
|
||
|
|
- If more detail is needed, explain step by step in natural spoken sentences.
|
||
|
|
- Ask at most one follow-up question at a time.
|
||
|
|
- If the answer contains several items, present them as natural speech, not as a list.
|
||
|
|
- Use wording that sounds good when spoken aloud.
|
||
|
|
- Avoid abbreviations when possible.
|
||
|
|
- Avoid symbols when words are better.
|
||
|
|
- Prefer complete spoken forms for dates, times, and numbers when useful.
|
||
|
|
|
||
|
|
Safety and honesty rules:
|
||
|
|
- If you are unsure, say so briefly and clearly.
|
||
|
|
- Do not invent facts.
|
||
|
|
- If current real-world information is needed and unavailable, say that clearly.
|
||
|
|
|
||
|
|
Always optimize your answer for listening, not for reading.
|
||
|
|
""".strip()
|
||
|
|
|
||
|
|
|
||
|
|
class OpenRouterLLMProvider(LLMProvider):
|
||
|
|
def __init__(self, api_key: str, model: str):
|
||
|
|
self.api_key = (api_key or "").strip()
|
||
|
|
self.model = (model or "").strip()
|
||
|
|
|
||
|
|
async def complete(self, text: str, session_id: str | None = None) -> str:
|
||
|
|
if not self.api_key:
|
||
|
|
raise ValueError("OPENROUTER_API_KEY is empty")
|
||
|
|
if not self.model:
|
||
|
|
raise ValueError("OPENROUTER_LLM_MODEL is empty")
|
||
|
|
if not text or not text.strip():
|
||
|
|
raise ValueError("LLM input text is empty")
|
||
|
|
|
||
|
|
payload = {
|
||
|
|
"model": self.model,
|
||
|
|
"messages": [
|
||
|
|
{"role": "system", "content": SYSTEM_PROMPT},
|
||
|
|
{"role": "user", "content": text.strip()},
|
||
|
|
],
|
||
|
|
}
|
||
|
|
|
||
|
|
timeout = httpx.Timeout(connect=10.0, read=120.0, write=30.0, pool=10.0)
|
||
|
|
|
||
|
|
async with httpx.AsyncClient(timeout=timeout) as client:
|
||
|
|
try:
|
||
|
|
response = await client.post(
|
||
|
|
"https://openrouter.ai/api/v1/chat/completions",
|
||
|
|
headers={
|
||
|
|
"Authorization": f"Bearer {self.api_key}",
|
||
|
|
"Content-Type": "application/json",
|
||
|
|
},
|
||
|
|
json=payload,
|
||
|
|
)
|
||
|
|
response.raise_for_status()
|
||
|
|
except httpx.HTTPStatusError as exc:
|
||
|
|
raise RuntimeError(
|
||
|
|
f"OpenRouter LLM error {exc.response.status_code}: {exc.response.text}"
|
||
|
|
) from exc
|
||
|
|
except httpx.TimeoutException as exc:
|
||
|
|
raise RuntimeError("OpenRouter LLM timeout") from exc
|
||
|
|
except httpx.HTTPError as exc:
|
||
|
|
raise RuntimeError(f"OpenRouter LLM transport error: {exc}") from exc
|
||
|
|
|
||
|
|
data = response.json()
|
||
|
|
|
||
|
|
try:
|
||
|
|
content = data["choices"][0]["message"]["content"]
|
||
|
|
except (KeyError, IndexError, TypeError) as exc:
|
||
|
|
raise RuntimeError(f"Unexpected OpenRouter LLM response: {data}") from exc
|
||
|
|
|
||
|
|
if not content or not str(content).strip():
|
||
|
|
raise RuntimeError("OpenRouter LLM returned empty content")
|
||
|
|
|
||
|
|
return str(content).strip()
|
||
|
|
|