github-agent 2.4.3__py3-none-any.whl → 2.4.4__py3-none-any.whl

github_agent/__init__.py

@@ -1 +1 @@
-__version__ = "2.4.3"
+__version__ = "2.4.4"

github_agent/github_gql.py

@@ -4,11 +4,10 @@
 import logging
 from typing import Any
 
-from agent_connector_sdk.exceptions import MissingParameterError, ParameterError
+from agent_connector_sdk.exceptions import MissingParameterError
 from agent_connector_sdk.tls.profile import ResolvedTLSProfile
 from agent_connector_sdk.tls.resolve import resolve_tls_profile
-from gql import Client, gql
-from gql.transport.requests import RequestsHTTPTransport
+from agent_connector_sdk.transports.graphql import GraphQLAuth, GraphQLClient
 
 
 class GraphQL:
@@ -44,29 +43,21 @@ class GraphQL:
         # GitHub GraphQL authenticates with a bearer token (classic PAT,
         # fine-grained PAT, or OAuth/GitHub-App installation token).
         self.headers = {"Authorization": f"Bearer {token}"}
-        self.transport = RequestsHTTPTransport(
-            url=self.url,
-            headers=self.headers,
-            **self.tls_profile.requests_kwargs(),
-        )
-        self.client = Client(
-            transport=self.transport, fetch_schema_from_transport=False
+        self._gql = GraphQLClient(
+            self.url,
+            auth=GraphQLAuth.bearer(token),
+            tls_profile=self.tls_profile,
+            tls_service="github",
         )
 
+    @property
+    def client(self) -> Any:
+        """The underlying protocol client, built on first use."""
+        return self._gql.client
+
     def close(self) -> None:
-        """Release GraphQL transport state and runtime-only TLS material.
-
-        ``Client.close_sync()`` unconditionally asserts ``self.session`` is a
-        ``SyncClientSession`` — but under gql 4.0 that attribute is only ever
-        set by ``connect_sync()`` (called implicitly by ``execute_gql`` /
-        ``client.execute``). If this ``GraphQL`` was constructed but never
-        used to run a query, the underlying ``Client`` has no ``session``
-        attribute yet and ``close_sync()`` raises ``AttributeError`` instead
-        of being a no-op. Only close a session that was actually opened.
-        """
-        if hasattr(self.client, "session"):
-            self.client.close_sync()
-        self.tls_profile.cleanup()
+        """Release GraphQL transport state and runtime-only TLS material."""
+        self._gql.close()
 
     @staticmethod
     def _graphql_endpoint(url: str) -> str:
@@ -91,17 +82,7 @@ class GraphQL:
         Raises:
             ParameterError: If the query fails or the response carries errors.
         """
-        try:
-            query = gql(query_str)
-            result = self.client.execute(
-                query, variable_values=variables, operation_name=operation_name
-            )
-            if "errors" in result:
-                raise ParameterError(f"GraphQL errors: {result['errors']}")
-            return result
-        except Exception as e:
-            logging.error("GraphQL execution failed: error_type=%s", type(e).__name__)
-            raise ParameterError(f"Query execution failed: {type(e).__name__}") from e
+        return self._gql.execute(query_str, variables, operation_name)
 
     def enable_pull_request_auto_merge(
         self, pull_request_id: str, merge_method: str = "MERGE"

github_agent/mcp_server.py

@@ -39,7 +39,7 @@ from github_agent.auth import (
 )
 from github_agent.github_response_models import PagesAlreadyEnabled, PagesNotEnabled
 
-__version__ = "2.4.3"
+__version__ = "2.4.4"
 logger = get_logger("GithubMCPServer")
 logger.setLevel(logging.INFO)
 

{github_agent-2.4.3.dist-info → github_agent-2.4.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: github-agent
-Version: 2.4.3
+Version: 2.4.4
 Summary: GitHub Agent for MCP
 Author-email: Project Maintainers <maintainers@users.noreply.github.com>
 License: MIT
@@ -12,9 +12,8 @@ Classifier: Programming Language :: Python :: 3
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: gql[requests]>=4.4.0
 Requires-Dist: requests>=2.34.2
-Requires-Dist: agent-connector-sdk[mcp]<1.0.0,>=0.3.11
+Requires-Dist: agent-connector-sdk[graphql,mcp]<1.0.0,>=0.3.13
 Provides-Extra: mcp
 Provides-Extra: all
 Provides-Extra: test
@@ -45,7 +44,7 @@ Dynamic: license-file
 ![PyPI - Wheel](https://img.shields.io/pypi/wheel/github-agent)
 ![PyPI - Implementation](https://img.shields.io/pypi/implementation/github-agent)
 
-*Version: 2.4.3*
+*Version: 2.4.4*
 
 > **Documentation** — Installation, deployment, and usage across the MCP, API, and
 > [official documentation](https://knuckles-team.github.io/github-agent/).

{github_agent-2.4.3.dist-info → github_agent-2.4.4.dist-info}/RECORD

@@ -1,14 +1,14 @@
-github_agent/__init__.py,sha256=OPBnVa8RJVtYMGheuiuNN0f5i8tn6elz4XCmUBGelSo,22
+github_agent/__init__.py,sha256=zUyWRJ2idk3EkNSUQlyMfzQtAlpD5CCHF2ifejOcf-w,22
 github_agent/__main__.py,sha256=O4Xg__fMvKfUsA6m61pLV56jlYY6EORiLhgDLlp0vJ8,92
 github_agent/api_client.py,sha256=O69YfkXBKyVd59EC8szaMlw3_ZxwEEvsLSvhWlfYnvk,1581
 github_agent/auth.py,sha256=isNd6DYbl_syB_wIbC399d_-157jqH1jqJfDmBn8h-0,4671
-github_agent/github_gql.py,sha256=GdkazLUDRwu5AKv2cikT8evpQAGHD1nBZ4f4BtJ2f84,5382
+github_agent/github_gql.py,sha256=8LxyFlUdtJArPO3s-YgZSL5WqnjGLp3ZuvGgVbPtVHM,4322
 github_agent/github_http.py,sha256=v-H9Nny1Q45yQH-1vsDpog4GOvAUSI786XFkctmXdik,4362
 github_agent/github_input_models.py,sha256=yqqPGHb206th736MZvbnxq9F1JRsBUBODNeNVc08F3A,15568
 github_agent/github_response_models.py,sha256=mllSt0frBgzklUtfuHywJ2Ald3m99So948Xikt3H2Tc,8859
 github_agent/kg_ingest.py,sha256=iH_NKpd8a6GtNxfZTp8Oah8VZkcXjI74KDf8P9NbtjA,17978
 github_agent/mcp_config.json,sha256=2OOXrwO1sDLyHQqpZwhvDHizPIe3by6YmK4KFE333gI,23
-github_agent/mcp_server.py,sha256=ChllOlIhkx8PzIJxtDwtDGzH2F6ybRTr0B5YuJmAgbg,126620
+github_agent/mcp_server.py,sha256=33fgoJXLclmFKHyGzFJDv78JhvYZ-bcQ3rKVEQ_JB_8,126620
 github_agent/api/__init__.py,sha256=ntaEPqk-dsAV3HyyDLc2hYpc9nTTyG5CKI-SVDDjc24,27
 github_agent/api/api_client_base.py,sha256=tMXKrZIXg6aQ6OK4kNKBDcNc8BrM0Vml_66SwA5ZgEw,7485
 github_agent/api/api_client_branches.py,sha256=bCmqGUWQ27GFTGxcFkQ7D-ZEltKp6APVsk_Fw9wzdjY,3940
@@ -92,7 +92,7 @@ github_agent/skills/github-triage-resolver/WORKFLOW.md,sha256=orvcRmaV-qb_JKoGSw
 github_agent/skills/github-triage-resolver/references/resolution-rules.md,sha256=lGPvfLCSHpNIeh2rKWcH2k_2lla3ovgBxzwvtMmyQ-0,4133
 github_agent/skills/github-triage-resolver/scripts/classify_safe.py,sha256=8BhcDF9Q8rjJIEVOmdyO-dO-eHBjW7T-bzaYC16uY5I,9556
 github_agent/skills/github-triage-resolver/scripts/gh_write.py,sha256=SL5oA1A3lFRLepNri4CzEcuEsg9D5WKJ1000fCRKBEM,4125
-github_agent-2.4.3.dist-info/licenses/LICENSE,sha256=5ALbh4fIALWVsUhO7q1nFT1bQb-CL9sWKf7p3GgLEG8,1070
+github_agent-2.4.4.dist-info/licenses/LICENSE,sha256=5ALbh4fIALWVsUhO7q1nFT1bQb-CL9sWKf7p3GgLEG8,1070
 scripts/security_sanitizer.py,sha256=gE65xtYMGx43oYUnK0kozBm73Hx97mCdrQUM7GGzIbk,9449
 scripts/verify_api_integration.py,sha256=3AK8HKW52o1yxCDaEUtXngnh7CjYw6NiG9Eb3EIoXNE,10256
 tests/conftest.py,sha256=kgpOLUp_q8IZaI6Xd7F1tbFuio81-_l5Gd4jaGiMrnY,3636
@@ -104,7 +104,7 @@ tests/test_comment_crud.py,sha256=mHA0scds_0W0oND-VI0QMvNaYarYkHjiXd4H9-9d5kE,27
 tests/test_fleet_name.py,sha256=qCLzYTPuiaS3y0I0NLuMRloXlf_o0sue0aVH-Sn-tYQ,1893
 tests/test_github_agent_api_brute_force_coverage.py,sha256=5jWW6oVU7cmXL2sS3OZ-3gbYGhyjlYPIWqH2rsdF2fw,7347
 tests/test_github_agent_brute_force_coverage.py,sha256=Z_voXtChOYRqaNEmNzMfzvwQMj2xFBn1m8mXx5ubUe8,4849
-tests/test_github_gql.py,sha256=mpn814s9UhN4jnlp1TB8XE3OZ04adR9nhfPM9j0Gpyc,1892
+tests/test_github_gql.py,sha256=M55X5V2orZ1kVek16s1zaMJIUb6FhlfbC8k0cKkD07Y,2709
 tests/test_github_http.py,sha256=uuSMrot71Kj8WQf-QyazO4rWzpWSucRTQ1tt5wRvC_E,9563
 tests/test_hardening.py,sha256=sdlKmtk7B5yncc-r1LuFgoJfro6wNlCqY0xI73IVC4M,3710
 tests/test_init_dynamics.py,sha256=fPE_4M-bONKiqvtWLn12A0q18CUlzQhdx1UoXYkbbPk,1839
@@ -123,8 +123,8 @@ tests/test_skill_org_remediation_classify_fix_risk.py,sha256=0Y8289lhxtJ3pxYUHzO
 tests/test_skill_pr_review_sweep_summarize_prs.py,sha256=NrNaucQvM7K-k1WBcOBKtTGjxJbW16VgTHbC7dSsjm4,3821
 tests/test_skill_triage_resolver_classify_safe.py,sha256=B-aDeVmsKTsxYO8TwMEx-cmpVSmZdlKXw5993Da2ubc,9489
 tests/test_startup.py,sha256=Ix_g69k0tsZbY0-Vkeh8ADWojUdbYwTwwmnM4_hVuCc,1245
-github_agent-2.4.3.dist-info/METADATA,sha256=p61ZwVHzoGYvHKxzHWA7ulhSIKJhb9w1ofIF_LneNBU,34536
-github_agent-2.4.3.dist-info/WHEEL,sha256=mOdXR9C3wMVov7cSY8PY-XCzo3SmmWCHvGQr5csJoao,91
-github_agent-2.4.3.dist-info/entry_points.txt,sha256=Wi6GhziOqmJl-22qTYkjfVuIHEHKRrBvyyloc7NqV80,367
-github_agent-2.4.3.dist-info/top_level.txt,sha256=53Ecr8n7o_EMIqZClc11PBU07uQIggYZOk5WnvBbeOQ,50
-github_agent-2.4.3.dist-info/RECORD,,
+github_agent-2.4.4.dist-info/METADATA,sha256=Fvbp3Urv5vaeNmDqM2HZ1JaOxcEsRmAhQk7qUH09EFI,34508
+github_agent-2.4.4.dist-info/WHEEL,sha256=mOdXR9C3wMVov7cSY8PY-XCzo3SmmWCHvGQr5csJoao,91
+github_agent-2.4.4.dist-info/entry_points.txt,sha256=Wi6GhziOqmJl-22qTYkjfVuIHEHKRrBvyyloc7NqV80,367
+github_agent-2.4.4.dist-info/top_level.txt,sha256=53Ecr8n7o_EMIqZClc11PBU07uQIggYZOk5WnvBbeOQ,50
+github_agent-2.4.4.dist-info/RECORD,,

tests/test_github_gql.py

@@ -37,19 +37,37 @@ def test_construction_sets_bearer_header():
 
 
 def test_close_without_session_does_not_raise():
-    """Regression guard: under gql 4.0, Client.close_sync() unconditionally
-    asserts ``self.session`` — an attribute only ever set by connect_sync()
-    (called implicitly the first time a query executes). A GraphQL client
-    that was constructed but never used to run a query has no such session
-    yet, so close() must be a safe no-op rather than raising AttributeError.
-    """
-    g = GraphQL(url="https://api.github.com", token="tok123")
-    g.close()
+    """close() on a never-used client is a safe no-op."""
+    GraphQL(url="https://api.github.com", token="tok123").close()
+
+
+@pytest.mark.spec("CONN-GITHUB-AGENT-CONNECTOR-IDENTITY-R002")
+def test_wrapper_delegates_to_shared_graphql_client():
+    from unittest.mock import patch
+
+    from agent_connector_sdk.tls.resolve import resolve_tls_profile
+    from agent_connector_sdk.transports.graphql import GraphQLClient
+
+    profile = resolve_tls_profile("github")
+    g = GraphQL(url="https://github.example/api/v3", token="tok123", tls_profile=profile)
+    assert isinstance(g._gql, GraphQLClient)
+    assert g._gql.endpoint == "https://github.example/api/graphql"
+    assert g._gql.headers["Authorization"] == "Bearer tok123"
+    assert g._gql.tls_profile is profile
+    with patch.object(GraphQLClient, "execute", return_value={"viewer": {}}) as ex:
+        assert g.execute_gql("query { viewer { login } }", {"a": 1}, "Op") == {
+            "viewer": {}
+        }
+    ex.assert_called_once_with("query { viewer { login } }", {"a": 1}, "Op")
+
+
+def test_execute_failure_message_is_redacted_class_name():
+    from unittest.mock import patch
 
+    from agent_connector_sdk.exceptions import ParameterError
+    from agent_connector_sdk.transports.graphql import GraphQLClient
 
-def test_close_after_session_opened_closes_cleanly():
-    """Once a sync session has actually been opened, close() still closes it."""
     g = GraphQL(url="https://api.github.com", token="tok123")
-    g.client.connect_sync()
-    assert hasattr(g.client, "session")
-    g.close()
+    with patch.object(GraphQLClient, "_attempt", side_effect=ValueError("boom")):
+        with pytest.raises(ParameterError, match="Query execution failed: ValueError"):
+            g.execute_gql("query { viewer { login } }")