pygeai 0.2.7b40__py3-none-any.whl → 0.2.7b42__py3-none-any.whl

pygeai/proxy/managers.py

@@ -52,13 +52,18 @@ class ServerManager:
                 await self.exit_stack.enter_async_context(server.exit_stack)
                 await server.initialize()
                 self.servers[server.name] = server
-                Console.write_stdout(f"Server {server.name} initialized successfully")       
+                Console.write_stdout(f"\nServer {server.name} initialized successfully", end="\n\n")        
             except Exception as e:
                 Console.write_exception(f"Failed to initialize server {server.name}:", e)
                 raise
 
         for server in self.servers.values():
-            Console.write_stdout(f"Listing tools for server {server.name}")
+            Console.write_stdout(f"Listing tools for server {server.name}", end="")
+            if server.public_preffix:
+                Console.write_stdout(f" | access scope:public prefix: {server.public_preffix}")
+            else:
+                Console.write_stdout(" ! access scope:private")
+
             tools = await server.list_tools()
             for tool in tools:
                 self.tools[tool.get_full_name()] = tool
@@ -75,17 +80,29 @@ class ServerManager:
         try:
             alias = self.settings.get_current_alias()
             client = ProxyClient(self.settings.get_api_key(alias), self.settings.get_base_url(alias), self.settings.get_proxy_id(alias))
-            Console.write_stdout(f"Registering proxy {self.settings.get_proxy_id(alias)} with name {self.settings.get_proxy_name(alias)} and description {self.settings.get_proxy_description(alias)}")
-            client.register(proxy_data=ToolProxyData(
+            Console.write_stdout(f"\nRegistering proxy {self.settings.get_proxy_id(alias)} with name {self.settings.get_proxy_name(alias)} and description {self.settings.get_proxy_description(alias)}")
+            result = client.register(proxy_data=ToolProxyData(
                 id=self.settings.get_proxy_id(alias),
                 name=self.settings.get_proxy_name(alias),
                 description=self.settings.get_proxy_description(alias),
                 affinity=self.settings.get_proxy_affinity(alias),
                 tools=list(self.tools.values())
             ))
-            Console.write_stdout("Proxy registered successfully")
+            Console.write_stdout("----------------------------------")
+            Console.write_stdout(f"Proxy registered successfully:")
+            if isinstance(result, dict) and isinstance(result.get("Messages"), list):
+                for message in result["Messages"]:
+                    description = message.get("Description", "")
+                    message_type = message.get("Type", None)
+
+                    if message_type == 1:
+                        Console.write_stderr(description)
+                    elif message_type == 2:
+                        Console.write_stdout(description)
+                Console.write_stdout("----------------------------------")
             return client
         except (ConnectionError, MaxRetryError):
+            Console.write_exception(f"Error registering proxy {self.settings.get_proxy_id(alias)}:")
             raise
     
     async def execute_tool(

pygeai/proxy/servers.py

@@ -105,6 +105,7 @@ class A2AServer(ToolServer):
         """
         try:
             self.httpx_client = httpx.AsyncClient(timeout=60.0)
+            self.public_preffix = self.config.get("publicpreffix")
             headers = self.config.get("headers")
             if headers:
                 self.httpx_client.headers.update(headers)

{pygeai-0.2.7b40.dist-info → pygeai-0.2.7b42.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pygeai
-Version: 0.2.7b40
+Version: 0.2.7b42
 Summary: Software Development Kit to interact with Globant Enterprise AI.
 Author-email: Globant <geai-sdk@globant.com>
 License-Expression: MIT

{pygeai-0.2.7b40.dist-info → pygeai-0.2.7b42.dist-info}/RECORD

@@ -186,8 +186,8 @@ pygeai/organization/limits/mappers.py,sha256=nINHaXOnZLnpc39PG3xf_7HX6tS9_-cT-H5
 pygeai/proxy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pygeai/proxy/clients.py,sha256=YPRUS3JAbEXgIGZ_dg8VwEXLxvG_hl51o70uudqjYB0,7857
 pygeai/proxy/config.py,sha256=41fACnnhB21OhnohVPW4sbM8aZbAAEgGo7-dUaM_fpM,5057
-pygeai/proxy/managers.py,sha256=i2cfddEUxhtre2hRN6OgMpHqvs8TPIH20wgcLANMstM,10168
-pygeai/proxy/servers.py,sha256=c1Z94aXXSlWBcbffc6O1xKwI0aKAacGR3-SI9aeQUFk,11169
+pygeai/proxy/managers.py,sha256=swORE-WLtcTPglNpRc2AuHVjnbq0SX2ppnWQWYA44iQ,11134
+pygeai/proxy/servers.py,sha256=ys2bzIRIoBxEuzQOxviFaQGznm3CuymXHSkeKu4xm6U,11236
 pygeai/proxy/tool.py,sha256=BB6G5cg0768qz1tARB1pL0wBhSKr8SeWVGna0hfGmzg,2245
 pygeai/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pygeai/tests/admin/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -475,9 +475,9 @@ pygeai/vendor/a2a/utils/helpers.py,sha256=6Tbd8SVfXvdNEk6WYmLOjrAxkzFf1aIg8dkFfB
 pygeai/vendor/a2a/utils/message.py,sha256=gc_EKO69CJ4HkR76IFgsy-kENJz1dn7CfSgWJWvt-gs,2197
 pygeai/vendor/a2a/utils/task.py,sha256=BYRA_L1HpoUGJAVlyHML0lCM9Awhf2Ovjj7oPFXKbh0,1647
 pygeai/vendor/a2a/utils/telemetry.py,sha256=VvSp1Ztqaobkmq9-3sNhhPEilJS32-JTSfKzegkj6FU,10861
-pygeai-0.2.7b40.dist-info/licenses/LICENSE,sha256=eHfqo7-AWS8cMq0cg03lq7owsLeCmZA-xS5L0kuHnl8,1474
-pygeai-0.2.7b40.dist-info/METADATA,sha256=UhHBh-lpk6ZKG13kD2atfFs23HasjoRhq9WOohDF8fU,6883
-pygeai-0.2.7b40.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pygeai-0.2.7b40.dist-info/entry_points.txt,sha256=OAmwuXVCQBTCE3HeVegVd37hbhCcp9TPahvdrCuMYWw,178
-pygeai-0.2.7b40.dist-info/top_level.txt,sha256=bJFwp2tURmCfB94yXDF7ylvdSJXFDDJsyUOb-7PJgwc,7
-pygeai-0.2.7b40.dist-info/RECORD,,
+pygeai-0.2.7b42.dist-info/licenses/LICENSE,sha256=eHfqo7-AWS8cMq0cg03lq7owsLeCmZA-xS5L0kuHnl8,1474
+pygeai-0.2.7b42.dist-info/METADATA,sha256=8ib_II6z_LMPUaR07YA4dt6gzeGwtxtfJFxYnHq2kPo,6883
+pygeai-0.2.7b42.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pygeai-0.2.7b42.dist-info/entry_points.txt,sha256=OAmwuXVCQBTCE3HeVegVd37hbhCcp9TPahvdrCuMYWw,178
+pygeai-0.2.7b42.dist-info/top_level.txt,sha256=bJFwp2tURmCfB94yXDF7ylvdSJXFDDJsyUOb-7PJgwc,7
+pygeai-0.2.7b42.dist-info/RECORD,,