{agentType === "a2a" ? (
+ ) : selectedAgentTypeInfo?.use_a2a_form_fields ? (
+ // A2A-compatible agents (like Pydantic AI) use full A2A form fields
+ // plus any additional credential fields
+ <>
+
+ {selectedAgentTypeInfo.credential_fields.length > 0 && (
+
+
+ {selectedAgentTypeInfo.agent_type_display_name} Settings
+
+ {selectedAgentTypeInfo.credential_fields.map((field) => (
+
+ {field.field_type === "password" ? (
+
+ ) : (
+
+ )}
+
+ ))}
+
+ )}
+
) : selectedAgentTypeInfo ? (
) : null}
diff --git a/ui/litellm-dashboard/src/components/networking.tsx b/ui/litellm-dashboard/src/components/networking.tsx
index 65c41c5aab4..ddabc6f5212 100644
--- a/ui/litellm-dashboard/src/components/networking.tsx
+++ b/ui/litellm-dashboard/src/components/networking.tsx
@@ -215,6 +215,7 @@ export interface AgentCreateInfo {
credential_fields: AgentCredentialFieldMetadata[];
litellm_params_template?: Record
| null;
model_template?: string | null;
+ use_a2a_form_fields?: boolean;
}
export interface PublicModelHubInfo {
diff --git a/ui/litellm-dashboard/src/components/templates/key_edit_view.test.tsx b/ui/litellm-dashboard/src/components/templates/key_edit_view.test.tsx
index b245119cfee..e23bc88dda8 100644
--- a/ui/litellm-dashboard/src/components/templates/key_edit_view.test.tsx
+++ b/ui/litellm-dashboard/src/components/templates/key_edit_view.test.tsx
@@ -20,8 +20,8 @@ Object.defineProperty(window, "matchMedia", {
describe("KeyEditView", () => {
const MOCK_KEY_DATA: KeyResponse = {
- token: "40b7608ea43423400d5b82bb5ee11042bfb2ed4655f05b5992b5abbc2f294931",
- token_id: "40b7608ea43423400d5b82bb5ee11042bfb2ed4655f05b5992b5abbc2f294931",
+ token: "test-token-123",
+ token_id: "test-token-123",
key_name: "sk-...TUuw",
key_alias: "asdasdas",
spend: 0,
diff --git a/ui/litellm-dashboard/src/components/templates/key_info_view.test.tsx b/ui/litellm-dashboard/src/components/templates/key_info_view.test.tsx
index bd6e560a1c9..8137a70124f 100644
--- a/ui/litellm-dashboard/src/components/templates/key_info_view.test.tsx
+++ b/ui/litellm-dashboard/src/components/templates/key_info_view.test.tsx
@@ -5,8 +5,8 @@ import KeyInfoView from "./key_info_view";
describe("KeyInfoView", () => {
const MOCK_KEY_DATA: KeyResponse = {
- token: "40b7608ea43423400d5b82bb5ee11042bfb2ed4655f05b5992b5abbc2f294931",
- token_id: "40b7608ea43423400d5b82bb5ee11042bfb2ed4655f05b5992b5abbc2f294931",
+ token: "test-token-123",
+ token_id: "test-token-123",
key_name: "sk-...TUuw",
key_alias: "asdasdas",
spend: 0,
diff --git a/ui/litellm-dashboard/src/components/vector_store_management/VectorStoreForm.test.tsx b/ui/litellm-dashboard/src/components/vector_store_management/VectorStoreForm.test.tsx
new file mode 100644
index 00000000000..97d18e33640
--- /dev/null
+++ b/ui/litellm-dashboard/src/components/vector_store_management/VectorStoreForm.test.tsx
@@ -0,0 +1,27 @@
+import { render, screen } from "@testing-library/react";
+import { describe, expect, it, vi } from "vitest";
+import { CredentialItem } from "../networking";
+import VectorStoreForm from "./VectorStoreForm";
+
+vi.mock("../networking");
+
+describe("VectorStoreForm", () => {
+ it("should render the form when visible", () => {
+ const mockOnCancel = vi.fn();
+ const mockOnSuccess = vi.fn();
+ const mockAccessToken = "test-token";
+ const mockCredentials: CredentialItem[] = [];
+
+ render(
+ ,
+ );
+
+ expect(screen.getByText("Add New Vector Store")).toBeInTheDocument();
+ });
+});
diff --git a/ui/litellm-dashboard/src/components/vector_store_management/VectorStoreForm.tsx b/ui/litellm-dashboard/src/components/vector_store_management/VectorStoreForm.tsx
index d1cd3c5e443..8be879b2239 100644
--- a/ui/litellm-dashboard/src/components/vector_store_management/VectorStoreForm.tsx
+++ b/ui/litellm-dashboard/src/components/vector_store_management/VectorStoreForm.tsx
@@ -1,4 +1,4 @@
-import React, { useState } from "react";
+import React, { useState, useEffect } from "react";
import { TextInput, Button as TremorButton } from "@tremor/react";
import { Modal, Form, Select, Tooltip, Input, Alert } from "antd";
import { InfoCircleOutlined } from "@ant-design/icons";
@@ -10,6 +10,7 @@ import {
getProviderSpecificFields,
VectorStoreFieldConfig,
} from "../vector_store_providers";
+import { fetchAvailableModels, ModelGroup } from "../playground/llm_calls/fetch_models";
import NotificationsManager from "../molecules/notifications_manager";
interface VectorStoreFormProps {
@@ -30,6 +31,24 @@ const VectorStoreForm: React.FC = ({
const [form] = Form.useForm();
const [metadataJson, setMetadataJson] = useState("{}");
const [selectedProvider, setSelectedProvider] = useState("bedrock");
+ const [modelInfo, setModelInfo] = useState([]);
+
+ useEffect(() => {
+ if (!accessToken) return;
+
+ const loadModels = async () => {
+ try {
+ const uniqueModels = await fetchAvailableModels(accessToken);
+ if (uniqueModels.length > 0) {
+ setModelInfo(uniqueModels);
+ }
+ } catch (error) {
+ console.error("Error fetching model info:", error);
+ }
+ };
+
+ loadModels();
+ }, [accessToken]);
const handleCreate = async (formValues: any) => {
if (!accessToken) return;
@@ -207,23 +226,62 @@ const VectorStoreForm: React.FC = ({
{/* Provider-specific fields */}
- {getProviderSpecificFields(selectedProvider).map((field: VectorStoreFieldConfig) => (
-
- {field.label}{" "}
-
-
-
-
- }
- name={field.name}
- rules={field.required ? [{ required: true, message: `Please input the ${field.label.toLowerCase()}` }] : []}
- >
-
-
- ))}
+ {getProviderSpecificFields(selectedProvider).map((field: VectorStoreFieldConfig) => {
+ if (field.type === "select") {
+ const embeddingModels = modelInfo
+ .filter((option: ModelGroup) => option.mode === "embedding")
+ .map((option: ModelGroup) => ({
+ value: option.model_group,
+ label: option.model_group,
+ }));
+
+ return (
+
+ {field.label}{" "}
+
+
+
+
+ }
+ name={field.name}
+ rules={
+ field.required ? [{ required: true, message: `Please select the ${field.label.toLowerCase()}` }] : []
+ }
+ >
+
+ );
+ }
+
+ return (
+
+ {field.label}{" "}
+
+
+
+
+ }
+ name={field.name}
+ rules={
+ field.required ? [{ required: true, message: `Please input the ${field.label.toLowerCase()}` }] : []
+ }
+ >
+
+
+ );
+ })}
= {
@@ -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
@@ -31,7 +34,7 @@ export interface VectorStoreFieldConfig {
tooltip: string;
placeholder?: string;
required: boolean;
- type?: "text" | "password";
+ type?: "text" | "password" | "select";
}
// Provider-specific field configurations
@@ -84,6 +87,33 @@ 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",
+ },
+ {
+ name: "embedding_model",
+ label: "Embedding Model",
+ tooltip: "Select the embedding model to use",
+ placeholder: "text-embedding-3-small",
+ required: true,
+ type: "select",
+ },
+ ],
};
export const getVectorStoreProviderLogoAndName = (providerValue: string): { logo: string; displayName: string } => {
diff --git a/ui/litellm-dashboard/src/components/view_logs/index.test.tsx b/ui/litellm-dashboard/src/components/view_logs/index.test.tsx
index 294dc84761b..a44a8ada43d 100644
--- a/ui/litellm-dashboard/src/components/view_logs/index.test.tsx
+++ b/ui/litellm-dashboard/src/components/view_logs/index.test.tsx
@@ -1,4 +1,4 @@
-import { render } from "@testing-library/react";
+import { render, screen } from "@testing-library/react";
import { describe, expect, it } from "vitest";
import { RequestViewer } from "./index";
import type { LogEntry } from "./columns";
@@ -43,14 +43,14 @@ const createRow = (overrides: Partial = {}): Row =>
describe("Request Viewer", () => {
it("renders the request details heading", () => {
- const { getByText } = render();
- expect(getByText("Request Details")).toBeInTheDocument();
+ render();
+ expect(screen.getByText("Request Details")).toBeInTheDocument();
});
it("should truncate the request id if it is longer than 64 characters", () => {
const LONG_REQUEST_ID = "a".repeat(128);
const TRUNCATED_REQUEST_ID = `${"a".repeat(64)}...`;
- const { getByText } = render(
+ render(
{
/>,
);
- expect(getByText(TRUNCATED_REQUEST_ID)).toBeInTheDocument();
+ expect(screen.getByText(TRUNCATED_REQUEST_ID)).toBeInTheDocument();
+ });
+
+ it("should display LiteLLM Overhead when litellm_overhead_time_ms is present in metadata", () => {
+ render(
+ ,
+ );
+
+ expect(screen.getByText("LiteLLM Overhead:")).toBeInTheDocument();
+ expect(screen.getByText("150 ms")).toBeInTheDocument();
+ });
+
+ it("should not display LiteLLM Overhead when litellm_overhead_time_ms is not present in metadata", () => {
+ render();
+
+ expect(screen.queryByText("LiteLLM Overhead:")).not.toBeInTheDocument();
});
});
diff --git a/ui/litellm-dashboard/src/components/view_logs/index.tsx b/ui/litellm-dashboard/src/components/view_logs/index.tsx
index 04c81d832d6..f6a969a9eed 100644
--- a/ui/litellm-dashboard/src/components/view_logs/index.tsx
+++ b/ui/litellm-dashboard/src/components/view_logs/index.tsx
@@ -930,6 +930,12 @@ export function RequestViewer({ row }: { row: Row }) {
Duration:
{row.original.duration} s.