Refatora a inicialização dos serviços no main.py e remove importações duplicadas no auth_routes.py. Adiciona serviços de sessão, artefatos e memória para melhor gerenciamento de estado.
This commit is contained in:
parent
215e76012e
commit
7af234ef48
@ -12,6 +12,7 @@ from src.schemas.user import (
|
|||||||
MessageResponse,
|
MessageResponse,
|
||||||
)
|
)
|
||||||
from src.services.user_service import (
|
from src.services.user_service import (
|
||||||
|
authenticate_user,
|
||||||
create_user,
|
create_user,
|
||||||
verify_email,
|
verify_email,
|
||||||
resend_verification,
|
resend_verification,
|
||||||
@ -19,7 +20,6 @@ from src.services.user_service import (
|
|||||||
reset_password,
|
reset_password,
|
||||||
)
|
)
|
||||||
from src.services.auth_service import (
|
from src.services.auth_service import (
|
||||||
authenticate_user,
|
|
||||||
create_access_token,
|
create_access_token,
|
||||||
get_current_admin_user,
|
get_current_admin_user,
|
||||||
get_current_user,
|
get_current_user,
|
||||||
|
22
src/main.py
22
src/main.py
@ -9,6 +9,18 @@ sys.path.append(str(root_dir))
|
|||||||
from fastapi import FastAPI
|
from fastapi import FastAPI
|
||||||
from fastapi.middleware.cors import CORSMiddleware
|
from fastapi.middleware.cors import CORSMiddleware
|
||||||
from src.config.database import engine, Base
|
from src.config.database import engine, Base
|
||||||
|
from src.config.settings import settings
|
||||||
|
from src.utils.logger import setup_logger
|
||||||
|
from google.adk.artifacts.in_memory_artifact_service import InMemoryArtifactService
|
||||||
|
from google.adk.sessions import DatabaseSessionService
|
||||||
|
from google.adk.memory import InMemoryMemoryService
|
||||||
|
|
||||||
|
# Initialize service instances
|
||||||
|
session_service = DatabaseSessionService(db_url=settings.POSTGRES_CONNECTION_STRING)
|
||||||
|
artifacts_service = InMemoryArtifactService()
|
||||||
|
memory_service = InMemoryMemoryService()
|
||||||
|
|
||||||
|
# Import routers after service initialization to avoid circular imports
|
||||||
from src.api.auth_routes import router as auth_router
|
from src.api.auth_routes import router as auth_router
|
||||||
from src.api.admin_routes import router as admin_router
|
from src.api.admin_routes import router as admin_router
|
||||||
from src.api.chat_routes import router as chat_router
|
from src.api.chat_routes import router as chat_router
|
||||||
@ -18,20 +30,10 @@ from src.api.contact_routes import router as contact_router
|
|||||||
from src.api.mcp_server_routes import router as mcp_server_router
|
from src.api.mcp_server_routes import router as mcp_server_router
|
||||||
from src.api.tool_routes import router as tool_router
|
from src.api.tool_routes import router as tool_router
|
||||||
from src.api.client_routes import router as client_router
|
from src.api.client_routes import router as client_router
|
||||||
from src.config.settings import settings
|
|
||||||
from src.utils.logger import setup_logger
|
|
||||||
from google.adk.artifacts.in_memory_artifact_service import InMemoryArtifactService
|
|
||||||
from google.adk.sessions import DatabaseSessionService
|
|
||||||
from google.adk.memory import InMemoryMemoryService
|
|
||||||
|
|
||||||
# Configure logger
|
# Configure logger
|
||||||
logger = setup_logger(__name__)
|
logger = setup_logger(__name__)
|
||||||
|
|
||||||
|
|
||||||
session_service = DatabaseSessionService(db_url=settings.POSTGRES_CONNECTION_STRING)
|
|
||||||
artifacts_service = InMemoryArtifactService()
|
|
||||||
memory_service = InMemoryMemoryService()
|
|
||||||
|
|
||||||
# FastAPI initialization
|
# FastAPI initialization
|
||||||
app = FastAPI(
|
app = FastAPI(
|
||||||
title=settings.API_TITLE,
|
title=settings.API_TITLE,
|
||||||
|
Loading…
Reference in New Issue
Block a user