Add support for vector store files endpoints

litellm/__init__.py

@@ -1372,6 +1372,20 @@ def add_known_models():
 from .realtime_api.main import _arealtime
 from .fine_tuning.main import *
 from .files.main import *
+from .vector_store_files.main import (
+    acreate as avector_store_file_create,
+    adelete as avector_store_file_delete,
+    alist as avector_store_file_list,
+    aretrieve as avector_store_file_retrieve,
+    aretrieve_content as avector_store_file_content,
+    aupdate as avector_store_file_update,
+    create as vector_store_file_create,
+    delete as vector_store_file_delete,
+    list as vector_store_file_list,
+    retrieve as vector_store_file_retrieve,
+    retrieve_content as vector_store_file_content,
+    update as vector_store_file_update,
+)
 from .scheduler import *
 from .cost_calculator import response_cost_calculator, cost_per_token
 

litellm/llms/base_llm/vector_store_files/transformation.py

@@ -0,0 +1,226 @@
+from abc import ABC, abstractmethod
+from typing import TYPE_CHECKING, Any, Dict, Optional, Tuple, Union
+
+import httpx
+
+from litellm.types.router import GenericLiteLLMParams
+from litellm.types.vector_store_files import (
+    VectorStoreFileAuthCredentials,
+    VectorStoreFileChunkingStrategy,
+    VectorStoreFileContentResponse,
+    VectorStoreFileCreateRequest,
+    VectorStoreFileDeleteResponse,
+    VectorStoreFileListQueryParams,
+    VectorStoreFileListResponse,
+    VectorStoreFileObject,
+    VectorStoreFileUpdateRequest,
+)
+
+if TYPE_CHECKING:
+    from litellm.litellm_core_utils.litellm_logging import Logging as _LiteLLMLoggingObj
+
+    from ..chat.transformation import BaseLLMException as _BaseLLMException
+
+    LiteLLMLoggingObj = _LiteLLMLoggingObj
+    BaseLLMException = _BaseLLMException
+else:
+    LiteLLMLoggingObj = Any
+    BaseLLMException = Any
+
+
+class BaseVectorStoreFilesConfig(ABC):
+    """Base configuration contract for provider-specific vector store file implementations."""
+
+    def get_supported_openai_params(
+        self,
+        operation: str,
+    ) -> Tuple[str, ...]:
+        """Return the set of OpenAI params supported for the given operation."""
+
+        return tuple()
+
+    def map_openai_params(
+        self,
+        *,
+        operation: str,
+        non_default_params: Dict[str, Any],
+        optional_params: Dict[str, Any],
+        drop_params: bool,
+    ) -> Dict[str, Any]:
+        """Map non-default OpenAI params to provider-specific params."""
+
+        return optional_params
+
+    @abstractmethod
+    def get_auth_credentials(
+        self, litellm_params: Dict[str, Any]
+    ) -> VectorStoreFileAuthCredentials:
+        ...
+
+    @abstractmethod
+    def get_vector_store_file_endpoints_by_type(self) -> Dict[
+        str, Tuple[Tuple[str, str], ...]
+    ]:
+        ...
+
+    @abstractmethod
+    def validate_environment(
+        self,
+        *,
+        headers: Dict[str, str],
+        litellm_params: Optional[GenericLiteLLMParams],
+    ) -> Dict[str, str]:
+        return {}
+
+    @abstractmethod
+    def get_complete_url(
+        self,
+        *,
+        api_base: Optional[str],
+        vector_store_id: str,
+        litellm_params: Dict[str, Any],
+    ) -> str:
+        if api_base is None:
+            raise ValueError("api_base is required")
+        return api_base
+
+    @abstractmethod
+    def transform_create_vector_store_file_request(
+        self,
+        *,
+        vector_store_id: str,
+        create_request: VectorStoreFileCreateRequest,
+        api_base: str,
+    ) -> Tuple[str, Dict[str, Any]]:
+        ...
+
+    @abstractmethod
+    def transform_create_vector_store_file_response(
+        self,
+        *,
+        response: httpx.Response,
+    ) -> VectorStoreFileObject:
+        ...
+
+    @abstractmethod
+    def transform_list_vector_store_files_request(
+        self,
+        *,
+        vector_store_id: str,
+        query_params: VectorStoreFileListQueryParams,
+        api_base: str,
+    ) -> Tuple[str, Dict[str, Any]]:
+        ...
+
+    @abstractmethod
+    def transform_list_vector_store_files_response(
+        self,
+        *,
+        response: httpx.Response,
+    ) -> VectorStoreFileListResponse:
+        ...
+
+    @abstractmethod
+    def transform_retrieve_vector_store_file_request(
+        self,
+        *,
+        vector_store_id: str,
+        file_id: str,
+        api_base: str,
+    ) -> Tuple[str, Dict[str, Any]]:
+        ...
+
+    @abstractmethod
+    def transform_retrieve_vector_store_file_response(
+        self,
+        *,
+        response: httpx.Response,
+    ) -> VectorStoreFileObject:
+        ...
+
+    @abstractmethod
+    def transform_retrieve_vector_store_file_content_request(
+        self,
+        *,
+        vector_store_id: str,
+        file_id: str,
+        api_base: str,
+    ) -> Tuple[str, Dict[str, Any]]:
+        ...
+
+    @abstractmethod
+    def transform_retrieve_vector_store_file_content_response(
+        self,
+        *,
+        response: httpx.Response,
+    ) -> VectorStoreFileContentResponse:
+        ...
+
+    @abstractmethod
+    def transform_update_vector_store_file_request(
+        self,
+        *,
+        vector_store_id: str,
+        file_id: str,
+        update_request: VectorStoreFileUpdateRequest,
+        api_base: str,
+    ) -> Tuple[str, Dict[str, Any]]:
+        ...
+
+    @abstractmethod
+    def transform_update_vector_store_file_response(
+        self,
+        *,
+        response: httpx.Response,
+    ) -> VectorStoreFileObject:
+        ...
+
+    @abstractmethod
+    def transform_delete_vector_store_file_request(
+        self,
+        *,
+        vector_store_id: str,
+        file_id: str,
+        api_base: str,
+    ) -> Tuple[str, Dict[str, Any]]:
+        ...
+
+    @abstractmethod
+    def transform_delete_vector_store_file_response(
+        self,
+        *,
+        response: httpx.Response,
+    ) -> VectorStoreFileDeleteResponse:
+        ...
+
+    def get_error_class(
+        self,
+        *,
+        error_message: str,
+        status_code: int,
+        headers: Union[Dict[str, Any], httpx.Headers],
+    ) -> BaseLLMException:
+        from ..chat.transformation import BaseLLMException
+
+        raise BaseLLMException(
+            status_code=status_code,
+            message=error_message,
+            headers=headers,
+        )
+
+    def sign_request(
+        self,
+        *,
+        headers: Dict[str, str],
+        optional_params: Dict[str, Any],
+        request_data: Dict[str, Any],
+        api_base: str,
+        api_key: Optional[str] = None,
+    ) -> Tuple[Dict[str, str], Optional[bytes]]:
+        return headers, None
+
+    def prepare_chunking_strategy(
+        self,
+        chunking_strategy: Optional[VectorStoreFileChunkingStrategy],
+    ) -> Optional[VectorStoreFileChunkingStrategy]:
+        return chunking_strategy