chore: initial public snapshot for github upload

llm-gateway-competitors/litellm-wheel-src/litellm/vector_stores/utils.py

@@ -0,0 +1,58 @@
+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)