zenaight commited on
Commit ·
5f13f2a
1
Parent(s): fdeebb1
cheapest ai
Browse files
main.py
CHANGED
|
@@ -21,7 +21,7 @@ if OPENAI_API_KEY:
|
|
| 21 |
# Pydantic model for chat requests
|
| 22 |
class ChatRequest(BaseModel):
|
| 23 |
message: str
|
| 24 |
-
model: Optional[str] = "gpt-
|
| 25 |
max_tokens: Optional[int] = 150
|
| 26 |
|
| 27 |
# Pydantic model for chat responses
|
|
@@ -79,7 +79,7 @@ async def generate_ai_response(user_message: str) -> str:
|
|
| 79 |
|
| 80 |
try:
|
| 81 |
response = openai_client.chat.completions.create(
|
| 82 |
-
model="gpt-
|
| 83 |
messages=[
|
| 84 |
{"role": "system", "content": "You are a helpful assistant. Keep your responses concise and friendly."},
|
| 85 |
{"role": "user", "content": user_message}
|
|
|
|
| 21 |
# Pydantic model for chat requests
|
| 22 |
class ChatRequest(BaseModel):
|
| 23 |
message: str
|
| 24 |
+
model: Optional[str] = "gpt-4o-mini" # Using the cheapest OpenAI model
|
| 25 |
max_tokens: Optional[int] = 150
|
| 26 |
|
| 27 |
# Pydantic model for chat responses
|
|
|
|
| 79 |
|
| 80 |
try:
|
| 81 |
response = openai_client.chat.completions.create(
|
| 82 |
+
model="gpt-4o-mini",
|
| 83 |
messages=[
|
| 84 |
{"role": "system", "content": "You are a helpful assistant. Keep your responses concise and friendly."},
|
| 85 |
{"role": "user", "content": user_message}
|