diff options
Diffstat (limited to 'autogpts/autogpt/autogpt/core/resource/model_providers/schema.py')
-rw-r--r-- | autogpts/autogpt/autogpt/core/resource/model_providers/schema.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/autogpts/autogpt/autogpt/core/resource/model_providers/schema.py b/autogpts/autogpt/autogpt/core/resource/model_providers/schema.py index 43d4bd296..cc0030995 100644 --- a/autogpts/autogpt/autogpt/core/resource/model_providers/schema.py +++ b/autogpts/autogpt/autogpt/core/resource/model_providers/schema.py @@ -2,6 +2,7 @@ import abc import enum import math from typing import ( + Any, Callable, ClassVar, Generic, @@ -68,12 +69,12 @@ class ChatMessageDict(TypedDict): class AssistantFunctionCall(BaseModel): name: str - arguments: str + arguments: dict[str, Any] class AssistantFunctionCallDict(TypedDict): name: str - arguments: str + arguments: dict[str, Any] class AssistantToolCall(BaseModel): |