feat: basic api key support

This commit is contained in:
ceerrep 2025-02-25 14:11:39 +08:00
parent 7b2a6690ab
commit f639fbc19e
3 changed files with 5 additions and 0 deletions

View file

@ -25,6 +25,9 @@ async def chat_completion(request:Request,create:ChatCompletionCreate):
input_message = [json.loads(m.model_dump_json()) for m in create.messages]
if Config().api_key != '':
assert request.headers.get('Authorization', '').split()[-1] == Config().api_key
if create.stream:
async def inner():
chunk = ChatCompletionChunk(id=id,object='chat.completion.chunk',created=int(time()))