pygeai 0.2.7b45__py3-none-any.whl → 0.2.7b46__py3-none-any.whl

pygeai/proxy/servers.py

@@ -199,7 +199,7 @@ class MCPServer(ToolServer):
     async def initialize(self) -> None:
         self.public_prefix = self.config.get("public_prefix")
         transport = self.config.get("transport") or (
-            "sse" if "uri" in self.config else "stdio"
+            "sse" if ("uri" in self.config  or "url" in self.config) else "stdio"
         )
         try:
             if transport == "stdio":
@@ -226,17 +226,19 @@ class MCPServer(ToolServer):
                 )
                 read, write = stdio_transport
             elif transport == "sse":
-                uri = self.config.get("uri")
+                uri = self.config.get("uri", self.config["url"])
                 if not uri:
                     raise ValueError("Missing 'uri' for sse transport")
-
-                sse_transport = await self.exit_stack.enter_async_context(
-                    sse_client(
-                        url=self.config["uri"],
-                        headers=self.config.get("headers")
+                try:
+                    sse_transport = await self.exit_stack.enter_async_context(
+                        sse_client(
+                            url=uri,
+                            headers=self.config.get("headers")
+                        )
                     )
-                )
-                read, write = sse_transport
+                    read, write = sse_transport
+                except httpx.HTTPStatusError as e:
+                    Console.write_exception(f"HTTP error initializing MCP server {self.name}:", e)
             else:
                 raise ValueError(f"Unsupported transport: {transport}")
             

{pygeai-0.2.7b45.dist-info → pygeai-0.2.7b46.dist-info}/METADATA

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

{pygeai-0.2.7b45.dist-info → pygeai-0.2.7b46.dist-info}/RECORD

@@ -187,7 +187,7 @@ 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=hXxVeWvkwfb7QrwOGcKyL9qW_wdNTxcpW84ppqwQ0SU,11132
-pygeai/proxy/servers.py,sha256=ZJEnfT5gogTrr9LYir0kRkUDzW6RJjNFWceqN1sIeKM,11231
+pygeai/proxy/servers.py,sha256=iQb6dpvcCsd5behZdxibl6ylqQOvyWvpYfMVKNRfL_U,11461
 pygeai/proxy/tool.py,sha256=vG1THJNHCPNr5Y_3J7NlTu20hlATzsejOe07soPMDAE,2238
 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.7b45.dist-info/licenses/LICENSE,sha256=eHfqo7-AWS8cMq0cg03lq7owsLeCmZA-xS5L0kuHnl8,1474
-pygeai-0.2.7b45.dist-info/METADATA,sha256=PBUcRGghW1sTQMmZ0BPxwaRLn54kLfXKnaKCz-Jt0Q0,6883
-pygeai-0.2.7b45.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pygeai-0.2.7b45.dist-info/entry_points.txt,sha256=OAmwuXVCQBTCE3HeVegVd37hbhCcp9TPahvdrCuMYWw,178
-pygeai-0.2.7b45.dist-info/top_level.txt,sha256=bJFwp2tURmCfB94yXDF7ylvdSJXFDDJsyUOb-7PJgwc,7
-pygeai-0.2.7b45.dist-info/RECORD,,
+pygeai-0.2.7b46.dist-info/licenses/LICENSE,sha256=eHfqo7-AWS8cMq0cg03lq7owsLeCmZA-xS5L0kuHnl8,1474
+pygeai-0.2.7b46.dist-info/METADATA,sha256=laJ_y-4UmPVh0shlWgMyU2JUZu_ert5cRq6QUK4a4bo,6883
+pygeai-0.2.7b46.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pygeai-0.2.7b46.dist-info/entry_points.txt,sha256=OAmwuXVCQBTCE3HeVegVd37hbhCcp9TPahvdrCuMYWw,178
+pygeai-0.2.7b46.dist-info/top_level.txt,sha256=bJFwp2tURmCfB94yXDF7ylvdSJXFDDJsyUOb-7PJgwc,7
+pygeai-0.2.7b46.dist-info/RECORD,,