diff options
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.py | 7 |
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 "", ) |