diff options
author | kafm <wdxdesperado@qq.com> | 2024-04-28 13:16:29 +0200 |
---|---|---|
committer | kafm <wdxdesperado@qq.com> | 2024-04-28 13:16:29 +0200 |
commit | 127352de2ab01d3478d36469fe4686b70ef348c2 (patch) | |
tree | 53956efdf45a409322111fb40bf585861e25d673 | |
parent | add authorization for g4f API (diff) | |
download | gpt4free-127352de2ab01d3478d36469fe4686b70ef348c2.tar gpt4free-127352de2ab01d3478d36469fe4686b70ef348c2.tar.gz gpt4free-127352de2ab01d3478d36469fe4686b70ef348c2.tar.bz2 gpt4free-127352de2ab01d3478d36469fe4686b70ef348c2.tar.lz gpt4free-127352de2ab01d3478d36469fe4686b70ef348c2.tar.xz gpt4free-127352de2ab01d3478d36469fe4686b70ef348c2.tar.zst gpt4free-127352de2ab01d3478d36469fe4686b70ef348c2.zip |
Diffstat (limited to '')
-rw-r--r-- | g4f/api/__init__.py | 18 |
1 files changed, 5 insertions, 13 deletions
diff --git a/g4f/api/__init__.py b/g4f/api/__init__.py index f986a14f..d379653f 100644 --- a/g4f/api/__init__.py +++ b/g4f/api/__init__.py @@ -20,11 +20,9 @@ import g4f.debug from g4f.client import AsyncClient from g4f.typing import Messages -global _g4f_api_key - -def create_app(): +def create_app(g4f_api_key:str = None): app = FastAPI() - api = Api(app, g4f_api_key=_g4f_api_key) + api = Api(app, g4f_api_key=g4f_api_key) api.register_routes() api.register_authorization() api.register_validation_exception_handler() @@ -53,7 +51,6 @@ class Api: self.app = app self.client = AsyncClient() self.g4f_api_key = g4f_api_key - print(g4f_api_key) self.get_g4f_api_key = APIKeyHeader(name="g4f-api-key") def register_authorization(self): @@ -64,7 +61,6 @@ class Api: user_g4f_api_key = await self.get_g4f_api_key(request) except HTTPException as e: if e.status_code == 403: - print(e) return JSONResponse( status_code=HTTP_401_UNAUTHORIZED, content=jsonable_encoder({"detail": "G4F API key required"}), @@ -184,13 +180,9 @@ def run_api( bind: str = None, debug: bool = False, workers: int = None, - g4f_api_key: str = None, - use_colors: bool = None + use_colors: bool = None, + g4f_api_key: str = None ) -> None: - - global _g4f_api_key - _g4f_api_key = g4f_api_key - print(f'Starting server... [g4f v-{g4f.version.utils.current_version}]' + (" (debug)" if debug else "")) if use_colors is None: use_colors = debug @@ -198,4 +190,4 @@ def run_api( host, port = bind.split(":") if debug: g4f.debug.logging = True - uvicorn.run("g4f.api:create_app", host=host, port=int(port), workers=workers, use_colors=use_colors, factory=True)
\ No newline at end of file + uvicorn.run(create_app(g4f_api_key), host=host, port=int(port), workers=workers, use_colors=use_colors)
\ No newline at end of file |