remove output value and mime type for retriever spans

Arize-ai · Jan 30, 2024 · 26966b0 · 26966b0
1 parent 28b94f4
commit 26966b0
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 12 deletions.
diff --git a/...ion/openinference-instrumentation-dspy/src/openinference/instrumentation/dspy/__init__.py b/...ion/openinference-instrumentation-dspy/src/openinference/instrumentation/dspy/__init__.py
@@ -295,10 +295,6 @@ def __call__(
                 dict(
                     _flatten(
                         {
-                            SpanAttributes.OUTPUT_VALUE: json.dumps(
-                                prediction, cls=DSPyJSONEncoder
-                            ),
-                            SpanAttributes.OUTPUT_MIME_TYPE: OpenInferenceMimeTypeValues.JSON.value,
                             SpanAttributes.RETRIEVAL_DOCUMENTS: [
                                 {
                                     DocumentAttributes.DOCUMENT_ID: str(uuid4()),

diff --git a/...erence-instrumentation-dspy/tests/openinference/instrumentation/dspy/test_instrumentor.py b/...erence-instrumentation-dspy/tests/openinference/instrumentation/dspy/test_instrumentor.py
@@ -218,14 +218,6 @@ def forward(self, question):
         "query_or_queries": "What's the capital of the United States?"
     }
     assert span.attributes[INPUT_MIME_TYPE] == JSON.value
-    assert json.loads(span.attributes[OUTPUT_VALUE]) == {
-        "passages": [
-            "first retrieved document text",
-            "second retrieved document text",
-            "third retrieved document text",
-        ]
-    }
-    assert span.attributes[OUTPUT_MIME_TYPE] == JSON.value
     assert isinstance(
         span.attributes[f"{RETRIEVAL_DOCUMENTS}.0.{DOCUMENT_ID}"],
         str,