Skip to content

Commit 053c17f

Browse files
committed
feat(langfuse): add handler to dependency
1 parent de52d3d commit 053c17f

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

backend/api/core/dependencies.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
from fastapi import Depends
55
from langchain_mcp_adapters.tools import load_mcp_tools
66
from langchain_openai import ChatOpenAI
7+
from langfuse.callback import CallbackHandler
78
from sqlalchemy.ext.asyncio import AsyncEngine, create_async_engine
89

910
from api.core.agent.persistence import checkpointer_context
@@ -47,3 +48,17 @@ async def setup_graph() -> AsyncGenerator[Resource]:
4748
tools=tools,
4849
session=session,
4950
)
51+
52+
53+
def get_langfuse_handler() -> CallbackHandler:
54+
55+
return CallbackHandler(
56+
public_key=settings.langfuse_public_key,
57+
secret_key=settings.langfuse_secret_key,
58+
host=settings.langfuse_host,
59+
session_id=settings.environment,
60+
environment=settings.environment,
61+
)
62+
63+
64+
LangfuseHandlerDep = Annotated[CallbackHandler, Depends(get_langfuse_handler)]

0 commit comments

Comments
 (0)