Skip to content

Adding Gemini models to AutoGen Studio #6933

@metalshanked

Description

@metalshanked

What happened?

Describe the bug
The AutoGenStudio UI does not recognize non OpenAI models.
Error: model_info is needed for non openai models

Here is a workaround

Edit in JSON Editor since the UI form does not have all the needed fields

{
  "provider": "autogen_ext.models.openai.OpenAIChatCompletionClient",
  "component_type": "model",
  "version": 1,
  "component_version": 1,
  "description": "OpenAI API compatible Gemini 2.5 Pro",
  "label": "Gemini 2.5 Pro",
  "config": {
    "model": "gemini-2.5-pro",
    "api_key": "12345ABCDEF",
    "base_url": "https://generativelanguage.googleapis.com/v1beta/openai/",
    "component_type": "model",
    "model_capabilities": {
      "vision": false,
      "function_calling": true,
      "json_output": false
    }
  }
}
Image

Which packages was the bug in?

AutoGen Studio (autogensudio)

AutoGen library version.

Python dev (main branch)

Other library version.

No response

Model used

No response

Model provider

None

Other model provider

No response

Python version

None

.NET version

None

Operating system

None

Metadata

Metadata

Assignees

No one assigned

    Type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions