Explorar el Código

fix: fix on translate

latapp hace 9 meses
padre
commit
b5c2d5555d
Se han modificado 2 ficheros con 5 adiciones y 3 borrados
  1. 2 1
      auth/security.py
  2. 3 2
      routes/chat.py

+ 2 - 1
auth/security.py

@@ -10,6 +10,7 @@ from config.settings import SECRET_KEY
 from jose import jwt, JWTError
 from passlib.context import CryptContext
 
+from models.user import User
 from services.data_service import UserDataService
 
 
@@ -57,7 +58,7 @@ def authenticate_user(email: str, password: str) -> bool:
         return False
     return True
 
-async def get_current_user(credentials: HTTPAuthorizationCredentials = Depends(security)):
+async def get_current_user(credentials: HTTPAuthorizationCredentials = Depends(security)) -> User:
     credentials_exception = HTTPException(
         status_code=401,
         detail="No se pudieron validar las credenciales",

+ 3 - 2
routes/chat.py

@@ -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: