nvidia-nat-data-flywheel 1.3.0a20251005__py3-none-any.whl → 1.3.0a20251007__py3-none-any.whl

nat/plugins/data_flywheel/observability/exporter/dfw_elasticsearch_exporter.py

@@ -43,11 +43,11 @@ class DFWElasticsearchExporter(ElasticsearchMixin, DFWExporter):
             max_queue_size: The maximum queue size for exporting spans.
             drop_on_overflow: Whether to drop spans on overflow.
             shutdown_timeout: The shutdown timeout in seconds.
-            **elasticsearch_kwargs: Additional arguments for ElasticsearchMixin:
-                - endpoint: The elasticsearch endpoint.
-                - index: The elasticsearch index name.
-                - elasticsearch_auth: The elasticsearch authentication credentials.
-                - headers: The elasticsearch headers.
+            elasticsearch_kwargs: Additional arguments for ElasticsearchMixin:
+            - endpoint: The elasticsearch endpoint.
+            - index: The elasticsearch index name.
+            - elasticsearch_auth: The elasticsearch authentication credentials.
+            - headers: The elasticsearch headers.
         """
         # Initialize both mixins - ElasticsearchMixin expects elasticsearch_kwargs,
         # DFWExporter expects the standard exporter parameters

nat/plugins/data_flywheel/observability/processor/trace_conversion/__init__.py

@@ -16,7 +16,7 @@
 from .span_extractor import extract_timestamp
 from .span_extractor import extract_token_usage
 from .span_extractor import extract_usage_info
-from .span_to_dfw_record import span_to_dfw_record
+from .span_to_dfw import span_to_dfw_record
 from .trace_adapter_registry import TraceAdapterRegistry
 from .trace_adapter_registry import register_adapter
 

nat/plugins/data_flywheel/observability/processor/trace_conversion/adapter/elasticsearch/openai_converter.py

@@ -24,12 +24,12 @@ from nat.plugins.data_flywheel.observability.schema.provider.openai_message impo
 from nat.plugins.data_flywheel.observability.schema.provider.openai_trace_source import OpenAITraceSource
 from nat.plugins.data_flywheel.observability.schema.sink.elasticsearch.dfw_es_record import AssistantMessage
 from nat.plugins.data_flywheel.observability.schema.sink.elasticsearch.dfw_es_record import DFWESRecord
+from nat.plugins.data_flywheel.observability.schema.sink.elasticsearch.dfw_es_record import ESRequest
 from nat.plugins.data_flywheel.observability.schema.sink.elasticsearch.dfw_es_record import FinishReason
 from nat.plugins.data_flywheel.observability.schema.sink.elasticsearch.dfw_es_record import Function
 from nat.plugins.data_flywheel.observability.schema.sink.elasticsearch.dfw_es_record import FunctionDetails
 from nat.plugins.data_flywheel.observability.schema.sink.elasticsearch.dfw_es_record import FunctionMessage
 from nat.plugins.data_flywheel.observability.schema.sink.elasticsearch.dfw_es_record import Message
-from nat.plugins.data_flywheel.observability.schema.sink.elasticsearch.dfw_es_record import Request
 from nat.plugins.data_flywheel.observability.schema.sink.elasticsearch.dfw_es_record import RequestTool
 from nat.plugins.data_flywheel.observability.schema.sink.elasticsearch.dfw_es_record import Response
 from nat.plugins.data_flywheel.observability.schema.sink.elasticsearch.dfw_es_record import ResponseChoice
@@ -77,7 +77,7 @@ def create_message_by_role(role: str, content: str | None, **kwargs) -> Message:
     Args:
         role (str): The message role
         content (str): The message content
-        **kwargs: Additional role-specific parameters
+        kwargs: Additional role-specific parameters
 
     Returns:
         Message: The appropriate message type for the role
@@ -314,11 +314,11 @@ def convert_langchain_openai(trace_source: TraceContainer) -> DFWESRecord:
     temperature = None
     max_tokens = None
 
-    request = Request(messages=messages,
-                      model=model_name,
-                      tools=request_tools if request_tools else None,
-                      temperature=temperature,
-                      max_tokens=max_tokens)
+    request = ESRequest(messages=messages,
+                        model=model_name,
+                        tools=request_tools if request_tools else None,
+                        temperature=temperature,
+                        max_tokens=max_tokens)
 
     # Transform chat responses
     response_choices = []

nat/plugins/data_flywheel/observability/schema/sink/elasticsearch/dfw_es_record.py

@@ -27,7 +27,8 @@ from pydantic import field_validator
 from pydantic import model_validator
 
 from nat.plugins.data_flywheel.observability.schema.schema_registry import register_schema
-from nat.plugins.data_flywheel.observability.schema.sink.elasticsearch.contract_version import ContractVersion
+
+from .contract_version import ContractVersion
 
 logger = logging.getLogger(__name__)
 
@@ -135,7 +136,7 @@ class RequestTool(BaseModel):
     function: FunctionDetails = Field(..., description="The function details.")
 
 
-class Request(BaseModel):
+class ESRequest(BaseModel):
     """Request structure used in requests."""
 
     model_config = ConfigDict(extra="allow")  # Allow extra fields
@@ -199,7 +200,7 @@ class DFWESRecord(BaseModel):
                                               description="Contract version for compatibility tracking")
 
     # Core fields (backward compatible)
-    request: Request = Field(..., description="The OpenAI ChatCompletion request.")
+    request: ESRequest = Field(..., description="The OpenAI ChatCompletion request.")
     response: Response = Field(..., description="The OpenAI ChatCompletion response.")
     client_id: str = Field(..., description="Identifier of the application or deployment that generated traffic.")
     workload_id: str = Field(..., description="Stable identifier for the logical task / route / agent node.")

nat/plugins/data_flywheel/observability/utils/__init__.py

@@ -0,0 +1,14 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.

{nvidia_nat_data_flywheel-1.3.0a20251005.dist-info → nvidia_nat_data_flywheel-1.3.0a20251007.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-data-flywheel
-Version: 1.3.0a20251005
+Version: 1.3.0a20251007
 Summary: Subpackage for NVIDIA Data Flywheel Blueprint integration in NeMo Agent Toolkit
 Keywords: ai,observability,nemo,data flywheel
 Classifier: Programming Language :: Python
@@ -10,7 +10,7 @@ Classifier: Programming Language :: Python :: 3.13
 Requires-Python: <3.14,>=3.11
 Description-Content-Type: text/markdown
 Requires-Dist: elasticsearch~=8.1
-Requires-Dist: nvidia-nat==v1.3.0a20251005
+Requires-Dist: nvidia-nat==v1.3.0a20251007
 
 <!--
 SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.

{nvidia_nat_data_flywheel-1.3.0a20251005.dist-info → nvidia_nat_data_flywheel-1.3.0a20251007.dist-info}/RECORD

@@ -2,21 +2,21 @@ nat/meta/pypi.md,sha256=kqTY91XBsWVIQNt63BVOjEj02f8YYNG0WVoFn7HfWsc,1137
 nat/plugins/data_flywheel/observability/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
 nat/plugins/data_flywheel/observability/register.py,sha256=2D1UvjTtz3fwhqadXbV-flz4EsBBrTHpr8vCPMgQTiI,3310
 nat/plugins/data_flywheel/observability/exporter/__init__.py,sha256=GUJrgGtpvyMUCjUBvR3faAdv-tZzbU9W-izgx9aMEQg,680
-nat/plugins/data_flywheel/observability/exporter/dfw_elasticsearch_exporter.py,sha256=CREldRXPJAs4LVFipHPQks8qXX5V--cwVAyACRHNrCM,3722
+nat/plugins/data_flywheel/observability/exporter/dfw_elasticsearch_exporter.py,sha256=Ou9rZYhPmGUi-F3ECH5eIAPhpSUvbYpUS20Uictg-sk,3704
 nat/plugins/data_flywheel/observability/exporter/dfw_exporter.py,sha256=ZCxvB0HgSpnKN8lwYe6AfoPbMe7Un9T9C2FaVRFuGAQ,4336
 nat/plugins/data_flywheel/observability/mixin/__init__.py,sha256=GUJrgGtpvyMUCjUBvR3faAdv-tZzbU9W-izgx9aMEQg,680
 nat/plugins/data_flywheel/observability/mixin/elasticsearch_mixin.py,sha256=Taf4OypW6goBLHQ7TTzVqBCpYJdSFt9jQHQOygXO9dQ,3058
 nat/plugins/data_flywheel/observability/processor/__init__.py,sha256=Lj0nT8rGFxd3ITrrW5ZMRYsVwkwtTWzmJYGRsW-r_VA,1072
 nat/plugins/data_flywheel/observability/processor/dfw_record_processor.py,sha256=qGegvaxyOkSiFji-rCtPBlnLySO7jfB3t3DeSA5b19s,3265
-nat/plugins/data_flywheel/observability/processor/trace_conversion/__init__.py,sha256=1ZME9GRzJCti2ExzNHRqYK9fhhnRAkrep7Db4AH-Sbg,1159
+nat/plugins/data_flywheel/observability/processor/trace_conversion/__init__.py,sha256=FSSR9Yx_hXXtWoHIzRdT5YLgWNX16Tbkywx5MO5ojEo,1152
 nat/plugins/data_flywheel/observability/processor/trace_conversion/span_extractor.py,sha256=nX2uKR0DUMaxC5jpeZc741cJR1PiakQPMGDrrwPbjJY,2750
-nat/plugins/data_flywheel/observability/processor/trace_conversion/span_to_dfw_record.py,sha256=bqS4BJ0iJ-v_Mh-jnSOHq_Dh4eKyMmnpDS0T4kv8rlQ,4929
+nat/plugins/data_flywheel/observability/processor/trace_conversion/span_to_dfw.py,sha256=bqS4BJ0iJ-v_Mh-jnSOHq_Dh4eKyMmnpDS0T4kv8rlQ,4929
 nat/plugins/data_flywheel/observability/processor/trace_conversion/trace_adapter_registry.py,sha256=qrNdSuYjqbTTtbu7mN3cPfBs7r5sn_MpBUPw2RWbabM,9911
 nat/plugins/data_flywheel/observability/processor/trace_conversion/adapter/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
 nat/plugins/data_flywheel/observability/processor/trace_conversion/adapter/register.py,sha256=Xm05kMVRJ8h65QCKTh3UsmIZ8dOMZhKHBDjklGZ9m_E,1061
 nat/plugins/data_flywheel/observability/processor/trace_conversion/adapter/elasticsearch/__init__.py,sha256=GUJrgGtpvyMUCjUBvR3faAdv-tZzbU9W-izgx9aMEQg,680
 nat/plugins/data_flywheel/observability/processor/trace_conversion/adapter/elasticsearch/nim_converter.py,sha256=XXbWePbVrpVywlr79VE9Hk6sWTG_zQI8AV7ykg6CTD4,1763
-nat/plugins/data_flywheel/observability/processor/trace_conversion/adapter/elasticsearch/openai_converter.py,sha256=pV1XIJiy6vSRJoiADU3RFmF0u_2Kh74oUNSz7jIq3V4,14767
+nat/plugins/data_flywheel/observability/processor/trace_conversion/adapter/elasticsearch/openai_converter.py,sha256=gfCfUKWjDlEgwf1m_tb5RRG0_AnMRV9BzAIC3I1aX00,14777
 nat/plugins/data_flywheel/observability/schema/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
 nat/plugins/data_flywheel/observability/schema/register.py,sha256=P7WfvItuvYn22rYSPrZgydZPe-a9JkfUN7UUyiqsMRo,929
 nat/plugins/data_flywheel/observability/schema/schema_registry.py,sha256=tYwkFDhCJauNWlNATd2YpVgydp6HWXZCAZyN0f6ECbg,4971
@@ -29,10 +29,11 @@ nat/plugins/data_flywheel/observability/schema/provider/openai_trace_source.py,s
 nat/plugins/data_flywheel/observability/schema/sink/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
 nat/plugins/data_flywheel/observability/schema/sink/elasticsearch/__init__.py,sha256=zij8aeeVtQqZODJwrJ9xg-CY61yythnUMeJzbUJfjO8,899
 nat/plugins/data_flywheel/observability/schema/sink/elasticsearch/contract_version.py,sha256=eAxkIZG2uYl0e8QNzcLiDhqzYbaibx0XBV6jsfNm5UM,1153
-nat/plugins/data_flywheel/observability/schema/sink/elasticsearch/dfw_es_record.py,sha256=104nkmI2ZbiiP_qrdVUZkn4i4N_OBYCyUM-q-a4Orjc,9176
+nat/plugins/data_flywheel/observability/schema/sink/elasticsearch/dfw_es_record.py,sha256=4SwDwf5Ng2DvTMjDBvxL36_Rp00gceTI2IRS2HQV6eM,9116
+nat/plugins/data_flywheel/observability/utils/__init__.py,sha256=GUJrgGtpvyMUCjUBvR3faAdv-tZzbU9W-izgx9aMEQg,680
 nat/plugins/data_flywheel/observability/utils/deserialize.py,sha256=3dMPFQzcReX9gL1Di2iyP1czqiFQzOI7glIn9i8m3Jo,1568
-nvidia_nat_data_flywheel-1.3.0a20251005.dist-info/METADATA,sha256=142rjWlF35MjS_gx6aGX_uOFvGgzS3bVIbI0tZHIOmA,1693
-nvidia_nat_data_flywheel-1.3.0a20251005.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat_data_flywheel-1.3.0a20251005.dist-info/entry_points.txt,sha256=ElemL8x3DrrcLB-NsVsrUWKbHMD6rapsvs8qoIGVWrY,295
-nvidia_nat_data_flywheel-1.3.0a20251005.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
-nvidia_nat_data_flywheel-1.3.0a20251005.dist-info/RECORD,,
+nvidia_nat_data_flywheel-1.3.0a20251007.dist-info/METADATA,sha256=mosj1nHXR33X7rr6mr212mOJM_MzPM-scjykgX2pHK0,1693
+nvidia_nat_data_flywheel-1.3.0a20251007.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_data_flywheel-1.3.0a20251007.dist-info/entry_points.txt,sha256=ElemL8x3DrrcLB-NsVsrUWKbHMD6rapsvs8qoIGVWrY,295
+nvidia_nat_data_flywheel-1.3.0a20251007.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_data_flywheel-1.3.0a20251007.dist-info/RECORD,,