cortexhub 0.1.4__py3-none-any.whl → 0.1.5__py3-none-any.whl

cortexhub/adapters/claude_agents.py

@@ -73,6 +73,7 @@ class ClaudeAgentsAdapter(ToolAdapter):
     def __init__(self, cortex_hub: Any):
         super().__init__(cortex_hub)
         self._hook_spans: dict[str, Any] = {}
+        self._active_run_session_id: str | None = None
 
     def _get_framework_modules(self) -> list[str]:
         return ["claude_agent_sdk"]
@@ -206,6 +207,7 @@ class ClaudeAgentsAdapter(ToolAdapter):
         try:
             import claude_agent_sdk
             from claude_agent_sdk import ClaudeSDKClient, ResultMessage
+            adapter = self
 
             if getattr(claude_agent_sdk, _PATCHED_RUN_ATTR, False):
                 return
@@ -218,6 +220,7 @@ class ClaudeAgentsAdapter(ToolAdapter):
                 status = None
                 failed = False
                 cortex_hub.start_run(framework="claude_agents")
+                adapter._active_run_session_id = cortex_hub.session_id
                 try:
                     async for message in original_query(*args, **kwargs):
                         if isinstance(message, ResultMessage):
@@ -231,6 +234,7 @@ class ClaudeAgentsAdapter(ToolAdapter):
                         status = "failed"
                     if status:
                         cortex_hub.finish_run(framework="claude_agents", status=status)
+                    adapter._active_run_session_id = None
 
             claude_agent_sdk.query = patched_query
 
@@ -251,10 +255,12 @@ class ClaudeAgentsAdapter(ToolAdapter):
 
             async def patched_client_query(self, *args, **kwargs):
                 cortex_hub.start_run(framework="claude_agents")
+                adapter._active_run_session_id = cortex_hub.session_id
                 try:
                     return await original_client_query(self, *args, **kwargs)
                 except Exception:
                     cortex_hub.finish_run(framework="claude_agents", status="failed")
+                    adapter._active_run_session_id = None
                     raise
 
             async def patched_receive_response(self, *args, **kwargs):
@@ -273,6 +279,7 @@ class ClaudeAgentsAdapter(ToolAdapter):
                         status = "failed"
                     if status:
                         cortex_hub.finish_run(framework="claude_agents", status=status)
+                    adapter._active_run_session_id = None
 
             ClaudeSDKClient.query = patched_client_query
             ClaudeSDKClient.receive_response = patched_receive_response
@@ -303,6 +310,10 @@ class ClaudeAgentsAdapter(ToolAdapter):
         """
         cortex_hub = self.cortex_hub
         span_store = self._hook_spans
+        adapter = self
+
+        def _current_session_id() -> str:
+            return adapter._active_run_session_id or cortex_hub.session_id
 
         def _start_tool_span(
             *,
@@ -316,7 +327,7 @@ class ClaudeAgentsAdapter(ToolAdapter):
                 name="tool.invoke",
                 kind=SpanKind.INTERNAL,
             )
-            span.set_attribute("cortexhub.session.id", cortex_hub.session_id)
+            span.set_attribute("cortexhub.session.id", _current_session_id())
             span.set_attribute("cortexhub.agent.id", cortex_hub.agent_id)
             span.set_attribute("cortexhub.tool.name", tool_name)
             span.set_attribute("cortexhub.tool.framework", "claude_agents")

{cortexhub-0.1.4.dist-info → cortexhub-0.1.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cortexhub
-Version: 0.1.4
+Version: 0.1.5
 Summary: CortexHub Python SDK - Policy-as-Code for AI Agents
 Project-URL: Homepage, https://cortexhub.ai
 Project-URL: Documentation, https://docs.cortexhub.ai

{cortexhub-0.1.4.dist-info → cortexhub-0.1.5.dist-info}/RECORD

@@ -8,7 +8,7 @@ cortexhub/pipeline.py,sha256=EvkrEz7dGnzc0tlv02B3Kx2KQfdX4gsTuhUxj-MwuFI,3256
 cortexhub/version.py,sha256=L1gsFyWfMZYvly7nYjyH0nlWRAsm65S61_B-imDPgXE,68
 cortexhub/adapters/__init__.py,sha256=Lb6RquGPGs4exdfSB6Qb6q6HyGczozuLWSaLCtgZdyo,130
 cortexhub/adapters/base.py,sha256=_ODoJFLQLYnowse3MHQk8bN9KjE1oQr8QDNNlvPSwSs,4094
-cortexhub/adapters/claude_agents.py,sha256=etuDHQjWikItFB-rHVwDM9fFm-0WNderkOhIgOtRxjk,22499
+cortexhub/adapters/claude_agents.py,sha256=RzFxMCylL3yCgnTASKnpa5F60t0RD-IrvoDfpRKYPFY,23038
 cortexhub/adapters/crewai.py,sha256=SuyMBF9bzrqDq58ZsGZV-3uAzaAtlLD3xEeyMB2Hgig,24899
 cortexhub/adapters/langgraph.py,sha256=X0UutdATOowXcZYRv_ImiJGSO97MKgYcuxFKV-jTxJA,20850
 cortexhub/adapters/openai_agents.py,sha256=63rNZ8HItZdoxERk1BTYzzpVe8Qq_8j56jfrEICzjpU,21511
@@ -33,7 +33,7 @@ cortexhub/policy/models.py,sha256=81NPX36yru0AZ22Wivytv3uxlwir4VsTN4MZgq4bdiA,38
 cortexhub/policy/sync.py,sha256=7kAc3rd5WftipR8XKwvU8SwukIoY-0gcGW2fsf-ij_I,5999
 cortexhub/telemetry/__init__.py,sha256=RSY38hHnYtbRURLoWAhbYYxYMtCItVYZBNN6Bfny8uA,1049
 cortexhub/telemetry/otel.py,sha256=T2pC-K_S_KmrLvfcd3oHMbEwC5ibGZ-PXTZSY0xJgcQ,16633
-cortexhub-0.1.4.dist-info/METADATA,sha256=415eS3_LSI7Df293BxsVWLCm_7o7HSrF89GC4j2z7Jc,8539
-cortexhub-0.1.4.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-cortexhub-0.1.4.dist-info/licenses/LICENSE,sha256=R-lPsEVN69q2o19ge2_O9Vt9cJuGM557oEYX3XdnvIk,1066
-cortexhub-0.1.4.dist-info/RECORD,,
+cortexhub-0.1.5.dist-info/METADATA,sha256=oZPJ9gFAkIQ6D7jYLElFa9gobzaz3yyqqTGgsMRb7MM,8539
+cortexhub-0.1.5.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+cortexhub-0.1.5.dist-info/licenses/LICENSE,sha256=R-lPsEVN69q2o19ge2_O9Vt9cJuGM557oEYX3XdnvIk,1066
+cortexhub-0.1.5.dist-info/RECORD,,