diff --git a/litellm/utils.py b/litellm/utils.py index 019fbc2add8..c26314b6225 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -8910,7 +8910,7 @@ class ProviderConfigManager: def get_provider_vector_store_files_config( provider: LlmProviders, ) -> Optional[BaseVectorStoreFilesConfig]: - if litellm.LlmProviders.OPENAI == provider: + if litellm.LlmProviders.OPENAI == provider or litellm.LlmProviders.PG_VECTOR == provider: from litellm.llms.openai.vector_store_files.transformation import ( OpenAIVectorStoreFilesConfig, )