mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-11 03:38:38 +00:00
fix: qualify list annotations for Python 3.14 runtime inspection
This commit is contained in:
parent
cfb0b59e34
commit
1988dfc4cc
4 changed files with 13 additions and 9 deletions
|
|
@ -32,7 +32,7 @@ class ModelsManagementClient:
|
|||
headers["Authorization"] = f"Bearer {self._api_key}"
|
||||
return headers
|
||||
|
||||
def list(self, return_request: bool = False) -> list[dict[str, Any]] | requests.Request:
|
||||
def list(self, return_request: bool = False) -> builtins.list[dict[str, Any]] | requests.Request:
|
||||
"""
|
||||
Get the list of models supported by the server.
|
||||
|
||||
|
|
|
|||
|
|
@ -40,7 +40,7 @@ class TeamsManagementClient:
|
|||
self,
|
||||
user_id: str | None = None,
|
||||
organization_id: str | None = None,
|
||||
) -> list[dict[str, Any]]:
|
||||
) -> builtins.list[dict[str, Any]]:
|
||||
"""
|
||||
List teams that the user belongs to.
|
||||
|
||||
|
|
|
|||
|
|
@ -39,7 +39,7 @@ base_llm_http_handler = BaseLLMHTTPHandler()
|
|||
|
||||
|
||||
def mock_vector_store_search_response(
|
||||
mock_results: list[VectorStoreSearchResult] | None = None,
|
||||
mock_results: builtins.list[VectorStoreSearchResult] | None = None,
|
||||
):
|
||||
"""Mock response for vector store search"""
|
||||
if mock_results is None:
|
||||
|
|
@ -93,7 +93,7 @@ def mock_vector_store_create_response(
|
|||
@client
|
||||
async def acreate(
|
||||
name: str | None = None,
|
||||
file_ids: list[str] | None = None,
|
||||
file_ids: builtins.list[str] | None = None,
|
||||
expires_after: dict | None = None,
|
||||
chunking_strategy: dict | None = None,
|
||||
metadata: dict[str, str] | None = None,
|
||||
|
|
@ -157,7 +157,7 @@ async def acreate(
|
|||
@client
|
||||
def create(
|
||||
name: str | None = None,
|
||||
file_ids: list[str] | None = None,
|
||||
file_ids: builtins.list[str] | None = None,
|
||||
expires_after: dict | None = None,
|
||||
chunking_strategy: dict | None = None,
|
||||
metadata: dict[str, str] | None = None,
|
||||
|
|
@ -270,7 +270,7 @@ def create(
|
|||
@client
|
||||
async def asearch(
|
||||
vector_store_id: str,
|
||||
query: str | list[str],
|
||||
query: str | builtins.list[str],
|
||||
filters: dict | None = None,
|
||||
max_num_results: int | None = None,
|
||||
ranking_options: dict | None = None,
|
||||
|
|
@ -339,7 +339,7 @@ async def asearch(
|
|||
@client
|
||||
def search(
|
||||
vector_store_id: str,
|
||||
query: str | list[str],
|
||||
query: str | builtins.list[str],
|
||||
filters: dict | None = None,
|
||||
max_num_results: int | None = None,
|
||||
ranking_options: dict | None = None,
|
||||
|
|
|
|||
|
|
@ -9,6 +9,8 @@ serialization trap).
|
|||
|
||||
from unittest.mock import MagicMock, patch
|
||||
|
||||
import pytest
|
||||
|
||||
import litellm.vector_stores.main as vector_stores_main
|
||||
from litellm.vector_stores.main import search
|
||||
|
||||
|
|
@ -19,7 +21,8 @@ MOCK_SEARCH_RESPONSE = {
|
|||
}
|
||||
|
||||
|
||||
def test_search_threads_router_to_handler():
|
||||
@pytest.mark.parametrize("query", ["q", ["q", "another question"]])
|
||||
def test_search_threads_router_to_handler(query: str | list[str]):
|
||||
"""search() must pass its router param through to the HTTP handler"""
|
||||
mock_router = MagicMock()
|
||||
logger = MagicMock()
|
||||
|
|
@ -37,7 +40,7 @@ def test_search_threads_router_to_handler():
|
|||
):
|
||||
response = search(
|
||||
vector_store_id="bkt:idx",
|
||||
query="q",
|
||||
query=query,
|
||||
custom_llm_provider="s3_vectors",
|
||||
router=mock_router,
|
||||
litellm_logging_obj=logger,
|
||||
|
|
@ -46,6 +49,7 @@ def test_search_threads_router_to_handler():
|
|||
assert response == MOCK_SEARCH_RESPONSE
|
||||
mock_handler.assert_called_once()
|
||||
assert mock_handler.call_args.kwargs["router"] is mock_router
|
||||
assert mock_handler.call_args.kwargs["query"] == query
|
||||
|
||||
|
||||
def test_search_router_not_in_litellm_params():
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue