connector-py 4.183.0__tar.gz → 4.185.0__tar.gz

{connector_py-4.183.0 → connector_py-4.185.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: connector-py
-Version: 4.183.0
+Version: 4.185.0
 Summary: An Abstract Tool to Perform Actions on Integrations.
 Author-email: teamlumos <security@lumos.com>
 License-Expression: Apache-2.0

connector_py-4.185.0/connector/__about__.py

@@ -0,0 +1 @@
+__version__ = "4.185.0"

{connector_py-4.183.0 → connector_py-4.185.0}/connector/oai/base_clients.py

@@ -10,6 +10,7 @@ from connector_sdk_types.generated import ErrorCode
 from gql import Client
 from gql.client import AsyncClientSession
 from gql.dsl import DSLSchema
+from gql.transport.httpx import HTTPXAsyncTransport as GqlHTTPXAsyncTransport
 from graphql import DocumentNode, GraphQLSchema, build_client_schema, build_schema
 from httpx import Response
 from typing_extensions import Self
@@ -181,7 +182,7 @@ class RateLimitedClient(AsyncClient):
 class RateLimitedHTTPXAsyncTransport(HTTPXAsyncTransport):
     """A wrapper around HTTPXAsyncTransport that applies rate limiting to GraphQL requests."""
 
-    def __init__(self, base_transport: HTTPXAsyncTransport, rate_limit_config: RateLimitConfig):
+    def __init__(self, base_transport: GqlHTTPXAsyncTransport, rate_limit_config: RateLimitConfig):
         # Copy all attributes from base transport, but exclude 'execute' to avoid shadowing our method
         base_dict = {k: v for k, v in base_transport.__dict__.items() if k != "execute"}
         self.__dict__.update(base_dict)
@@ -348,7 +349,8 @@ class BaseGraphQLSession(AsyncClientSession):
         rate_limiting = rate_limit_config or cls._rate_limit_config
         if rate_limiting is not None and "transport" in client_args:
             transport = client_args["transport"]
-            if isinstance(transport, HTTPXAsyncTransport):
+            # wrap any gql HTTPX transport (tests use the GqlHTTPXAsyncTransport, client uses the SDK's HTTPXAsyncTransport)
+            if isinstance(transport, GqlHTTPXAsyncTransport):
                 client_args["transport"] = RateLimitedHTTPXAsyncTransport(transport, rate_limiting)
 
         return Client(**client_args)

{connector_py-4.183.0 → connector_py-4.185.0}/connector/oai/errors.py

@@ -144,7 +144,7 @@ class DefaultHandler(ExceptionHandler):
             error_message in str(response.error.message).lower() for error_message in error_messages
         ):
             response.error.message = "Failed to connect to the API. Please verify the URL or try at a later time due to a potential temporary network issue."
-            response.error.error_code = ErrorCode.API_ERROR
+            response.error.error_code = ErrorCode.CONNECTION_TIMEOUT
 
         if isinstance(e, json.JSONDecodeError):
             response.error.error_code = ErrorCode.API_ERROR

{connector_py-4.183.0 → connector_py-4.185.0}/connector_py.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: connector-py
-Version: 4.183.0
+Version: 4.185.0
 Summary: An Abstract Tool to Perform Actions on Integrations.
 Author-email: teamlumos <security@lumos.com>
 License-Expression: Apache-2.0

{connector_py-4.183.0 → connector_py-4.185.0}/tests/oai/test_exception_handler_handled_cases.py

@@ -474,7 +474,7 @@ def case_connect_error() -> Case:
         is_error=True,
         error=Error(
             message="Failed to connect to the API. Please verify the URL or try at a later time due to a potential temporary network issue.",
-            error_code=ErrorCode.API_ERROR,
+            error_code=ErrorCode.CONNECTION_TIMEOUT,
             app_id=app_id,
             status_code=None,
             raised_by="ConnectError",
@@ -570,7 +570,7 @@ def case_connection_error_message() -> Case:
         is_error=True,
         error=Error(
             message="Failed to connect to the API. Please verify the URL or try at a later time due to a potential temporary network issue.",
-            error_code=ErrorCode.API_ERROR,
+            error_code=ErrorCode.CONNECTION_TIMEOUT,
             app_id=app_id,
             status_code=None,
             raised_by="Exception",

connector_py-4.183.0/connector/__about__.py

@@ -1 +0,0 @@
-__version__ = "4.183.0"