ci: fix type check for llama-index (#1164)

Arize-ai · Dec 10, 2024 · 4e5d219 · 4e5d219
1 parent 81cb963
commit 4e5d219
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/...ce-instrumentation-llama-index/src/openinference/instrumentation/llama_index/_callback.py b/...ce-instrumentation-llama-index/src/openinference/instrumentation/llama_index/_callback.py
@@ -502,7 +502,7 @@ def _get_span_kind(event_type: Optional[CBEventType]) -> OpenInferenceSpanKindVa
 
 def _message_payload_to_attributes(message: Any) -> Dict[str, Optional[str]]:
     if isinstance(message, ChatMessage):
-        message_attributes = {
+        message_attributes: Dict[str, Any] = {
             MESSAGE_ROLE: message.role.value,
             MESSAGE_CONTENT: message.content,
         }

diff --git a/python/tox.ini b/python/tox.ini
@@ -57,7 +57,7 @@ commands_pre =
   openai-latest: uv pip install -U openai 'httpx<0.28'
   vertexai: uv pip install --reinstall {toxinidir}/instrumentation/openinference-instrumentation-vertexai[test]
   vertexai-latest: uv pip install -U vertexai 'httpx<0.28'
-  llama_index: uv pip install --reinstall {toxinidir}/instrumentation/openinference-instrumentation-llama-index[test]
+  llama_index: uv pip install --reinstall {toxinidir}/instrumentation/openinference-instrumentation-llama-index[test] 'httpx<0.28'
   llama_index-latest: uv pip install -U llama-index llama-index-core 'httpx<0.28'
   dspy: uv pip install --reinstall {toxinidir}/instrumentation/openinference-instrumentation-dspy[test]
   dspy-latest: uv pip install -U dspy-ai 'httpx<0.28'