Add langfuse

.gitignore

@@ -1,3 +1,4 @@
 __pycache__
 .venv
-tool_tests.ipynb
+tool_tests.ipynb
+.env

basic_agent.py

@@ -7,11 +7,15 @@ from llama_index.tools.wikipedia import WikipediaToolSpec
 from llama_index.core.tools.tool_spec.load_and_search import LoadAndSearchToolSpec
 from llama_index.readers.web import SimpleWebPageReader
 from llama_index.core.tools.ondemand_loader_tool import OnDemandLoaderTool
+from langfuse.llama_index import LlamaIndexInstrumentor
 
 class BasicAgent:
     def __init__(self):
         llm = HuggingFaceInferenceAPI(model_name="Qwen/Qwen2.5-Coder-32B-Instruct")
 
+        # Langfuse
+        self.instrumentor = LlamaIndexInstrumentor()
+
         # Initialize tools
         tool_spec = DuckDuckGoSearchToolSpec()
         search_tool = FunctionTool.from_defaults(tool_spec.duckduckgo_full_search)
@@ -47,4 +51,5 @@ class BasicAgent:
 
     async def __call__(self, question: str) -> str:
         response = await self.agent.run(user_msg=question) # ctx=self.ctx)
+        self.instrumentor.flush()
         return response.response.content

requirements.txt

@@ -6,4 +6,5 @@ llama_index-tools-duckduckgo
 llama_index-tools-wikipedia
 llama-index-indices-managed-bge-m3
 llama-index-embeddings-huggingface
-llama-index-readers-web
+llama-index-readers-web
+langfuse llama-index