feat(lineage): capture tool data-source refs and agent build version in span data

.github/workflows/ci.yml

@@ -18,7 +18,7 @@ jobs:
   lint:
     timeout-minutes: 10
     name: lint
-    runs-on: ${{ github.repository == 'stainless-sdks/agentex-sdk-python' && 'depot-ubuntu-24.04' || 'ubuntu-latest' }}
+    runs-on: ${{ startsWith(github.repository, 'stainless-sdks/') && 'depot-ubuntu-24.04' || 'ubuntu-latest' }}
     if: (github.event_name == 'push' || github.event.pull_request.head.repo.fork) && (github.event_name != 'push' || github.event.head_commit.message != 'codegen metadata')
     steps:
       - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6.0.2
@@ -44,7 +44,7 @@ jobs:
     permissions:
       contents: read
       id-token: write
-    runs-on: ${{ github.repository == 'stainless-sdks/agentex-sdk-python' && 'depot-ubuntu-24.04' || 'ubuntu-latest' }}
+    runs-on: ${{ startsWith(github.repository, 'stainless-sdks/') && 'depot-ubuntu-24.04' || 'ubuntu-latest' }}
     steps:
       - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6.0.2
 
@@ -87,7 +87,7 @@ jobs:
   test:
     timeout-minutes: 10
     name: test
-    runs-on: ${{ github.repository == 'stainless-sdks/agentex-sdk-python' && 'depot-ubuntu-24.04' || 'ubuntu-latest' }}
+    runs-on: ${{ startsWith(github.repository, 'stainless-sdks/') && 'depot-ubuntu-24.04' || 'ubuntu-latest' }}
     if: github.event_name == 'push' || github.event.pull_request.head.repo.fork
     steps:
       - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6.0.2

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 75
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/sgp/agentex-sdk-b2df5f506330ad5fba5a0d518ab8a4bcf876e8c3684a4fe0d0cc5102fd9c569e.yml
-openapi_spec_hash: 132e9efdb0535d9594abadf799431cf5
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/sgp/agentex-sdk-132fed556e527565b396688a09e873b51ae42c6ad11757d49fc5ef1f2091c943.yml
+openapi_spec_hash: 0c1d8518f9f60bb8dc325e26fe1ebadc
 config_hash: 593e89b291976a5e84e4c3c3f8324354

src/agentex/lib/adk/__init__.py

@@ -29,6 +29,10 @@
 from agentex.lib.adk._modules.tasks import TasksModule
 from agentex.lib.adk._modules.tracing import TracingModule, TurnSpan
 
+# Data-source refs for lineage (SGP-6513); implementation lives in core.tracing
+from agentex.lib.core.tracing import lineage
+from agentex.lib.core.tracing.lineage import DataSourceRef, data_sources
+
 # Unified harness surface (AGX1-375)
 from agentex.lib.core.harness import (
     UnifiedEmitter,
@@ -67,6 +71,10 @@
     "events",
     "agent_task_tracker",
     "TurnSpan",
+    # Lineage data-source refs (SGP-6513)
+    "lineage",
+    "DataSourceRef",
+    "data_sources",
     # Checkpointing / LangGraph
     "create_checkpointer",
     "stream_langgraph_events",

src/agentex/lib/adk/providers/_modules/sync_provider.py

@@ -19,6 +19,7 @@
 from agentex import AsyncAgentex
 from agentex.lib.utils.logging import make_logger
 from agentex.lib.core.tracing.tracer import AsyncTracer
+from agentex.lib.core.tracing.lineage import merge_refs_into_data, resolve_refs_from_items
 
 logger = make_logger(__name__)
 
@@ -185,6 +186,9 @@ async def get_response(
                         "new_items": new_items,
                         "final_output": final_output,
                     }
+                    lineage_refs = resolve_refs_from_items(new_items)
+                    if lineage_refs:
+                        span.data = merge_refs_into_data(span.data, lineage_refs)
 
                 return response
         else:
@@ -303,6 +307,9 @@ async def stream_response(
                     "new_items": new_items,
                     "final_output": final_response_text if final_response_text else None,
                 }
+                lineage_refs = resolve_refs_from_items(new_items)
+                if lineage_refs:
+                    span.data = merge_refs_into_data(span.data, lineage_refs)
             finally:
                 # End the span after all events have been yielded
                 await trace.end_span(span)

src/agentex/lib/core/harness/tracer.py

@@ -6,6 +6,17 @@
 
 from agentex.lib.core.harness.types import OpenSpan, CloseSpan, SpanSignal
 
+try:
+    from agentex.lib.core.tracing.lineage import resolve_refs, merge_refs_into_data
+except Exception:  # keep the harness importable without optional tracing deps
+
+    def resolve_refs(tool_name: str, arguments: dict[str, Any] | None) -> list[dict[str, Any]]:  # noqa: ARG001
+        return []
+
+    def merge_refs_into_data(data: dict[str, Any] | None, refs: list[dict[str, Any]]) -> dict[str, Any]:  # noqa: ARG001
+        return dict(data or {})
+
+
 try:
     from agentex.lib.utils.logging import make_logger
 
@@ -80,6 +91,11 @@ async def handle(self, signal: SpanSignal) -> None:
                     task_id=self.task_id,
                 )
                 if span is not None:
+                    if signal.kind == "tool":
+                        refs = resolve_refs(signal.name, signal.input if isinstance(signal.input, dict) else {})
+                        if refs:
+                            data = span.data if isinstance(span.data, dict) else {}
+                            span.data = merge_refs_into_data(data, refs)
                     self._open[signal.key] = span
             elif isinstance(signal, CloseSpan):
                 span = self._open.pop(signal.key, None)

src/agentex/lib/core/services/adk/providers/openai.py

@@ -25,6 +25,7 @@
 from agentex.lib.utils.temporal import heartbeat_if_in_workflow
 from agentex.lib.core.tracing.tracer import AsyncTracer
 from agentex.lib.core.harness.emitter import UnifiedEmitter
+from agentex.lib.core.tracing.lineage import merge_refs_into_data, resolve_refs_from_items
 from agentex.types.task_message_update import StreamTaskMessageFull
 from agentex.types.task_message_content import (
     TextContent,
@@ -286,13 +287,17 @@ async def run_agent(
                     result = await Runner.run(starting_agent=agent, input=input_list)
 
                 if span:
+                    serialized_items = [
+                        item.raw_item.model_dump() if isinstance(item.raw_item, BaseModel) else item.raw_item
+                        for item in result.new_items
+                    ]
                     span.output = {
-                        "new_items": [
-                            item.raw_item.model_dump() if isinstance(item.raw_item, BaseModel) else item.raw_item
-                            for item in result.new_items
-                        ],
+                        "new_items": serialized_items,
                         "final_output": result.final_output,
                     }
+                    lineage_refs = resolve_refs_from_items(serialized_items)
+                    if lineage_refs:
+                        span.data = merge_refs_into_data(span.data, lineage_refs)
 
         return result
 
@@ -431,13 +436,17 @@ async def run_agent_auto_send(
                     result = await Runner.run(starting_agent=agent, input=input_list)
 
                 if span:
+                    serialized_items = [
+                        item.raw_item.model_dump() if isinstance(item.raw_item, BaseModel) else item.raw_item
+                        for item in result.new_items
+                    ]
                     span.output = {
-                        "new_items": [
-                            item.raw_item.model_dump() if isinstance(item.raw_item, BaseModel) else item.raw_item
-                            for item in result.new_items
-                        ],
+                        "new_items": serialized_items,
                         "final_output": result.final_output,
                     }
+                    lineage_refs = resolve_refs_from_items(serialized_items)
+                    if lineage_refs:
+                        span.data = merge_refs_into_data(span.data, lineage_refs)
 
                 tool_call_map: dict[str, Any] = {}
 
@@ -646,13 +655,17 @@ async def run_agent_streamed(
                     result = Runner.run_streamed(starting_agent=agent, input=input_list)
 
                 if span:
+                    serialized_items = [
+                        item.raw_item.model_dump() if isinstance(item.raw_item, BaseModel) else item.raw_item
+                        for item in result.new_items
+                    ]
                     span.output = {
-                        "new_items": [
-                            item.raw_item.model_dump() if isinstance(item.raw_item, BaseModel) else item.raw_item
-                            for item in result.new_items
-                        ],
+                        "new_items": serialized_items,
                         "final_output": result.final_output,
                     }
+                    lineage_refs = resolve_refs_from_items(serialized_items)
+                    if lineage_refs:
+                        span.data = merge_refs_into_data(span.data, lineage_refs)
 
         return result
 
@@ -906,12 +919,16 @@ async def run_agent_streamed_auto_send(
                     raise
 
                 if span:
+                    serialized_items = [
+                        item.raw_item.model_dump() if isinstance(item.raw_item, BaseModel) else item.raw_item
+                        for item in result.new_items
+                    ]
                     span.output = {
-                        "new_items": [
-                            item.raw_item.model_dump() if isinstance(item.raw_item, BaseModel) else item.raw_item
-                            for item in result.new_items
-                        ],
+                        "new_items": serialized_items,
                         "final_output": result.final_output,
                     }
+                    lineage_refs = resolve_refs_from_items(serialized_items)
+                    if lineage_refs:
+                        span.data = merge_refs_into_data(span.data, lineage_refs)
 
         return result

src/agentex/lib/core/temporal/plugins/openai_agents/models/temporal_streaming_model.py

@@ -64,6 +64,7 @@
 from agentex.lib import adk
 from agentex.lib.utils.logging import make_logger
 from agentex.lib.core.tracing.tracer import AsyncTracer
+from agentex.lib.core.tracing.lineage import merge_refs_into_data, resolve_refs_from_items
 from agentex.types.task_message_delta import TextDelta, ToolRequestDelta, ReasoningContentDelta, ReasoningSummaryDelta
 from agentex.types.task_message_update import StreamTaskMessageFull, StreamTaskMessageDelta
 from agentex.types.task_message_content import TextContent, ReasoningContent, ToolRequestContent, ToolResponseContent
@@ -1257,6 +1258,9 @@ async def get_response(
                         output_data["tool_outputs"] = tool_outputs
 
                     span.output = output_data
+                    lineage_refs = resolve_refs_from_items(new_items)
+                    if lineage_refs:
+                        span.data = merge_refs_into_data(span.data, lineage_refs)
 
                 # Streaming-only metrics. Token counters and the success request
                 # counter are emitted by LLMMetricsHooks.on_llm_end so they fire