pygeai 0.2.7b40__tar.gz → 0.2.7b42__tar.gz

{pygeai-0.2.7b40/pygeai.egg-info → pygeai-0.2.7b42}/PKG-INFO

@@ -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 → pygeai-0.2.7b42}/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-0.2.7b40 → pygeai-0.2.7b42}/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 → pygeai-0.2.7b42/pygeai.egg-info}/PKG-INFO

@@ -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 → pygeai-0.2.7b42}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "pygeai"
-version = "0.2.7b40"
+version = "0.2.7b42"
 description = "Software Development Kit to interact with Globant Enterprise AI."
 license = "MIT"
 requires-python = ">=3.10"