autogen icon indicating copy to clipboard operation
autogen copied to clipboard

Adding Gemini models to AutoGen Studio

Open metalshanked opened this issue 5 months ago • 1 comments

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

metalshanked avatar Aug 14 '25 15:08 metalshanked

I've opened a PR (#6896) that extends AutoGen Studio's UI to allow configuring model_info (and related fields) through a new Model Information section in the Edit Component menu, so the the JSON Editor no longer the only available option.

federicovilla55 avatar Aug 14 '25 21:08 federicovilla55