diff --git a/python/packages/autogen-core/src/autogen_core/_component_config.py b/python/packages/autogen-core/src/autogen_core/_component_config.py index bb603a839ed6..b3c7f8fd72ae 100644 --- a/python/packages/autogen-core/src/autogen_core/_component_config.py +++ b/python/packages/autogen-core/src/autogen_core/_component_config.py @@ -49,6 +49,8 @@ def _type_to_provider_str(t: type) -> str: "AzureOpenAIChatCompletionClient": "autogen_ext.models.openai.AzureOpenAIChatCompletionClient", "openai_chat_completion_client": "autogen_ext.models.openai.OpenAIChatCompletionClient", "OpenAIChatCompletionClient": "autogen_ext.models.openai.OpenAIChatCompletionClient", + "OllamaChatCompletionClient": "autogen_ext.models.ollama.OllamaChatCompletionClient", + "ollama_chat_completion_client": "autogen_ext.models.ollama.OllamaChatCompletionClient", }