use agent api key

Author: smoreinis

src/agentex/lib/adk/utils/_modules/client.py

@@ -5,8 +5,8 @@
 
 
 def create_async_agentex_client(**kwargs):
-    agent_id = refreshed_environment_variables.AGENT_ID
+    #agent_id = refreshed_environment_variables.AGENT_ID
     default_headers = {
-        "x-agent-identity": agent_id
+        "x-agent-identity": os.environ["AGENT_API_KEY"]
     }
     return AsyncAgentex(default_headers=default_headers, **kwargs)

src/agentex/lib/sdk/fastacp/base/base_acp_server.py

@@ -393,11 +393,15 @@ async def _register_agent(self, env_vars: EnvironmentVariables):
                     if response.status_code == 200:
                         agent = response.json()
                         agent_id, agent_name = agent["id"], agent["name"]
+                        print(agent)
+                        agent_api_key = agent["agent_api_key"]
 
                         os.environ["AGENT_ID"] = agent_id
                         os.environ["AGENT_NAME"] = agent_name
+                        os.environ["AGENT_API_KEY"] = agent_api_key
                         #refreshed_environment_variables.AGENT_ID = agent_id
                         #refreshed_environment_variables.AGENT_NAME = agent_name
+                        #refreshed_environment_variables.AGENT_API_KEY = agent_api_key
 
                         logger.info(
                             f"Successfully registered agent '{agent_name}' with Agentex server with acp_url: {full_acp_url}. Registration data: {registration_data}"