diff --git a/python/packages/autogen-core/src/autogen_core/components/models/_model_info.py b/python/packages/autogen-core/src/autogen_core/components/models/_model_info.py index f54df6fade..2440d5b186 100644 --- a/python/packages/autogen-core/src/autogen_core/components/models/_model_info.py +++ b/python/packages/autogen-core/src/autogen_core/components/models/_model_info.py @@ -112,7 +112,7 @@ def resolve_model(model: str) -> str: return model -def get_capabilties(model: str) -> ModelCapabilities: +def get_capabilities(model: str) -> ModelCapabilities: resolved_model = resolve_model(model) return _MODEL_CAPABILITIES[resolved_model] diff --git a/python/packages/autogen-core/src/autogen_core/components/models/_openai_client.py b/python/packages/autogen-core/src/autogen_core/components/models/_openai_client.py index 813bb59b52..8ce8ddff2c 100644 --- a/python/packages/autogen-core/src/autogen_core/components/models/_openai_client.py +++ b/python/packages/autogen-core/src/autogen_core/components/models/_openai_client.py @@ -333,7 +333,7 @@ class BaseOpenAIChatCompletionClient(ChatCompletionClient): if model_capabilities is None and isinstance(client, AsyncAzureOpenAI): raise ValueError("AzureOpenAIChatCompletionClient requires explicit model capabilities") elif model_capabilities is None: - self._model_capabilities = _model_info.get_capabilties(create_args["model"]) + self._model_capabilities = _model_info.get_capabilities(create_args["model"]) else: self._model_capabilities = model_capabilities diff --git a/python/packages/autogen-ext/src/autogen_ext/models/_openai/_model_info.py b/python/packages/autogen-ext/src/autogen_ext/models/_openai/_model_info.py index 79747ab679..aea2bfb5d1 100644 --- a/python/packages/autogen-ext/src/autogen_ext/models/_openai/_model_info.py +++ b/python/packages/autogen-ext/src/autogen_ext/models/_openai/_model_info.py @@ -112,7 +112,7 @@ def resolve_model(model: str) -> str: return model -def get_capabilties(model: str) -> ModelCapabilities: +def get_capabilities(model: str) -> ModelCapabilities: resolved_model = resolve_model(model) return _MODEL_CAPABILITIES[resolved_model] diff --git a/python/packages/autogen-ext/src/autogen_ext/models/_openai/_openai_client.py b/python/packages/autogen-ext/src/autogen_ext/models/_openai/_openai_client.py index ee2fc92054..aa9f772d0b 100644 --- a/python/packages/autogen-ext/src/autogen_ext/models/_openai/_openai_client.py +++ b/python/packages/autogen-ext/src/autogen_ext/models/_openai/_openai_client.py @@ -334,7 +334,7 @@ class BaseOpenAIChatCompletionClient(ChatCompletionClient): if model_capabilities is None and isinstance(client, AsyncAzureOpenAI): raise ValueError("AzureOpenAIChatCompletionClient requires explicit model capabilities") elif model_capabilities is None: - self._model_capabilities = _model_info.get_capabilties(create_args["model"]) + self._model_capabilities = _model_info.get_capabilities(create_args["model"]) else: self._model_capabilities = model_capabilities