Fix mypy issues

This commit is contained in:
Sameer Kankute 2026-02-17 20:51:51 +05:30
parent 90be817b32
commit 842d5a0762
3 changed files with 9 additions and 4 deletions

View file

@ -9544,7 +9544,7 @@ class BaseLLMHTTPHandler:
try:
response = await async_httpx_client.get(
url=url, headers=headers, params=query_params, timeout=timeout
url=url, headers=headers, params=query_params
)
except Exception as e:
raise self._handle_error(
@ -9649,7 +9649,7 @@ class BaseLLMHTTPHandler:
try:
response = await async_httpx_client.get(
url=url, headers=headers, timeout=timeout
url=url, headers=headers
)
except Exception as e:
raise self._handle_error(

View file

@ -532,6 +532,11 @@ class ProxyBaseLLMRequestProcessing:
"aupdate_eval",
"adelete_eval",
"acancel_eval",
"acreate_run",
"alist_runs",
"aget_run",
"acancel_run",
"adelete_run",
],
version: Optional[str] = None,
user_model: Optional[str] = None,

View file

@ -775,7 +775,7 @@ async def list_runs(
general_settings=general_settings,
proxy_config=proxy_config,
select_data_generator=select_data_generator,
model=data.get("model"),
model=str(data.get("model")) if data.get("model") is not None else None,
user_model=user_model,
user_temperature=user_temperature,
user_request_timeout=user_request_timeout,
@ -848,7 +848,7 @@ async def get_run(
if model:
data["model"] = model
if "custom_llm_provider" not in data:
if "custom_llm_provider" not in data and custom_llm_provider is not None:
data["custom_llm_provider"] = custom_llm_provider
# Process request using ProxyBaseLLMRequestProcessing