diff --git a/agent_service/main.py b/agent_service/main.py index dbac35b..aee302d 100644 --- a/agent_service/main.py +++ b/agent_service/main.py @@ -105,7 +105,7 @@ async def lifespan(app: FastAPI): # 5. Memory manager try: from .memory.memory_manager import MemoryManager - memory_mgr = MemoryManager(pool=pool, llm=llm_router) if pool else None + memory_mgr = MemoryManager(pool=pool, llm_router=llm_router) if pool else None except Exception as exc: logger.error('Failed to init memory manager: %s', exc) memory_mgr = None diff --git a/agent_service/migrations/alembic.ini b/agent_service/migrations/alembic.ini index b9f1ad8..3832eb3 100644 --- a/agent_service/migrations/alembic.ini +++ b/agent_service/migrations/alembic.ini @@ -1,5 +1,5 @@ [alembic] -script_location = migrations +script_location = agent_service/migrations prepend_sys_path = . version_path_separator = os sqlalchemy.url = postgresql+asyncpg://%(POSTGRES_USER)s:%(POSTGRES_PASSWORD)s@%(POSTGRES_HOST)s:%(POSTGRES_PORT)s/%(POSTGRES_DB)s