phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 19:24:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat - support /create assistants endpoint

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-09 10:03:47 -07:00
parent 0f43869706
commit bce7b5f8c8
2 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/litellm_pre_call_utils.py
View file

@ -33,11 +33,11 @@ def _get_metadata_variable_name(request: Request) -> str:
""" """
Helper to return what the "metadata" field should be called in the request data Helper to return what the "metadata" field should be called in the request data
For all /thread endpoints we need to call this "litellm_metadata" For all /thread or /assistant endpoints we need to call this "litellm_metadata"
For ALL other endpoints we call this "metadata For ALL other endpoints we call this "metadata
""" """
if "thread" in request.url.path: if "thread" in request.url.path or "assistant" in request.url.path:
return "litellm_metadata" return "litellm_metadata"
else: else:
return "metadata" return "metadata"

2
litellm/proxy/proxy_server.py
View file

@ -3975,10 +3975,10 @@ async def create_assistant(
API Reference docs - https://platform.openai.com/docs/api-reference/assistants/createAssistant API Reference docs - https://platform.openai.com/docs/api-reference/assistants/createAssistant
""" """
global proxy_logging_obj global proxy_logging_obj
data: Dict = {}
try: try:
# Use orjson to parse JSON data, orjson speeds up requests significantly # Use orjson to parse JSON data, orjson speeds up requests significantly
body = await request.body() body = await request.body()
data = orjson.loads(body)
# Include original request and headers in the data # Include original request and headers in the data
data = await add_litellm_data_to_request( data = await add_litellm_data_to_request(