AI 및 fastAPI 초기 구조 셋팅

2026-06-22 16:14:02 +09:00
parent 37133ef694
commit 3a76b24df0
10 changed files with 106 additions and 0 deletions
@@ -0,0 +1,44 @@
+from fastapi import FastAPI
+from starlette.staticfiles import StaticFiles
+from contextlib import asynccontextmanager
+
+from backend.routers.stock_router import router as stock_router
+
+from backend.repository.db_init import Base, SessionLocal, engine
+
+# from backend.routers.evalu_router import router as evalu_router
+# from backend.routers.assistant_router import router as assistant_router
+# from backend.routers.call_router import router as call_router
+# from backend.repository.seed import seed_customers
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    print("서버 시작")
+
+    # Base 에 등록된 모든 모델에 테이블 자동 생성
+    Base.metadata.create_all(bind=engine)
+    print("[DB] 테이블 생성 완료 (또는 이미 존재)")
+
+    yield
+    print("서버 종료")
+
+# app = FastAPI()
+# app = FastAPI(title="상담 LLM", version="1.0", lifespan=lifespan)
+app = FastAPI(title="$ Stock AI", version="1.0", lifespan=lifespan, description="""
+    #### $주식 AI 분석 FastAPI + LangGraph + yfinance + ChromaDB
+    
+    ### 주요 기능
+    | 번호 | 기능 | 엔드포인트 |
+    | --- | --- | --- |
+    |1 | 기업 종합 분석 | 'POST /api/stock/analysis'
+    |2 | 투자 의견 조회 | 'GET /api/stock/opinion/{analysis_id}'
+    """,
+)
+
+# static 폴더 지정
+# app.mount("/static", StaticFiles(directory="backend/static"), name="static")
+
+# 라우터 등록
+app.include_router(stock_router)
+# app.include_router(assistant_router)
+# app.include_router(evalu_router)