Toursim-Test / src /api /__init__.py
zhuhai111's picture
Upload 43 files
7cc8bc0 verified
from typing import Dict, Any
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from .llm_api import DeepseekInterface
from .search_api import GoogleSearch
def create_app(config: Dict[str, Any] = None) -> FastAPI:
"""
创建并配置 FastAPI 应用
"""
app = FastAPI(
title="Travel RAG API",
description="Travel recommendation system using RAG",
version="1.0.0"
)
# 配置 CORS
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
# 初始化配置
if config:
app.state.config = config
# 初始化 LLM
app.state.llm = DeepseekInterface(
api_key=config['deepseek_api_key'],
base_url=config['llm_settings']['deepseek']['base_url'],
model=config['llm_settings']['deepseek']['models'][0]
)
# 初始化搜索引擎在 init_app 中完成
from .routes import init_app
init_app(app)
# 导入和注册路由
from .routes import router
app.include_router(router)
return app
__all__ = ['create_app']