|
|
@@ -2,6 +2,7 @@ from fastapi import Request, HTTPException, Depends
|
|
|
from fastapi.responses import JSONResponse
|
|
|
from httpx import get
|
|
|
from models.chat import ChatCompletionRequest
|
|
|
+from models.user import User
|
|
|
from services.openai_service.openai_service import generate_completion
|
|
|
from services.logging_service import log_llm_response
|
|
|
from auth.security import get_current_user
|
|
|
@@ -14,13 +15,13 @@ chat_router = APIRouter()
|
|
|
|
|
|
|
|
|
@chat_router.post("/completions")
|
|
|
-async def chat_completions(request_data: ChatCompletionRequest, request: Request, current_user=Depends(get_current_user)):
|
|
|
+async def chat_completions(request_data: ChatCompletionRequest, request: Request, current_user: User = Depends(get_current_user)):
|
|
|
"""Get chat completions from OpenAI"""
|
|
|
# Uses session_token (which is the antiAbuseToken) as an identifier for logging
|
|
|
session_identifier = request.session.get("antiAbuseToken", "unknown_session")
|
|
|
|
|
|
try:
|
|
|
- openai_response = await generate_completion(request_data.messages, session_identifier, current_user.nombre, current_user.email)
|
|
|
+ openai_response = await generate_completion(request_data.messages, session_identifier, current_user.name, current_user.email)
|
|
|
log_llm_response(request_data.user, openai_response)
|
|
|
return JSONResponse({"response": openai_response, "message": SuccessResponse.CHAT_RESPONSE_SUCCESS})
|
|
|
except HTTPException as e:
|