mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-25 01:02:15 +00:00
b02be1ba70
* feat(milvus/): initial commit adding milvus vector store support to LiteLLM allows querying milvus vector store through litellm * feat(bedrock/vector_stores): support translating openai filters param to aws kb adds filtering to aws kb * feat(milvus/): add milvus vector store unified search support allows calling milvus vector store in through chat completions * docs(milvus_vector_stores.md): document new milvus vector search integration * feat(pass_through_endpoints.py): support passing form data through to a passthrough endpoint Closes LIT-1147 * fix: fix linting errors
58 lines
2.1 KiB
Python
58 lines
2.1 KiB
Python
from typing import Any, Dict, cast, get_type_hints
|
|
|
|
from litellm.llms.base_llm.vector_store.transformation import BaseVectorStoreConfig
|
|
from litellm.types.vector_stores import (
|
|
VectorStoreCreateOptionalRequestParams,
|
|
VectorStoreSearchOptionalRequestParams,
|
|
)
|
|
|
|
|
|
class VectorStoreRequestUtils:
|
|
"""Helper utils for constructing Vector Store search requests"""
|
|
|
|
@staticmethod
|
|
def get_requested_vector_store_search_optional_param(
|
|
params: Dict[str, Any],
|
|
vector_store_provider_config: BaseVectorStoreConfig,
|
|
) -> VectorStoreSearchOptionalRequestParams:
|
|
"""
|
|
Filter parameters to only include those defined in VectorStoreSearchOptionalRequestParams.
|
|
|
|
Args:
|
|
params: Dictionary of parameters to filter
|
|
|
|
Returns:
|
|
VectorStoreSearchOptionalRequestParams instance with only the valid parameters
|
|
"""
|
|
valid_keys = get_type_hints(VectorStoreSearchOptionalRequestParams).keys()
|
|
filtered_params = {
|
|
k: v for k, v in params.items() if k in valid_keys and v is not None
|
|
}
|
|
|
|
optional_params = vector_store_provider_config.map_openai_params(
|
|
non_default_params=params,
|
|
optional_params=filtered_params,
|
|
drop_params=False,
|
|
)
|
|
|
|
return cast(VectorStoreSearchOptionalRequestParams, optional_params)
|
|
|
|
@staticmethod
|
|
def get_requested_vector_store_create_optional_param(
|
|
params: Dict[str, Any],
|
|
) -> VectorStoreCreateOptionalRequestParams:
|
|
"""
|
|
Filter parameters to only include those defined in VectorStoreCreateOptionalRequestParams.
|
|
|
|
Args:
|
|
params: Dictionary of parameters to filter
|
|
|
|
Returns:
|
|
VectorStoreCreateOptionalRequestParams instance with only the valid parameters
|
|
"""
|
|
valid_keys = get_type_hints(VectorStoreCreateOptionalRequestParams).keys()
|
|
filtered_params = {
|
|
k: v for k, v in params.items() if k in valid_keys and v is not None
|
|
}
|
|
|
|
return cast(VectorStoreCreateOptionalRequestParams, filtered_params)
|