Upload folder using huggingface_hub

common/aagents/healthcare_agent.py

@@ -3,6 +3,8 @@ import os
 from agents import Agent, OpenAIChatCompletionsModel
 from dotenv import load_dotenv
 from openai import AsyncOpenAI
+from langsmith import wrappers
+
 
 # Import tools
 from common.mcp.tools.rag_tool import rag_search, UserContext
@@ -21,6 +23,7 @@ load_dotenv()
 GEMINI_BASE_URL = "https://generativelanguage.googleapis.com/v1beta/openai/"
 google_api_key = os.getenv('GOOGLE_API_KEY')
 gemini_client = AsyncOpenAI(base_url=GEMINI_BASE_URL, api_key=google_api_key)
+gemini_client = wrappers.wrap_openai(gemini_client)
 gemini_model = OpenAIChatCompletionsModel(model="gemini-2.0-flash-exp", openai_client=gemini_client)
 
 GROQ_BASE_URL = "https://api.groq.com/openai/v1"

pyproject.toml

@@ -121,7 +121,11 @@ dependencies = [
     # =======================
     # OBSERVABILITY
     # =======================
-
+    "openinference-instrumentation-autogen>=0.1.0",
+    "openinference-instrumentation-openai>=0.1.0",
+    "opentelemetry-sdk>=1.20.0",
+    "opentelemetry-exporter-otlp>=1.20.0",
+    "opentelemetry-api>=1.20.0",
 ]
 
 [dependency-groups]

src/deep-research/app.py

@@ -3,8 +3,13 @@ import asyncio
 import time
 import html
 from io import BytesIO
-import os
 import sys
+import os
+
+# LangSmith Configuration (Overwrites)
+os.environ["LANGCHAIN_TRACING_V2"] = "true"
+os.environ["LANGCHAIN_PROJECT"] = "deep-research"
+
 from pathlib import Path
 
 # Add project root

src/deep-research/appagents/orchestrator.py

@@ -6,6 +6,8 @@ from appagents.email_agent import email_agent
 from agents.exceptions import InputGuardrailTripwireTriggered
 from core.logger import log_call
 import asyncio
+from langsmith import traceable
+
 
 class Orchestrator:
 
@@ -13,6 +15,7 @@ class Orchestrator:
         self.session = session or SQLiteSession()
 
     @log_call
+    @traceable(name="Deep Research Run")
     async def run(self, query: str):
         """ Run the deep research process, yielding the status updates and the final report"""
         trace_id = gen_trace_id()

src/deep-research/appagents/search_agent.py

@@ -1,6 +1,8 @@
 import os
 from agents import Agent, OpenAIChatCompletionsModel, WebSearchTool
 from openai import AsyncOpenAI
+from langsmith import wrappers
+
 
 from agents.model_settings import ModelSettings
 from tools.google_tools import GoogleTools
@@ -27,6 +29,7 @@ essence and ignore any fluff. Do not include any additional commentary other tha
 GEMINI_BASE_URL = "https://generativelanguage.googleapis.com/v1beta/openai/"
 google_api_key = os.getenv('GOOGLE_API_KEY')
 gemini_client = AsyncOpenAI(base_url=GEMINI_BASE_URL, api_key=google_api_key)
+gemini_client = wrappers.wrap_openai(gemini_client)
 gemini_model = OpenAIChatCompletionsModel(model="gemini-2.0-flash", openai_client=gemini_client)
 
 # search_agent = Agent(