scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 86

fix(types): resolve mypy errors in search_tool_management.py

Browse source 
Add type casts for TypedDict .get() calls and rename loop variable to
avoid type shadowing between config and DB search tool iterations.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
yuneng-jiang 2026-03-21 17:31:48 -07:00
parent 5da2f35fd3
commit 0ca2472d83
1 changed files with 9 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
litellm/proxy/search_endpoints/search_tool_management.py
View file

@ -2,7 +2,7 @@
CRUD ENDPOINTS FOR SEARCH TOOLS
"""
from datetime import datetime
from typing import Any, Dict, List, Union
from typing import Any, Dict, List, Optional, Union, cast
from fastapi import APIRouter, Depends, HTTPException
from pydantic import BaseModel
@ -130,7 +130,7 @@ async def list_search_tools(
search_tool_id=None,
search_tool_name=tool_name,
litellm_params=masked_litellm_params_dict,
search_tool_info=search_tool.get("search_tool_info"),
search_tool_info=cast(Optional[dict], search_tool.get("search_tool_info")),
created_at=None,
updated_at=None,
is_from_config=True,
@ -143,8 +143,8 @@ async def list_search_tools(
if tool.get("search_tool_name") not in db_tool_names
]
for search_tool in search_tools_from_db:
litellm_params_dict = dict(search_tool.get("litellm_params", {}))
for db_tool in search_tools_from_db:
litellm_params_dict = dict(db_tool.get("litellm_params", {}))
masked_litellm_params_dict = _get_masked_values(
litellm_params_dict,
unmasked_length=4,
@ -153,12 +153,12 @@ async def list_search_tools(
search_tool_configs.append(
SearchToolInfoResponse(
search_tool_id=search_tool.get("search_tool_id"),
search_tool_name=search_tool.get("search_tool_name", ""),
search_tool_id=cast(Optional[str], db_tool.get("search_tool_id")),
search_tool_name=db_tool.get("search_tool_name", ""),
litellm_params=masked_litellm_params_dict,
search_tool_info=search_tool.get("search_tool_info"),
created_at=_convert_datetime_to_str(search_tool.get("created_at")),
updated_at=_convert_datetime_to_str(search_tool.get("updated_at")),
search_tool_info=cast(Optional[dict], db_tool.get("search_tool_info")),
created_at=_convert_datetime_to_str(cast(Optional[Union[datetime, str]], db_tool.get("created_at"))),
updated_at=_convert_datetime_to_str(cast(Optional[Union[datetime, str]], db_tool.get("updated_at"))),
is_from_config=False,
)
)