bug + test for prompt decorator

Author: Andrei Bratu

src/humanloop/otel/exporter/__init__.py

@@ -74,7 +74,7 @@ def export(self, spans: Sequence[ReadableSpan]) -> SpanExportResult:
 
         for span in spans:
             # only process spans that are relevant to Humanloop
-            if not is_humanloop_span(span) or not is_llm_provider_call(span):
+            if not is_humanloop_span(span) and not is_llm_provider_call(span):
                 continue
 
             file_type = span.attributes.get(HUMANLOOP_FILE_TYPE_KEY)  # type: ignore [union-attr]

tests/integration/test_decorators.py

@@ -0,0 +1,44 @@
+import time
+from typing import Any
+
+from openai import OpenAI
+from humanloop.client import Humanloop
+
+
+def test_prompt_decorator(
+    humanloop_test_client: Humanloop,
+    sdk_test_dir: str,
+    test_prompt_config: dict[str, Any],
+    openai_key: str,
+):
+    try:
+        prompt_path = f"{sdk_test_dir}/test_prompt"
+        prompt_response = humanloop_test_client.prompts.upsert(
+            path=prompt_path,
+            **test_prompt_config,
+        )
+
+        prompt_versions_response = humanloop_test_client.prompts.list_versions(id=prompt_response.id)
+        assert len(prompt_versions_response.records) == 1
+
+        @humanloop_test_client.prompt(path=prompt_path)
+        def my_prompt(question: str) -> str:
+            openai_client = OpenAI(api_key=openai_key)
+
+            response = openai_client.chat.completions.create(
+                model="gpt-4o-mini",
+                messages=[{"role": "user", "content": question}],
+            )
+
+            return response.choices[0].message.content
+
+        assert "paris" in my_prompt("What is the capital of the France?").lower()
+
+        time.sleep(5)
+        prompt_versions_response = humanloop_test_client.prompts.list_versions(id=prompt_response.id)
+        assert len(prompt_versions_response.records) == 2
+
+        logs = humanloop_test_client.logs.list(file_id=prompt_response.id, page=1, size=50)
+        assert len(logs.items) == 1
+    finally:
+        humanloop_test_client.prompts.delete(id=prompt_response.id)