Merge pull request #90 from bukosabino/develop-adding-langsmith

Adding langsmith to monitor the traces

requirements.txt

@@ -13,15 +13,18 @@ retry==0.9.2
 typer==0.9.0
 schedule==1.2.1
 
-langchain==0.0.305
+langchain==0.2.1
 # langchainplus-sdk==0.0.20
-langsmith==0.0.41
+langsmith==0.1.65
+langchain-openai==0.1.8
+langchain-core==0.2.3
+
 qdrant-client==1.8.0
 supabase==1.0.2
 pinecone-client==2.2.2
 sentence_transformers==2.2.2
-openai==1.3.8
-tavily-python==0.2.9
+openai==1.30.5
+tavily-python==0.3.3
 
 sendgrid==6.10.0
 

src/initialize.py

@@ -19,6 +19,7 @@
 from qdrant_client.models import VectorParams
 from supabase.client import Client, create_client
 from tavily import TavilyClient
+from langsmith.wrappers import wrap_openai
 
 from src.utils import StandardSupabaseVectorStore
 
@@ -152,6 +153,7 @@ def _init_openai_client():
     client = AsyncOpenAI(
         api_key=os.environ.get("OPENAI_API_KEY"),
     )
+    client = wrap_openai(client)
     logger.info("Initialized OpenAI client")
     return client