aboutsummaryrefslogtreecommitdiff
path: root/autogpts/autogpt/autogpt/core/prompting/base.py
diff options
context:
space:
mode:
Diffstat (limited to 'autogpts/autogpt/autogpt/core/prompting/base.py')
-rw-r--r--autogpts/autogpt/autogpt/core/prompting/base.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/autogpts/autogpt/autogpt/core/prompting/base.py b/autogpts/autogpt/autogpt/core/prompting/base.py
index c983be987..19e315f69 100644
--- a/autogpts/autogpt/autogpt/core/prompting/base.py
+++ b/autogpts/autogpt/autogpt/core/prompting/base.py
@@ -1,7 +1,7 @@
import abc
from autogpt.core.configuration import SystemConfiguration
-from autogpt.core.resource.model_providers import AssistantChatMessageDict
+from autogpt.core.resource.model_providers import AssistantChatMessage
from .schema import ChatPrompt, LanguageModelClassification
@@ -19,5 +19,5 @@ class PromptStrategy(abc.ABC):
...
@abc.abstractmethod
- def parse_response_content(self, response_content: AssistantChatMessageDict):
+ def parse_response_content(self, response_content: AssistantChatMessage):
...