File tree Expand file tree Collapse file tree 1 file changed +7
-1
lines changed
Expand file tree Collapse file tree 1 file changed +7
-1
lines changed Original file line number Diff line number Diff line change 11from __future__ import annotations
22
3+ import os
34import sys
45import json
56import time
@@ -678,7 +679,12 @@ def user_agent(self) -> str:
678679 # Format: "Gradient/package/version"
679680 package = self ._user_agent_package or "Python"
680681 version = self ._user_agent_version if self ._user_agent_package and self ._user_agent_version else self ._version
681- return f"{ self .__class__ .__name__ } /{ package } /{ version } "
682+ base_agent = f"{ self .__class__ .__name__ } /{ package } /{ version } "
683+
684+ deployment_uuid = os .environ .get ("AGENT_WORKSPACE_DEPLOYMENT_UUID" )
685+ if deployment_uuid :
686+ return f"{ base_agent } /GRADIENT_ADK_AGENT/{ deployment_uuid } "
687+ return base_agent
682688
683689 @property
684690 def base_url (self ) -> URL :
You can’t perform that action at this time.
0 commit comments