fix: remove mistralai dependency in mistralai instrumentor (#1271)

Author: RogerHYang

python/instrumentation/openinference-instrumentation-mistralai/pyproject.toml

@@ -37,12 +37,6 @@ instruments = [
   "mistralai >= 1.0.2",
 ]
 test = [
-  "mistralai == 1.0.2",
-  "opentelemetry-sdk",
-  "opentelemetry-instrumentation-httpx",
-  "respx",
-  "pytest-asyncio",
-  "pytest.recording",
 ]
 
 [project.urls]

python/instrumentation/openinference-instrumentation-mistralai/src/openinference/instrumentation/mistralai/_stream.py

@@ -1,5 +1,8 @@
+from __future__ import annotations
+
 import logging
 from typing import (
+    TYPE_CHECKING,
     Any,
     AsyncIterator,
     Generator,
@@ -13,10 +16,12 @@
 from opentelemetry.util.types import AttributeValue
 from wrapt import ObjectProxy
 
-from mistralai.models import CompletionEvent
 from openinference.instrumentation.mistralai._utils import _finish_tracing
 from openinference.instrumentation.mistralai._with_span import _WithSpan
 
+if TYPE_CHECKING:
+    from mistralai.models import CompletionEvent
+
 __all__ = (
     "_Stream",
     "_ResponseAccumulator",

python/instrumentation/openinference-instrumentation-mistralai/test-requirements.txt

@@ -0,0 +1,6 @@
+mistralai == 1.0.2
+opentelemetry-sdk
+opentelemetry-instrumentation-httpx
+respx
+pytest-asyncio
+pytest.recording

python/tox.ini

@@ -52,7 +52,10 @@ commands_pre =
   semconv: uv pip install --reinstall {toxinidir}/openinference-semantic-conventions
   bedrock: uv pip install --reinstall {toxinidir}/instrumentation/openinference-instrumentation-bedrock[test]
   bedrock-latest: uv pip install -U boto3 anthropic
-  mistralai: uv pip install --reinstall {toxinidir}/instrumentation/openinference-instrumentation-mistralai[test]
+  mistralai: uv pip uninstall -r test-requirements.txt
+  mistralai: uv pip install --reinstall-package openinference-instrumentation-mistralai .
+  mistralai: python -c 'import openinference.instrumentation.mistralai'
+  mistralai: uv pip install -r test-requirements.txt
   mistralai-latest: uv pip install -U mistralai
   openai: uv pip uninstall -r test-requirements.txt
   openai: uv pip install --reinstall-package openinference-instrumentation-openai .