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
@@ -13,6 +16,10 @@ from pydantic import BaseModel, Field
 from graphviz import Digraph
 
 
+
+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):
     """Trigger a callback when a blur events occurs or the enter key is pressed."""
@@ -76,7 +83,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 Node(BaseModel):
     id: int

requirements.txt

@@ -2,3 +2,4 @@ solara==1.31.0
 openai==1.17.0
 instructor==1.1.0
 graphviz==0.20.3
+langsmith==0.1.47