Spaces:
Sleeping
Sleeping
Update ai_service.py
Browse files- ai_service.py +1 -9
ai_service.py
CHANGED
|
@@ -241,12 +241,4 @@ async def get_ai_response(user_query: str, telegram_id: int):
|
|
| 241 |
completion = await loop.run_in_executor(None, lambda: call_hf(messages))
|
| 242 |
response_message = completion.choices[0].message
|
| 243 |
|
| 244 |
-
|
| 245 |
-
|
| 246 |
-
if db_manager:
|
| 247 |
-
await asyncio.gather(
|
| 248 |
-
db_manager.save_message(telegram_id, user_query, "user"),
|
| 249 |
-
db_manager.save_message(telegram_id, final_response, "assistant")
|
| 250 |
-
)
|
| 251 |
-
|
| 252 |
-
return final_response
|
|
|
|
| 241 |
completion = await loop.run_in_executor(None, lambda: call_hf(messages))
|
| 242 |
response_message = completion.choices[0].message
|
| 243 |
|
| 244 |
+
return clean_ai_response(response_message.content if response_message.content else "")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|