langtrace-python-sdk 1.3.6__py3-none-any.whl → 2.0.0__py3-none-any.whl

tests/openai/test_image_generation.py

@@ -30,13 +30,19 @@ def test_image_generation(openai_client, exporter):
     assert attributes.get("url.full") == "https://api.openai.com/v1/"
     assert attributes.get("llm.api") == APIS["IMAGES_GENERATION"]["ENDPOINT"]
     assert attributes.get("llm.model") == llm_model_value
-    assert attributes.get("llm.prompts") == json.dumps([prompt])
+    prompts = json.loads(attributes.get("llm.prompts"))
+    assert prompts[0]["content"] == prompt
 
     langtrace_responses = json.loads(attributes.get("llm.responses"))
+    assert isinstance(langtrace_responses, list)
     for langtrace_response in langtrace_responses:
-        assert response.data[0].url == langtrace_response.get("url")
-        assert response.data[0].revised_prompt == langtrace_response.get(
-            "revised_prompt"
+        assert isinstance(langtrace_response, dict)
+        assert "role" in langtrace_response
+        assert "content" in langtrace_response
+        assert response.data[0].url == langtrace_response["content"]["url"]
+        assert (
+            response.data[0].revised_prompt
+            == langtrace_response["content"]["revised_prompt"]
         )
 
 
@@ -67,11 +73,17 @@ async def test_async_image_generation(async_openai_client, exporter):
     assert attributes.get("url.full") == "https://api.openai.com/v1/"
     assert attributes.get("llm.api") == APIS["IMAGES_GENERATION"]["ENDPOINT"]
     assert attributes.get("llm.model") == llm_model_value
-    assert attributes.get("llm.prompts") == json.dumps([prompt])
+    prompts = json.loads(attributes.get("llm.prompts"))
+    assert prompts[0]["content"] == prompt
 
     langtrace_responses = json.loads(attributes.get("llm.responses"))
+    assert isinstance(langtrace_responses, list)
     for langtrace_response in langtrace_responses:
-        assert response.data[0].url == langtrace_response.get("url")
-        assert response.data[0].revised_prompt == langtrace_response.get(
-            "revised_prompt"
+        assert isinstance(langtrace_response, dict)
+        assert "role" in langtrace_response
+        assert "content" in langtrace_response
+        assert response.data[0].url == langtrace_response["content"]["url"]
+        assert (
+            response.data[0].revised_prompt
+            == langtrace_response["content"]["revised_prompt"]
         )

examples/openai/chat_completion.py

@@ -1,58 +0,0 @@
-from dotenv import find_dotenv, load_dotenv
-from openai import OpenAI
-
-from langtrace_python_sdk import langtrace
-from langtrace_python_sdk.utils.with_root_span import (
-    with_additional_attributes, with_langtrace_root_span)
-
-_ = load_dotenv(find_dotenv())
-
-langtrace.init(write_to_langtrace_cloud=False)
-client = OpenAI()
-
-
-@with_additional_attributes({"user.id": "1234", "user.feedback.rating": 1})
-def api1():
-    response = client.chat.completions.create(
-        model="gpt-4",
-        messages=[{"role": "user", "content": "Say this is a test three times"}],
-        stream=False,
-    )
-    return response
-
-
-@with_additional_attributes({"user.id": "5678", "user.feedback.rating": -1})
-def api2():
-    response = client.chat.completions.create(
-        model="gpt-4",
-        messages=[{"role": "user", "content": "Say this is a test three times"}],
-        stream=True,
-    )
-    return response
-
-
-@with_langtrace_root_span()
-def chat_completion():
-    response = api1()
-    response = api2()
-    result = []
-    for chunk in response:
-        if chunk.choices[0].delta.content is not None:
-            content = [
-                choice.delta.content if choice.delta and
-                choice.delta.content else ""
-                for choice in chunk.choices]
-            result.append(
-                content[0] if len(content) > 0 else "")
-
-    print("".join(result))
-    # return response
-
-
-# # print(response)
-# stream = client.chat.completions.create(
-#     model="gpt-4",
-#     messages=[{"role": "user", "content": "Say this is a test three times"}, {"role": "assistant", "content": "This is a test. This is a test. This is a test"},
-#               {"role": "user", "content": "Say this is a mock 4 times"}],
-#     stream=False,
-# )