diff --git a/ui/litellm-dashboard/src/components/vector_store_providers.tsx b/ui/litellm-dashboard/src/components/vector_store_providers.tsx index a6f222da67..68c5c3745c 100644 --- a/ui/litellm-dashboard/src/components/vector_store_providers.tsx +++ b/ui/litellm-dashboard/src/components/vector_store_providers.tsx @@ -4,6 +4,7 @@ export enum VectorStoreProviders { VertexRagEngine = "Vertex AI RAG Engine", OpenAI = "OpenAI", Azure = "Azure OpenAI", + Milvus = "Milvus", } export const vectorStoreProviderMap: Record = { @@ -12,6 +13,7 @@ export const vectorStoreProviderMap: Record = { VertexRagEngine: "vertex_ai", OpenAI: "openai", Azure: "azure", + Milvus: "milvus", }; const asset_logos_folder = "../ui/assets/logos/"; @@ -22,6 +24,7 @@ export const vectorStoreProviderLogoMap: Record = { [VectorStoreProviders.VertexRagEngine]: `${asset_logos_folder}google.svg`, [VectorStoreProviders.OpenAI]: `${asset_logos_folder}openai_small.svg`, [VectorStoreProviders.Azure]: `${asset_logos_folder}microsoft_azure.svg`, + [VectorStoreProviders.Milvus]: `${asset_logos_folder}milvus.svg`, }; // Define field types for provider-specific configurations @@ -84,6 +87,25 @@ export const vectorStoreProviderFields: Record type: "text", }, ], + milvus: [ + { + name: "api_key", + label: "API Key", + tooltip: + "To obtain a token, you should use a colon (:) to concatenate the username and password that you use to access your Milvus instance (e.g., username:password)", + placeholder: "username:password or api key", + required: true, + type: "password", + }, + { + name: "api_base", + label: "API Base", + tooltip: "Enter your Milvus endpoint (e.g., https://your-milvus-endpoint.com/)", + placeholder: "https://your-milvus-endpoint.com/", + required: true, + type: "text", + }, + ], }; export const getVectorStoreProviderLogoAndName = (providerValue: string): { logo: string; displayName: string } => {