diff --git a/python/packages/autogen-studio/frontend/src/components/types/component-templates.ts b/python/packages/autogen-studio/frontend/src/components/types/component-templates.ts index 4372ad20b..3ca2b5728 100644 --- a/python/packages/autogen-studio/frontend/src/components/types/component-templates.ts +++ b/python/packages/autogen-studio/frontend/src/components/types/component-templates.ts @@ -226,6 +226,27 @@ export const MODEL_TEMPLATES: ComponentTemplate[] = [ temperature: 0.7, } as AnthropicClientConfig, }, + { + id: "external-custom-model", + label: "External Model", + description: "Template for a model using an OpenAI-compatible endpoint", + provider: PROVIDERS.OPENAI, + component_type: "model", + version: 1, + component_version: 1, + config: { + model: "your-model-name", + base_url: "https://example.com/", + model_info: { + vision: false, + function_calling: true, + json_output: false, + structured_output: false, + family: "unknown", + multiple_system_messages: false, + }, + } as OpenAIClientConfig, + }, ]; // Tool Templates diff --git a/python/packages/autogen-studio/frontend/src/components/views/teambuilder/builder/component-editor/fields/model-fields.tsx b/python/packages/autogen-studio/frontend/src/components/views/teambuilder/builder/component-editor/fields/model-fields.tsx index 4126f7958..85326edf9 100644 --- a/python/packages/autogen-studio/frontend/src/components/views/teambuilder/builder/component-editor/fields/model-fields.tsx +++ b/python/packages/autogen-studio/frontend/src/components/views/teambuilder/builder/component-editor/fields/model-fields.tsx @@ -1,5 +1,5 @@ import React, { useCallback } from "react"; -import { Input, InputNumber, Select, Tooltip, Collapse } from "antd"; +import { Input, InputNumber, Select, Tooltip, Collapse, Switch, AutoComplete } from "antd"; import TextArea from "antd/es/input/TextArea"; import { HelpCircle, Settings, User, Wrench } from "lucide-react"; import { @@ -23,7 +23,7 @@ const InputWithTooltip: React.FC<{ tooltip: string; children: React.ReactNode; }> = ({ label, tooltip, children }) => ( -