add repro script

Author: agola11

python/bench/tracing_script.py

@@ -0,0 +1,33 @@
+from langsmith import traceable, wrappers
+from openai import Client
+
+import os
+os.environ["LANGCHAIN_PROJECT"] = "llm_messages_test_py"
+os.environ["LANGSMITH_USE_PYO3_CLIENT"] = "true"
+
+openai = wrappers.wrap_openai(Client())
+
+import openai
+from langsmith import traceable
+from langsmith.wrappers import wrap_openai
+
+client = wrap_openai(openai.Client())
+
+@traceable(run_type="tool", name="Retrieve Context")
+def my_tool(question: str) -> str:
+    return "During this morning's meeting, we solved all world conflict."
+
+@traceable(name="Chat Pipeline")
+def chat_pipeline(question: str):
+    context = my_tool(question)
+    messages = [
+        { "role": "system", "content": "You are a helpful assistant. Please respond to the user's request only based on the given context." },
+        { "role": "user", "content": f"Question: {question}\nContext: {context}"}
+    ]
+    chat_completion = client.chat.completions.create(
+        model="gpt-4o-mini", messages=messages
+    )
+    return chat_completion.choices[0].message.content
+
+if __name__ == "__main__":
+    chat_pipeline("Can you summarize this morning's meetings?")

python/langsmith/client.py

@@ -1288,6 +1288,7 @@ def create_run(
             and run_create.get("dotted_order") is not None
         ):
             if self._pyo3_client is not None:
+                print("RUN_CREATE", run_create)
                 self._pyo3_client.create_run(run_create)
             elif self.tracing_queue is not None:
                 serialized_op = serialize_run_dict("post", run_create)