tools

src/main.py

@@ -2,6 +2,8 @@ import os
 from dotenv import load_dotenv
 
 from nicegui import ui, app
+from langfuse import get_client
+from pydantic_ai.agent import Agent
 
 from components import Header, Sidebar
 from pages import DashboardPage, OllamaManagerPage
@@ -11,13 +13,22 @@ import logging
 from tools import TOOLS
 from tools.base_tool import ToolContext, set_tool_context
 
+load_dotenv()
+
 logging.basicConfig(
     level=logging.INFO,
     format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
 )
 logging.getLogger('watchfiles').setLevel(logging.WARNING)
 
-load_dotenv()
+langfuse = get_client()
+
+if langfuse.auth_check():
+    print("Langfuse client is authenticated and ready!")
+else:
+    print("Authentication failed. Please check your credentials and host.")
+
+Agent.instrument_all()
 
 app.add_static_files('/static', 'src/static')