add app-id variable to headers

PaddlePaddle · Apr 12, 2024 · c165511 · c165511
1 parent 6910387
commit c165511
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/erniebot-agent/src/erniebot_agent/tools/remote_tool.py b/erniebot-agent/src/erniebot_agent/tools/remote_tool.py
@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import os
 import dataclasses
 import logging
 from copy import deepcopy
@@ -118,6 +119,9 @@ async def send_request(self, tool_arguments: Dict[str, Any]) -> dict:
         headers = deepcopy(self.headers)
         headers["Content-Type"] = self.tool_view.parameters_content_type
 
+        if "EB_SDK_TRACE_APP_ID" in os.environ:
+            headers["X-EB-SDK-TRACE-APP-ID"] = os.getenv("EB_SDK_TRACE_APP_ID")
+
         requests_inputs = {
             "headers": headers,
         }