vadimwit
51 lines
1.7 KiB
Python
51 lines
1.7 KiB
Python
from api.config import settings
|
|
|
|
|
|
SYSTEM_PROMPT = """You are a friendly, beginner-aware music theory assistant for WhatTheFlat — a live jam helper app.
|
|
You help musicians understand what's happening harmonically during a jam session.
|
|
Keep answers short, warm, and jargon-free. If you use a music term, briefly explain it.
|
|
Never say someone played a "wrong" note — always frame it as "try these instead".
|
|
When given context about the current key and chord, use it in your response."""
|
|
|
|
|
|
async def chat(messages: list[dict], context: dict | None = None) -> str:
|
|
system = SYSTEM_PROMPT
|
|
if context:
|
|
parts = []
|
|
if context.get("key"):
|
|
parts.append(f"Current key: {context['key']}")
|
|
if context.get("chord"):
|
|
parts.append(f"Current chord: {context['chord']}")
|
|
if parts:
|
|
system += "\n\nLive session context:\n" + "\n".join(parts)
|
|
|
|
if settings.ai_mode == "online":
|
|
return await _chat_claude(messages, system)
|
|
else:
|
|
return await _chat_ollama(messages, system)
|
|
|
|
|
|
async def _chat_claude(messages: list[dict], system: str) -> str:
|
|
import anthropic
|
|
|
|
client = anthropic.AsyncAnthropic(api_key=settings.anthropic_api_key)
|
|
response = await client.messages.create(
|
|
model=settings.claude_model,
|
|
max_tokens=512,
|
|
system=system,
|
|
messages=messages,
|
|
)
|
|
return response.content[0].text
|
|
|
|
|
|
async def _chat_ollama(messages: list[dict], system: str) -> str:
|
|
import ollama
|
|
|
|
client = ollama.AsyncClient(host=settings.ollama_host)
|
|
full_messages = [{"role": "system", "content": system}] + messages
|
|
response = await client.chat(
|
|
model=settings.ollama_model,
|
|
messages=full_messages,
|
|
)
|
|
return response.message.content
|