diff --git a/autogpt/core/ability/base.py b/autogpt/core/ability/base.py index ac26f0267151..1faaaf3327d4 100644 --- a/autogpt/core/ability/base.py +++ b/autogpt/core/ability/base.py @@ -1,6 +1,6 @@ import abc from pprint import pformat -from typing import ClassVar +from typing import Any, ClassVar import inflection from pydantic import Field @@ -50,11 +50,11 @@ def required_arguments(cls) -> list[str]: return [] @abc.abstractmethod - async def __call__(self, *args, **kwargs) -> AbilityResult: + async def __call__(self, *args: Any, **kwargs: Any) -> AbilityResult: ... def __str__(self) -> str: - return pformat(self.dump) + return pformat(self.dump()) def dump(self) -> dict: return { @@ -88,5 +88,5 @@ def get_ability(self, ability_name: str) -> Ability: ... @abc.abstractmethod - def perform(self, ability_name: str, **kwargs) -> AbilityResult: + async def perform(self, ability_name: str, **kwargs: Any) -> AbilityResult: ... diff --git a/autogpt/core/resource/model_providers/openai.py b/autogpt/core/resource/model_providers/openai.py index 3707796a134a..df7bdb83bafb 100644 --- a/autogpt/core/resource/model_providers/openai.py +++ b/autogpt/core/resource/model_providers/openai.py @@ -109,7 +109,7 @@ class OpenAIModelProviderBudget(ModelProviderBudget): class OpenAISettings(ModelProviderSettings): configuration: OpenAIConfiguration - credentials: ModelProviderCredentials() + credentials: ModelProviderCredentials budget: OpenAIModelProviderBudget