增加鉴权系统
This commit is contained in:
@@ -4,6 +4,7 @@ from contextlib import asynccontextmanager
|
||||
from fastapi import FastAPI
|
||||
from openai import AsyncOpenAI
|
||||
|
||||
from auth import ApiKeyAuthenticator
|
||||
from config import Settings
|
||||
from routes import router
|
||||
from service import ChatService
|
||||
@@ -37,6 +38,9 @@ def create_app(
|
||||
max_tool_rounds=resolved_settings.max_tool_rounds,
|
||||
max_tool_calls_per_turn=resolved_settings.max_tool_calls_per_turn,
|
||||
)
|
||||
app.state.authenticator = ApiKeyAuthenticator(
|
||||
resolved_settings.api_key_whitelist
|
||||
)
|
||||
yield
|
||||
|
||||
if client is None:
|
||||
|
||||
Reference in New Issue
Block a user