Adding Milvus to Vector Store in UI

This commit is contained in:
yuneng-jiang
2025-12-15 18:31:35 -08:00
parent c754794bc3
commit fabbd28b45
@@ -4,6 +4,7 @@ export enum VectorStoreProviders {
VertexRagEngine = "Vertex AI RAG Engine",
OpenAI = "OpenAI",
Azure = "Azure OpenAI",
Milvus = "Milvus",
}
export const vectorStoreProviderMap: Record<string, string> = {
@@ -12,6 +13,7 @@ export const vectorStoreProviderMap: Record<string, string> = {
VertexRagEngine: "vertex_ai",
OpenAI: "openai",
Azure: "azure",
Milvus: "milvus",
};
const asset_logos_folder = "../ui/assets/logos/";
@@ -22,6 +24,7 @@ export const vectorStoreProviderLogoMap: Record<string, string> = {
[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<string, VectorStoreFieldConfig[]>
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 } => {