From 3e19da1258a5028c34e1a24835640c609bd80a42 Mon Sep 17 00:00:00 2001 From: Reinier van der Leer Date: Tue, 12 Dec 2023 10:52:18 +0100 Subject: [PATCH] refactor: Remove X- prefix from API request headers - Remove X- prefix from X-AutoGPT-UserID, X-AP-TaskID, and X-AP-StepID headers - Refactor literal references to "ask_user" to `ask_user.__name__` in AgentProtocolServer --- autogpts/autogpt/autogpt/app/agent_protocol_server.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/autogpts/autogpt/autogpt/app/agent_protocol_server.py b/autogpts/autogpt/autogpt/app/agent_protocol_server.py index e80ecdf1..1ee6f1ef 100644 --- a/autogpts/autogpt/autogpt/app/agent_protocol_server.py +++ b/autogpts/autogpt/autogpt/app/agent_protocol_server.py @@ -273,13 +273,13 @@ class AgentProtocolServer: + ("\n\n" if "\n" in str(execute_result) else " ") + f"{execute_result}\n\n" ) - if execute_command_args and execute_command != "ask_user" + if execute_command_args and execute_command != ask_user.__name__ else "" ) output += f"{raw_output['thoughts']['speak']}\n\n" output += ( f"Next Command: {next_command}({fmt_kwargs(next_command_args)})" - if next_command != "ask_user" + if next_command != ask_user.__name__ else next_command_args["question"] ) @@ -417,11 +417,11 @@ class AgentProtocolServer: task_llm_provider = copy.deepcopy(self.llm_provider) _extra_request_headers = task_llm_provider._configuration.extra_request_headers - _extra_request_headers["X-AP-TaskID"] = task.task_id + _extra_request_headers["AP-TaskID"] = task.task_id if step_id: - _extra_request_headers["X-AP-StepID"] = step_id + _extra_request_headers["AP-StepID"] = step_id if task.additional_input and (user_id := task.additional_input.get("user_id")): - _extra_request_headers["X-AutoGPT-UserID"] = user_id + _extra_request_headers["AutoGPT-UserID"] = user_id return task_llm_provider