phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(pass_through_endpoints.py): fix linting error

Browse source
This commit is contained in:
Krrish Dholakia 2025-03-12 12:00:05 -07:00
parent d2c79199b1
commit c7ceeaa4d7
1 changed files with 6 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm/proxy/pass_through_endpoints/pass_through_endpoints.py
View file

@ -3,8 +3,8 @@ import asyncio
import json
from base64 import b64encode
from datetime import datetime
from typing import List, Optional, Union, Dict
from urllib.parse import urlencode, parse_qs, urlparse
from typing import Dict, List, Optional, Union
from urllib.parse import parse_qs, urlencode, urlparse
import httpx
from fastapi import APIRouter, Depends, HTTPException, Request, Response, status
@ -260,6 +260,7 @@ async def chat_completion_pass_through_endpoint( # noqa: PLR0915
code=getattr(e, "status_code", 500),
)
class HttpPassThroughEndpointHelpers:
@staticmethod
def forward_headers_from_request(
@ -315,11 +316,11 @@ class HttpPassThroughEndpointHelpers:
existing_query_params = parse_qs(existing_query_string)
# parse_qs returns a dict where each value is a list, so let's flatten it
existing_query_params = {
updated_existing_query_params = {
k: v[0] if len(v) == 1 else v for k, v in existing_query_params.items()
}
# Merge the query params, giving priority to the existing ones
return {**request_query_params, **existing_query_params}
return {**request_query_params, **updated_existing_query_params}
@staticmethod
async def _make_non_streaming_http_request(
@ -352,6 +353,7 @@ class HttpPassThroughEndpointHelpers:
)
return response
async def pass_through_request( # noqa: PLR0915
request: Request,
target: str,