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, ) )