Add langsmith

app.py

@@ -1,3 +1,6 @@
+#from dotenv import load_dotenv, find_dotenv
+#_ = load_dotenv(find_dotenv())
+
 import solara
 from typing import Any, Callable, Optional, TypeVar, Union, cast, overload, List
 from typing_extensions import TypedDict
@@ -10,6 +13,9 @@ import openai
 from openai import OpenAI
 import instructor
 from pydantic import BaseModel, Field
+from langsmith import traceable
+from langsmith.wrappers import wrap_openai
+
 
 # NEEDED FOR INPUT TEXT AREA INSTEAD OF INPUT TEXT
 def use_change(el: reacton.core.Element, on_value: Callable[[Any], Any], enabled=True):
@@ -74,7 +80,12 @@ def InputTextarea(
 
 # EXTRACTION
 openai.api_key = os.environ['OPENAI_API_KEY']
-client = instructor.from_openai(OpenAI())
+
+# Wrap the OpenAI client with LangSmith
+client = wrap_openai(OpenAI())
+
+# Patch the client with instructor
+client = instructor.from_openai(client, mode=instructor.Mode.TOOLS)
 
 class Person(BaseModel):
     name: str
@@ -138,7 +149,8 @@ def Page():
                     aux.value = f"{obj}"
         def result():
             if messages.value != []:
-                response(messages.value[-1]["content"])
+                if messages.value[-1]["role"] == "user":
+                    response(messages.value[-1]["content"])
         result = solara.lab.use_task(result, dependencies=[user_message_count])
         InputTextarea("Enter text:", value=text_block, continuous_update=False)
         solara.Button(label="Extract names and ages of users", on_click=send)

requirements.txt

@@ -1,3 +1,4 @@
 solara==1.31.0
 openai==1.17.0
 instructor==1.1.0
+langsmith==0.1.47