martsforever
152 lines
4.4 KiB
Python
152 lines
4.4 KiB
Python
import datetime
|
|
import logging
|
|
from contextlib import asynccontextmanager
|
|
from typing import List, Union
|
|
|
|
from fastapi import FastAPI
|
|
from fastapi.openapi.docs import get_swagger_ui_oauth2_redirect_html, get_redoc_html, get_swagger_ui_html
|
|
from fastapi.responses import RedirectResponse
|
|
from langchain.chat_models import init_chat_model
|
|
from langchain_core.messages import HumanMessage, AIMessage, SystemMessage
|
|
from langchain_core.prompts import ChatPromptTemplate
|
|
from langchain_core.runnables import RunnableGenerator
|
|
from langchain_openai import ChatOpenAI
|
|
from langserve import add_routes
|
|
from pydantic import BaseModel, Field
|
|
from starlette.staticfiles import StaticFiles
|
|
|
|
from app.config.env import env
|
|
from app.controller.add_test_route import add_test_route
|
|
from app.utils.get_local_ips import get_local_ips
|
|
from app.utils.llm_utils import create_llm
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
print(f"应用启动：{datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
|
|
yield
|
|
print(f"应用销毁：{datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
|
|
|
|
|
|
app = FastAPI(
|
|
lifespan=lifespan,
|
|
docs_url=None, # 禁用默认 Swagger
|
|
redoc_url=None, # 禁用默认 ReDoc
|
|
)
|
|
app.mount("/static", StaticFiles(directory="static"), name="static")
|
|
|
|
|
|
# 自定义 Swagger 页面（使用本地资源）
|
|
@app.get("/docs", include_in_schema=False)
|
|
async def custom_swagger_ui():
|
|
"""
|
|
提供自定义的 Swagger UI 文档页面
|
|
|
|
Returns:
|
|
Swagger UI HTML 页面，使用本地静态资源加载
|
|
"""
|
|
return get_swagger_ui_html(
|
|
openapi_url=app.openapi_url,
|
|
title=app.title + " - Swagger UI",
|
|
oauth2_redirect_url=app.swagger_ui_oauth2_redirect_url,
|
|
swagger_js_url="/static/swagger-ui-bundle.min.js",
|
|
swagger_css_url="/static/swagger-ui.min.css",
|
|
)
|
|
|
|
|
|
@app.get(app.swagger_ui_oauth2_redirect_url, include_in_schema=False)
|
|
async def swagger_ui_redirect():
|
|
"""
|
|
处理 Swagger UI 的 OAuth2 重定向回调
|
|
|
|
Returns:
|
|
OAuth2 重定向 HTML 响应
|
|
"""
|
|
return get_swagger_ui_oauth2_redirect_html()
|
|
|
|
|
|
@app.get("/redoc", include_in_schema=False)
|
|
async def redoc_html():
|
|
"""
|
|
提供自定义的 ReDoc 文档页面
|
|
|
|
Returns:
|
|
ReDoc HTML 页面，使用本地静态资源加载
|
|
"""
|
|
return get_redoc_html(
|
|
openapi_url=app.openapi_url,
|
|
title=app.title + " - ReDoc",
|
|
redoc_js_url="/static/redoc.standalone.js",
|
|
)
|
|
|
|
|
|
@app.get("/")
|
|
async def redirect_root_to_docs():
|
|
"""
|
|
将根路径请求重定向到 Swagger 文档页面
|
|
|
|
Returns:
|
|
重定向响应到 /docs 路径
|
|
"""
|
|
return RedirectResponse("/docs")
|
|
|
|
|
|
@app.get("/chat")
|
|
async def chat():
|
|
model_qwen = create_llm(platform_code='bailian-qwen-turbo')
|
|
model_qwen_output = model_qwen.invoke("你好，你是谁").content
|
|
print(f"model_qwen output: {model_qwen_output}")
|
|
|
|
model_doubao = create_llm(platform_code='huoshan-doubao')
|
|
model_doubao_output = model_doubao.invoke("你好，你是谁").content
|
|
print(f"model_doubao output: {model_doubao_output}")
|
|
|
|
return {
|
|
"model_qwen_output": model_qwen_output,
|
|
"model_doubao_output": model_doubao_output
|
|
}
|
|
|
|
|
|
# Edit this to add the chain you want to add
|
|
# add_routes(app, NotImplemented)
|
|
|
|
model = init_chat_model(
|
|
model="doubao-1-5-pro-32k-250115", # 你的模型名
|
|
model_provider="openai", # 必须填 openai（兼容协议）
|
|
base_url="https://ark.cn-beijing.volces.com/api/v3/", # 你的自定义 API URL
|
|
api_key="b1e7c7e0-33af-480e-b0a8-2812ba97f7b1", # 本地模型随便填
|
|
temperature=0.7,
|
|
streaming=False
|
|
)
|
|
|
|
add_routes(app=app, runnable=model, path="/doubao")
|
|
|
|
add_test_route(app)
|
|
|
|
if __name__ == "__main__":
|
|
import uvicorn
|
|
|
|
# 获取环境变量中的端口号
|
|
port = int(env.server_port)
|
|
|
|
# 打印所有可用的访问地址
|
|
print("\n服务已启动，以下是可用的访问地址：")
|
|
print(f" - 本地访问: http://127.0.0.1:{port}")
|
|
|
|
for ip in get_local_ips():
|
|
print(f" - 网络访问: http://{ip}:{port}")
|
|
|
|
print("\n") # 空行美化输出
|
|
|
|
"""
|
|
uvicorn.run() 启动了一个异步事件循环来处理 HTTP 请求，
|
|
这个循环会一直运行直到服务器被手动停止（比如按下 Ctrl+C）。
|
|
因此，uvicorn.run() 之后的代码不会被执行，直到服务器关闭。
|
|
"""
|
|
# 启动Uvicorn服务器
|
|
uvicorn.run("main:app", host="0.0.0.0", port=port)
|
|
|
|
# uvicorn.run之后的代码永远都不会执行
|
|
# 使用FastAPI的 @app.on_event("startup")装饰器可以在服务器成功启动后执行代码
|
|
print('App is running...(Never Callable)')
|