nvidia-nat-data-flywheel 1.3.0a20250917__py3-none-any.whl → 1.3.0a20250923__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -56,7 +56,7 @@ class DFWToDictProcessor(Processor[DFWRecordT, dict]):
56
56
  return json.loads(item.model_dump_json(by_alias=True))
57
57
 
58
58
 
59
- class SpanToDFWRecordProcessor(Processor[Span, DFWRecordT], TypeIntrospectionMixin):
59
+ class SpanToDFWRecordProcessor(Processor[Span, DFWRecordT | None], TypeIntrospectionMixin):
60
60
  """Processor that converts a Span to a Data Flywheel record.
61
61
 
62
62
  Extracts trace data from spans and uses the trace adapter registry to convert
@@ -79,7 +79,9 @@ class SpanToDFWRecordProcessor(Processor[Span, DFWRecordT], TypeIntrospectionMix
79
79
 
80
80
  match item.attributes.get("nat.event_type"):
81
81
  case IntermediateStepType.LLM_START:
82
- dfw_record = span_to_dfw_record(span=item, to_type=self.output_type, client_id=self._client_id)
82
+ # Extract the concrete type from Optional[DFWRecordT] to avoid passing Optional to converters
83
+ target_type = self.extract_non_optional_type(self.output_type)
84
+ dfw_record = span_to_dfw_record(span=item, to_type=target_type, client_id=self._client_id)
83
85
  return cast(DFWRecordT | None, dfw_record)
84
86
  case _:
85
87
  logger.debug("Unsupported event type: '%s'", item.attributes.get("nat.event_type"))
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: nvidia-nat-data-flywheel
3
- Version: 1.3.0a20250917
3
+ Version: 1.3.0a20250923
4
4
  Summary: Subpackage for NVIDIA Data Flywheel Blueprint integration in NeMo Agent Toolkit
5
5
  Keywords: ai,observability,nemo,data flywheel
6
6
  Classifier: Programming Language :: Python
@@ -10,7 +10,7 @@ Classifier: Programming Language :: Python :: 3.13
10
10
  Requires-Python: <3.14,>=3.11
11
11
  Description-Content-Type: text/markdown
12
12
  Requires-Dist: elasticsearch~=8.1
13
- Requires-Dist: nvidia-nat==v1.3.0a20250917
13
+ Requires-Dist: nvidia-nat==v1.3.0a20250923
14
14
 
15
15
  <!--
16
16
  SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
@@ -7,7 +7,7 @@ nat/plugins/data_flywheel/observability/exporter/dfw_exporter.py,sha256=ZCxvB0Hg
7
7
  nat/plugins/data_flywheel/observability/mixin/__init__.py,sha256=GUJrgGtpvyMUCjUBvR3faAdv-tZzbU9W-izgx9aMEQg,680
8
8
  nat/plugins/data_flywheel/observability/mixin/elasticsearch_mixin.py,sha256=Taf4OypW6goBLHQ7TTzVqBCpYJdSFt9jQHQOygXO9dQ,3058
9
9
  nat/plugins/data_flywheel/observability/processor/__init__.py,sha256=Lj0nT8rGFxd3ITrrW5ZMRYsVwkwtTWzmJYGRsW-r_VA,1072
10
- nat/plugins/data_flywheel/observability/processor/dfw_record_processor.py,sha256=aRRhEokFKNNlWoeQizBbFvjaY2CgqEN2CI5wNJjfins,3074
10
+ nat/plugins/data_flywheel/observability/processor/dfw_record_processor.py,sha256=qGegvaxyOkSiFji-rCtPBlnLySO7jfB3t3DeSA5b19s,3265
11
11
  nat/plugins/data_flywheel/observability/processor/trace_conversion/__init__.py,sha256=1ZME9GRzJCti2ExzNHRqYK9fhhnRAkrep7Db4AH-Sbg,1159
12
12
  nat/plugins/data_flywheel/observability/processor/trace_conversion/span_extractor.py,sha256=nX2uKR0DUMaxC5jpeZc741cJR1PiakQPMGDrrwPbjJY,2750
13
13
  nat/plugins/data_flywheel/observability/processor/trace_conversion/span_to_dfw_record.py,sha256=bqS4BJ0iJ-v_Mh-jnSOHq_Dh4eKyMmnpDS0T4kv8rlQ,4929
@@ -31,8 +31,8 @@ nat/plugins/data_flywheel/observability/schema/sink/elasticsearch/__init__.py,sh
31
31
  nat/plugins/data_flywheel/observability/schema/sink/elasticsearch/contract_version.py,sha256=eAxkIZG2uYl0e8QNzcLiDhqzYbaibx0XBV6jsfNm5UM,1153
32
32
  nat/plugins/data_flywheel/observability/schema/sink/elasticsearch/dfw_es_record.py,sha256=104nkmI2ZbiiP_qrdVUZkn4i4N_OBYCyUM-q-a4Orjc,9176
33
33
  nat/plugins/data_flywheel/observability/utils/deserialize.py,sha256=WTRaxnNy034X1_qaHVDI0CWgvrZ7F1Oq0QnFL9_STYs,1569
34
- nvidia_nat_data_flywheel-1.3.0a20250917.dist-info/METADATA,sha256=sAQPfgUjjL0Gfr7lscb_zl-jsfbCcABxWm-bxpzS3Qo,1693
35
- nvidia_nat_data_flywheel-1.3.0a20250917.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
36
- nvidia_nat_data_flywheel-1.3.0a20250917.dist-info/entry_points.txt,sha256=ElemL8x3DrrcLB-NsVsrUWKbHMD6rapsvs8qoIGVWrY,295
37
- nvidia_nat_data_flywheel-1.3.0a20250917.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
38
- nvidia_nat_data_flywheel-1.3.0a20250917.dist-info/RECORD,,
34
+ nvidia_nat_data_flywheel-1.3.0a20250923.dist-info/METADATA,sha256=cb8aqP4EE22qyocfgA6AUbGeHq_mVndDPMLx4Mfooco,1693
35
+ nvidia_nat_data_flywheel-1.3.0a20250923.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
36
+ nvidia_nat_data_flywheel-1.3.0a20250923.dist-info/entry_points.txt,sha256=ElemL8x3DrrcLB-NsVsrUWKbHMD6rapsvs8qoIGVWrY,295
37
+ nvidia_nat_data_flywheel-1.3.0a20250923.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
38
+ nvidia_nat_data_flywheel-1.3.0a20250923.dist-info/RECORD,,