diff --git a/app/constants.py b/app/constants.py index 64f388a..b2acd7c 100644 --- a/app/constants.py +++ b/app/constants.py @@ -15,7 +15,7 @@ RECAPTCHA_SECRET = getenv("CAPTCHA_SECRET") RECAPTCHA_API_URL = getenv("API_RECAPTCHA_URL", "https://www.google.com/recaptcha/api/siteverify") -def inicializar_modelos() -> dict: +def inicializar_modelos_ml() -> dict: PATH_BASE = Path(__file__).resolve().parent with open(f"{PATH_BASE}/bin/explicador.pkl", "rb") as archivo: @@ -42,4 +42,4 @@ def cargar_credenciales_cliente_firebase() -> dict[str, str]: "measurementId": getenv("CLIENTE_FIREBASE_MEASUREMENT_ID"), "driveScopes": obtener_lista_dominios(getenv("CLIENTE_DRIVE_SCOPES", "")), "reCAPTCHA": getenv("CLIENTE_CAPTCHA"), - } \ No newline at end of file + } diff --git a/app/main.py b/app/main.py index f5e8775..7dc2d0f 100644 --- a/app/main.py +++ b/app/main.py @@ -21,7 +21,7 @@ async def inicializar_modelos(app: FastAPI): # Esto se ejecuta al iniciar el backend FIREBASE_APP = inicializar_firebase() - MODELOS = inicializar_modelos() + MODELOS = inicializar_modelos_ml() CREDS_FIREBASE_CLIENTE = cargar_credenciales_cliente_firebase() yield { @@ -120,4 +120,4 @@ async def manejar_error_interno(peticion: Request, excepcion: ErrorInterno): excepcion.mensaje, status_code=400, media_type="application/json", - ) \ No newline at end of file + ) diff --git a/articulo b/articulo new file mode 160000 index 0000000..2c88c5f --- /dev/null +++ b/articulo @@ -0,0 +1 @@ +Subproject commit 2c88c5f340779b968a5303a64ae50d53613bdcb5 diff --git a/tests/scripts/api/test_main.py b/tests/scripts/api/test_main.py index c84a632..32114a6 100644 --- a/tests/scripts/api/test_main.py +++ b/tests/scripts/api/test_main.py @@ -7,7 +7,7 @@ from firebase_admin.credentials import Certificate from app.main import app from contextlib import asynccontextmanager -from app.constants import cargar_credenciales_cliente_firebase, inicializar_modelos +from app.constants import cargar_credenciales_cliente_firebase, inicializar_modelos_ml # Constantes de prueba TEST_CREDS = { @@ -201,8 +201,8 @@ def mock_sesion(x, providers): mocker.patch("app.constants.jload", return_value=MOCK_TEXTOS) mocker.patch("app.constants.InferenceSession", side_effect=mock_sesion) - RES = inicializar_modelos() + RES = inicializar_modelos_ml() assert RES["explicador"] == "explicador_mock" assert RES["textos"] == MOCK_TEXTOS - assert RES["modelo"] == "modelo_mock" \ No newline at end of file + assert RES["modelo"] == "modelo_mock"