diff --git a/litellm/proxy/_types.py b/litellm/proxy/_types.py index 67cbb6063ea..b944df3a3e5 100644 --- a/litellm/proxy/_types.py +++ b/litellm/proxy/_types.py @@ -1,7 +1,7 @@ from pydantic import BaseModel, Extra, Field from typing import Optional, List, Union, Dict, Literal from datetime import datetime -import uuid +import uuid, json ######### Request Class Definition ###### class ProxyChatCompletionRequest(BaseModel): model: str @@ -84,6 +84,9 @@ class GenerateKeyRequest(BaseModel): user_id: Optional[str] = None max_parallel_requests: Optional[int] = None + def json(self, **kwargs) -> str: + return json.dumps(self.dict(), **kwargs) + class GenerateKeyResponse(BaseModel): key: str expires: datetime diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index e63083c9309..b4bcbedb66f 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -1128,9 +1128,7 @@ async def generate_key_fn(request: Request, data: GenerateKeyRequest, Authorizat - user_id: (str) Unique user id - used for tracking spend across multiple keys for same user id. """ # data = await request.json() - print(f"type of object: {type(data)}") - data_json = json.loads(data.model_dump_json()) # type: ignore - print(f"data_json: {data_json}") + data_json = json.loads(data.json()) # type: ignore response = await generate_key_helper_fn(**data_json) return GenerateKeyResponse(key=response["token"], expires=response["expires"], user_id=response["user_id"])