b4401d0339
- 更新了 Dockerfile，在同步过程中包含 Langfuse 集成。 - 在文档中添加了新的部分，详细介绍 Langfuse 集成的目的、配置和在 Yuxi 中的使用方法。 - 更新了 Vitepress 配置，包含了指向新 Langfuse 集成文档的链接。 - 增强了路线图，反映了对 Langfuse 自托管支持的添加，并明确了集成方法。
111 lines
3.6 KiB
Python
111 lines
3.6 KiB
Python
from __future__ import annotations
|
|
|
|
from yuxi.services import langfuse_service as svc
|
|
|
|
|
|
class _FakeLangfuseClient:
|
|
def __init__(self, **kwargs):
|
|
self.kwargs = kwargs
|
|
|
|
def create_trace_id(self, *, seed: str | None = None) -> str:
|
|
return f"trace-{seed}"
|
|
|
|
def get_trace_url(self, *, trace_id: str | None = None) -> str | None:
|
|
if trace_id is None:
|
|
return None
|
|
return f"https://langfuse.local/trace/{trace_id}"
|
|
|
|
def flush(self) -> None:
|
|
return None
|
|
|
|
|
|
class _FakeCallbackHandler:
|
|
def __init__(self, *, public_key=None, trace_context=None):
|
|
self.public_key = public_key
|
|
self.trace_context = trace_context
|
|
self.last_trace_id = None
|
|
|
|
|
|
def test_build_run_context_includes_trace_metadata(monkeypatch):
|
|
monkeypatch.setenv("LANGFUSE_PUBLIC_KEY", "pk-test")
|
|
monkeypatch.setenv("LANGFUSE_SECRET_KEY", "sk-test")
|
|
monkeypatch.setenv("LANGFUSE_BASE_URL", "https://cloud.langfuse.example")
|
|
monkeypatch.delenv("LANGFUSE_ENABLED", raising=False)
|
|
monkeypatch.setattr(svc, "Langfuse", _FakeLangfuseClient)
|
|
monkeypatch.setattr(svc, "CallbackHandler", _FakeCallbackHandler)
|
|
svc.get_langfuse_client.cache_clear()
|
|
|
|
run_context = svc.build_run_context(
|
|
user_id="user-1",
|
|
thread_id="thread-1",
|
|
agent_id="agent-a",
|
|
request_id="req-1",
|
|
operation="agent_chat_stream",
|
|
agent_config_id=42,
|
|
message_type="text",
|
|
username="alice",
|
|
login_user_id="alice-login",
|
|
department_id=7,
|
|
)
|
|
|
|
assert run_context.trace_id == "trace-req-1"
|
|
assert len(run_context.callbacks) == 1
|
|
assert run_context.callbacks[0].trace_context == {"trace_id": "trace-req-1"}
|
|
assert run_context.metadata["langfuse_user_id"] == "user-1"
|
|
assert run_context.metadata["langfuse_session_id"] == "thread-1"
|
|
assert run_context.metadata["agent_config_id"] == "42"
|
|
assert run_context.metadata["department_id"] == "7"
|
|
assert run_context.tags == [
|
|
"yuxi",
|
|
"chat",
|
|
"agent_chat_stream",
|
|
"agent:agent-a",
|
|
"message_type:text",
|
|
]
|
|
|
|
|
|
def test_get_trace_info_prefers_handler_last_trace_id(monkeypatch):
|
|
monkeypatch.setenv("LANGFUSE_PUBLIC_KEY", "pk-test")
|
|
monkeypatch.setenv("LANGFUSE_SECRET_KEY", "sk-test")
|
|
monkeypatch.setattr(svc, "Langfuse", _FakeLangfuseClient)
|
|
monkeypatch.setattr(svc, "CallbackHandler", _FakeCallbackHandler)
|
|
svc.get_langfuse_client.cache_clear()
|
|
|
|
run_context = svc.build_run_context(
|
|
user_id="user-1",
|
|
thread_id="thread-1",
|
|
agent_id="agent-a",
|
|
request_id="req-1",
|
|
operation="agent_chat_stream",
|
|
)
|
|
run_context.callbacks[0].last_trace_id = "trace-runtime"
|
|
|
|
trace_info = svc.get_trace_info(run_context)
|
|
|
|
assert trace_info == {
|
|
"langfuse_trace_id": "trace-runtime",
|
|
"langfuse_user_id": "user-1",
|
|
"langfuse_session_id": "thread-1",
|
|
}
|
|
|
|
|
|
async def test_get_trace_url_async_returns_trace_url(monkeypatch):
|
|
monkeypatch.setenv("LANGFUSE_PUBLIC_KEY", "pk-test")
|
|
monkeypatch.setenv("LANGFUSE_SECRET_KEY", "sk-test")
|
|
monkeypatch.setattr(svc, "Langfuse", _FakeLangfuseClient)
|
|
monkeypatch.setattr(svc, "CallbackHandler", _FakeCallbackHandler)
|
|
svc.get_langfuse_client.cache_clear()
|
|
|
|
run_context = svc.build_run_context(
|
|
user_id="user-1",
|
|
thread_id="thread-1",
|
|
agent_id="agent-a",
|
|
request_id="req-1",
|
|
operation="agent_chat_stream",
|
|
)
|
|
run_context.callbacks[0].last_trace_id = "trace-runtime"
|
|
|
|
trace_url = await svc.get_trace_url_async(run_context)
|
|
|
|
assert trace_url == "https://langfuse.local/trace/trace-runtime"
|