diff options
Diffstat (limited to 'autogpts/autogpt/autogpt/core/resource/schema.py')
| -rw-r--r-- | autogpts/autogpt/autogpt/core/resource/schema.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/autogpts/autogpt/autogpt/core/resource/schema.py b/autogpts/autogpt/autogpt/core/resource/schema.py index d8cc1de31..0da275ee2 100644 --- a/autogpts/autogpt/autogpt/core/resource/schema.py +++ b/autogpts/autogpt/autogpt/core/resource/schema.py @@ -1,5 +1,6 @@ import abc import enum +import math from pydantic import BaseModel, SecretBytes, SecretField, SecretStr @@ -25,9 +26,9 @@ class ProviderUsage(SystemConfiguration, abc.ABC): class ProviderBudget(SystemConfiguration): - total_budget: float = UserConfigurable() - total_cost: float - remaining_budget: float + total_budget: float = UserConfigurable(math.inf) + total_cost: float = 0 + remaining_budget: float = math.inf usage: ProviderUsage @abc.abstractmethod |