aboutsummaryrefslogtreecommitdiff
path: root/autogpts/autogpt/autogpt/core/ability/builtins/query_language_model.py
diff options
context:
space:
mode:
Diffstat (limited to 'autogpts/autogpt/autogpt/core/ability/builtins/query_language_model.py')
-rw-r--r--autogpts/autogpt/autogpt/core/ability/builtins/query_language_model.py7
1 files changed, 5 insertions, 2 deletions
diff --git a/autogpts/autogpt/autogpt/core/ability/builtins/query_language_model.py b/autogpts/autogpt/autogpt/core/ability/builtins/query_language_model.py
index 67d51087f..7a6ae68ee 100644
--- a/autogpts/autogpt/autogpt/core/ability/builtins/query_language_model.py
+++ b/autogpts/autogpt/autogpt/core/ability/builtins/query_language_model.py
@@ -45,7 +45,10 @@ class QueryLanguageModel(Ability):
parameters: ClassVar[dict[str, JSONSchema]] = {
"query": JSONSchema(
type=JSONSchema.Type.STRING,
- description="A query for a language model. A query should contain a question and any relevant context.",
+ description=(
+ "A query for a language model. "
+ "A query should contain a question and any relevant context."
+ ),
)
}
@@ -59,5 +62,5 @@ class QueryLanguageModel(Ability):
ability_name=self.name(),
ability_args={"query": query},
success=True,
- message=model_response.response["content"],
+ message=model_response.response.content or "",
)