Upload 3 files

agent.py

@@ -14,6 +14,28 @@ from dotenv import load_dotenv
 
 load_dotenv()
 
+# Configure tracing
+try:
+    if os.getenv("ARIZE_SPACE_ID") and os.getenv("ARIZE_API_KEY"):
+        from arize.otel import register
+        from openinference.instrumentation.google_genai import GoogleGenAIInstrumentor
+        from openinference.instrumentation.langchain import LangChainInstrumentor
+        
+        tracer_provider = register(
+            space_id=os.getenv("ARIZE_SPACE_ID"),
+            api_key=os.getenv("ARIZE_API_KEY"),
+            project_name=os.getenv("ARIZE_PROJECT_NAME", "langgraph-agent-test")
+        )
+        GoogleGenAIInstrumentor().instrument(tracer_provider=tracer_provider)
+        LangChainInstrumentor().instrument(tracer_provider=tracer_provider)
+        print("Tracing configured with Arize.")
+    else:
+        print("Arize tracing skipped: ARIZE_SPACE_ID or ARIZE_API_KEY not set.")
+except ImportError:
+    print("Tracing libraries not installed. Skipping tracing.")
+except Exception as e:
+    print(f"Error configuring tracing: {e}")
+
 # 1. Define the state
 class AgentState(TypedDict):
     messages: Annotated[List[BaseMessage], operator.add]

requirements.txt

@@ -9,4 +9,7 @@ langchain-google-genai
 google-auth
 langchain-tavily
 google-cloud-aiplatform
-youtube-transcript-api
+youtube-transcript-api
+arize-otel
+openinference-instrumentation-google-genai
+openinference-instrumentation-langchain