litellm/litellm/vector_stores/utils.py

from typing import Any, Final, 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: Final = get_type_hints(VectorStoreSearchOptionalRequestParams).keys()
        filtered_params: Final = {k: v for k, v in params.items() if k in valid_keys and v is not None}

        optional_params: Final = 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: Final = get_type_hints(VectorStoreCreateOptionalRequestParams).keys()
        filtered_params: Final = {k: v for k, v in params.items() if k in valid_keys and v is not None}

        return cast(VectorStoreCreateOptionalRequestParams, filtered_params)