From 0ca2472d83d9dc4e2a861d51e3f1c3f9061984aa Mon Sep 17 00:00:00 2001 From: yuneng-jiang Date: Sat, 21 Mar 2026 17:31:48 -0700 Subject: [PATCH] fix(types): resolve mypy errors in search_tool_management.py 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) --- .../search_endpoints/search_tool_management.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/litellm/proxy/search_endpoints/search_tool_management.py b/litellm/proxy/search_endpoints/search_tool_management.py index fac12ebf256..c9d415c0b9a 100644 --- a/litellm/proxy/search_endpoints/search_tool_management.py +++ b/litellm/proxy/search_endpoints/search_tool_management.py @@ -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, ) )