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 f54df6fad..2440d5b18 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 813bb59b5..8ce8ddff2 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 79747ab67..aea2bfb5d 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 ee2fc9205..aa9f772d0 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