跳至主要內容

向量儲存庫 - 總覽

LiteLLM 有三種不同的方式可與向量儲存庫搭配運作。它們是彼此獨立的 API,且各自有不同的端點,了解您需要哪一種可省去很多混淆:

  1. LiteLLM 代管向量儲存庫:您已在提供者上有一個向量儲存庫(例如 Bedrock Knowledge Base、Vertex AI Search datastore、Azure AI Search index,...)。您只要在 LiteLLM 註冊一次,LiteLLM 就會儲存提供者、憑證和 id 對應。之後代理伺服器上的每個金鑰都能透過一個相容 OpenAI 的端點查詢它,或將其附加到 /chat/completions 和 /v1/responses 請求以進行 RAG。註冊可在 config.yaml、透過 管理 API(POST /vector_store/new),或在管理員 UI 中完成。

  2. 相容 OpenAI 的向量儲存庫 API:透過 LiteLLM 使用 OpenAI API 形式,在提供者本身上建立並管理向量儲存庫。POST /v1/vector_stores 會在上游建立新的儲存庫,/v1/vector_stores/{id}/files 會管理其檔案,而 /rag/ingest 會將上傳、分塊、嵌入與建立儲存庫包裝成一次呼叫。

  3. 透傳提供者 API:透過代理伺服器呼叫提供者的原生 API(原生請求與回應格式),例如 /vertex_ai/discovery/... 或 /bedrock/knowledgebases/...。當您需要統一 API 未公開的提供者功能時,請使用此方式。

術語

LiteLLM 中的 代管向量儲存庫 是一種註冊,不是新的儲存庫:LiteLLM 會儲存儲存庫位於哪個提供者,以及如何向其驗證,因此請求可以透過 id 來參照它。提供者端不會建立任何新內容。較舊的文件會將此概念稱為「知識庫」;兩者是相同的東西。

我需要哪個端點?​

您想要使用文件
透過單一統一 API 查詢現有的提供者儲存庫先註冊,然後 POST /v1/vector_stores/{id}/search代管向量儲存庫
在 /chat/completions 中為模型提供 RAG 內容搭配已註冊的儲存庫使用 tools: [{"type": "file_search", "vector_store_ids": [...]}]搭配 Chat Completions 使用向量儲存庫
在 /v1/responses 上使用 file_search已註冊的儲存庫 + file_search 工具File Search 教學
在提供者上建立全新的儲存庫POST /v1/vector_stores建立
一次上傳、分塊、嵌入並儲存文件POST /rag/ingestRAG 擷取
一次完成搜尋加重排序加完成POST /rag/queryRAG 查詢
管理儲存庫內的檔案/v1/vector_stores/{id}/files檔案
直接呼叫提供者的原生 API透傳路由Vertex AI Search、Azure AI(透傳)

請注意兩個名稱相似的建立端點。POST /v1/vector_stores(複數)會在提供者上建立新的儲存庫。POST /vector_store/new(單數)則會在 LiteLLM 中註冊一個既有的儲存庫。完整的管理 API 請參見 代管向量儲存庫。

提供者支援​

統一端點的支援情況會因提供者而異。Search 為 POST /v1/vector_stores/{id}/search;Create 為 POST /v1/vector_stores。

提供者 (custom_llm_provider)搜尋建立備註
openai是是也支援 files API
azure(Azure OpenAI)是是
bedrock(Knowledge Bases)是否設定
vertex_ai(RAG Engine)是是
vertex_ai/search_api(Vertex AI Search)是否將 datastore 註冊為 代管向量儲存庫
azure_ai(Azure AI Search)是否設定
gemini(File Search)是是設定
milvus是否搜尋既有 collection,設定
mongodb(BETA)是否搜尋 Atlas 或自我管理部署上的既有 MongoDB Vector Search 索引。設定、chat completions、完整範例
pg_vector是是需要 litellm-pgvector 連接器
s3_vectors是否透過 /rag/ingest 建立,設定
valkey是否搜尋既有 valkey-search 索引,設定
ragflow否是僅限資料集管理,設定

Retrieve、list、update 和 delete(GET/POST/DELETE /v1/vector_stores/{id})會原封不動地轉送 OpenAI 的請求形式,因此請將它們與公開 OpenAI 形式向量儲存庫 API 的提供者(OpenAI、Azure OpenAI)一起使用。路由細節請參見 建立。

LiteLLM Enterprise
為正式環境打造的 SSO/SAML、稽核記錄、支出追蹤、多團隊管理與防護欄。
深入瞭解 →