fix sqlalchemy dep loading in deps.py

This commit is contained in:
phernandez
2024-12-30 14:08:21 -06:00
parent a4283d893f
commit 566a796d68
3 changed files with 69 additions and 21 deletions
+17 -8
View File
@@ -1,25 +1,34 @@
"""FastAPI application for basic-memory knowledge graph API."""
from contextlib import asynccontextmanager
from fastapi import FastAPI
from loguru import logger
from basic_memory import db
from .routers import documents
from .routers import knowledge
from .routers import discovery
@asynccontextmanager
async def lifespan(app: FastAPI):
"""Lifecycle manager for the FastAPI app."""
logger.info("Starting Basic Memory API")
yield
logger.info("Shutting down Basic Memory API")
await db.shutdown_db()
# Initialize FastAPI app
app = FastAPI(
title="Basic Memory API", description="Knowledge graph API for basic-memory", version="0.1.0"
title="Basic Memory API",
description="Knowledge graph API for basic-memory",
version="0.1.0",
lifespan=lifespan,
)
# Include routers
app.include_router(knowledge.router)
app.include_router(documents.router)
app.include_router(discovery.router)
# Add startup event
@app.on_event("startup")
async def startup_event():
"""Log when the API starts."""
logger.info("Starting Basic Memory API")