| from openai import AsyncOpenAI |
| from config import OPENAI_API_KEY |
|
|
| client = AsyncOpenAI(api_key=OPENAI_API_KEY) |
|
|
|
|
| SYSTEM_PROMPT = ( |
| "You are a helpful assistant. Respond concisely in the same language the user writes in. " |
| "Keep responses under 1000 characters when possible." |
| ) |
|
|
|
|
| async def ask_gpt(messages: list, model: str = "gpt-3.5-turbo") -> str: |
| full_messages = [{"role": "system", "content": SYSTEM_PROMPT}] |
| full_messages.extend(messages) |
| r = await client.chat.completions.create(model=model, messages=full_messages) |
| return r.choices[0].message.content |
|
|
