From 3c63f6f3ef817ebc8f49cff7aeb722ec498b85de Mon Sep 17 00:00:00 2001 From: Rohan Thacker Date: Thu, 31 Oct 2024 02:09:45 +0530 Subject: [PATCH] Corrected typo in get_capabilities in _model_info.py (#4002) --- .../src/autogen_core/components/models/_model_info.py | 2 +- .../src/autogen_core/components/models/_openai_client.py | 2 +- .../autogen-ext/src/autogen_ext/models/_openai/_model_info.py | 2 +- .../src/autogen_ext/models/_openai/_openai_client.py | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) 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