.env.example

@@ -3,6 +3,3 @@ PROVIDER_API_KEY=your-api-key
 PROVIDER_MODEL=gpt-4
 COMPOSIO_API_KEY=your-api-key
 AGENT_ID=user-12345
-LANGFUSE_PUBLIC_KEY=pk-lf-...
-LANGFUSE_SECRET_KEY=sk-lf-...
-LANGFUSE_HOST=http://localhost:3000

src/agent/service.py

@@ -2,9 +2,7 @@ from typing import AsyncIterator, AsyncContextManager, Self
 from abc import abstractmethod
 import os
 from pathlib import Path
-from langfuse import get_client, propagate_attributes,observe
+
-import langfuse
-from langfuse.langchain import CallbackHandler
 from src.agent.base import create_agent
 from src.core.logger import get_logger
 from lambda_agent_api.server import (
@@ -108,18 +106,8 @@ class AgentService:
     async def __astream(
         self, chat_id: int, text: str, attachments: list[str] = None
     ) -> AsyncIterator[AgentEventUnion]:
+        config = {"configurable": {"thread_id": chat_id}}
 
-        langfuse_client = get_client()
-        langfuse_handler = CallbackHandler()
-        config = {"configurable": {"thread_id": chat_id},"callbacks": [langfuse_handler]}
-
-        try:
-            # возможно сюда стоит включить обработку attachmanets, только  __describe_attachments нужно обернуть в @observe
-            with propagate_attributes(
-                session_id=str(chat_id),
-                user_id=str(chat_id),   # Или реальный ID юзера, если есть
-                tags=["agent-stream"]
-            ):
         new_message = text
         if attachments:
             logger.debug(f"Processing {attachments} attachments for chat {chat_id}")
@@ -127,13 +115,13 @@ class AgentService:
             new_message += "\n" + attachments_description
 
         logger.debug(f"Starting agent stream for chat {chat_id}")
+        
         # astream_events, чтобы выходили промежуточные ивенты по мере их генерации
         async for event in self._agent.astream_events(
             {"messages": [{"role": "user", "content": new_message}]},
             config=config,
             version="v2",
         ):
-        
             kind = event["event"]
 
             # пришли чанки от LLM
@@ -174,11 +162,7 @@ class AgentService:
                     yield MsgEventSendFile(path=file_path)
         
         logger.debug(f"Agent stream completed for chat {chat_id}")
-        finally:
-            logger.debug(f"Flushing Langfuse logs for chat {chat_id}")
-            langfuse_client.flush()
 
-    @observe(as_type="span", name="parse_attachments")
     async def __describe_attachments(self, raw_paths: list[str]) -> str:
         lines = []
         logger.debug(f"Describing {len(raw_paths)} attachments")