bendtherules/ask262
1
0
Fork 0
mirror of https://github.com/bendtherules/ask262.git synced 2026-08-18 21:31:46 +00:00
Code Issues 1 Packages Projects Releases Wiki Activity

feat(langfuse): add tool-level spans with input/output and trace-level I/O

Browse Source 
Each MCP tool handler now sets:
- LANGFUSE_TRACE_OUTPUT_ATTR on tool spans (maps to top-level trace output)
- LANGFUSE_OBSERVATION_INPUT_ATTR and LANGFUSE_OBSERVATION_OUTPUT_ATTR
  for detailed child observation views

HTTP root span renamed to 'mcp_http_request' for consistency with stdio.
All attribute keys use constants from langfuse-transport.ts.
This commit is contained in:
bendtherules
2026-04-27 18:56:50 +05:30
parent 54bc20cb7f
commit d71726205d
3 changed files with 162 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mcp-server-stdio.ts
+62 -4
View File
@@ -37,6 +37,15 @@ import {
import { STORAGE_DIR as STORAGE_DIR_REL } from "./constants.js";
import { createEmbeddings } from "./lib/embeddings-factory.js";
import { LogOperation, logger } from "./lib/logger.js";
import { trace } from "@opentelemetry/api";
import {
LANGFUSE_OBSERVATION_INPUT_ATTR,
LANGFUSE_OBSERVATION_OUTPUT_ATTR,
LANGFUSE_TRACE_INPUT_ATTR,
LANGFUSE_TRACE_NAME_ATTR,
LANGFUSE_TRACE_OUTPUT_ATTR,
TRACE_NAME_STDIO,
} from "./lib/langfuse-transport.js";
import {
createProcessScopedTrace,
getSessionMetadata,
@@ -149,12 +158,29 @@ export async function main() {
sessionTraceId,
"ask262_search_spec_sections",
{
"langfuse.observation.input": JSON.stringify({ query }),
LANGFUSE_TRACE_NAME_ATTR: TRACE_NAME_STDIO,
LANGFUSE_TRACE_INPUT_ATTR: JSON.stringify({
tool: searchSpecToolName,
input: { query },
}),
LANGFUSE_OBSERVATION_INPUT_ATTR: JSON.stringify({ query }),
tool: searchSpecToolName,
query,
},
async () => {
const result = await searchSpecTool({ query });
trace
.getActiveSpan()
?.setAttribute(
LANGFUSE_OBSERVATION_OUTPUT_ATTR,
JSON.stringify(result),
);
trace
.getActiveSpan()
?.setAttribute(
LANGFUSE_TRACE_OUTPUT_ATTR,
JSON.stringify(result),
);
return {
content: [{ type: "text", text: JSON.stringify(result, null, 2) }],
structuredContent: result,
@@ -187,7 +213,12 @@ export async function main() {
sessionTraceId,
"ask262_get_section_content",
{
"langfuse.observation.input": JSON.stringify({
LANGFUSE_TRACE_NAME_ATTR: TRACE_NAME_STDIO,
LANGFUSE_TRACE_INPUT_ATTR: JSON.stringify({
tool: sectionContentToolName,
input: { sectionIds, recursive },
}),
LANGFUSE_OBSERVATION_INPUT_ATTR: JSON.stringify({
sectionIds,
recursive,
}),
@@ -196,6 +227,18 @@ export async function main() {
},
async () => {
const result = await getSectionContentTool({ sectionIds, recursive });
trace
.getActiveSpan()
?.setAttribute(
LANGFUSE_OBSERVATION_OUTPUT_ATTR,
JSON.stringify(result),
);
trace
.getActiveSpan()
?.setAttribute(
LANGFUSE_TRACE_OUTPUT_ATTR,
JSON.stringify(result),
);
return {
content: [{ type: "text", text: JSON.stringify(result, null, 2) }],
structuredContent: result,
@@ -225,14 +268,29 @@ export async function main() {
sessionTraceId,
"ask262_evaluate_in_engine262",
{
"langfuse.observation.input": JSON.stringify({
code: code.slice(0, 200),
LANGFUSE_TRACE_NAME_ATTR: TRACE_NAME_STDIO,
LANGFUSE_TRACE_INPUT_ATTR: JSON.stringify({
tool: evaluateToolName,
input: { code },
}),
LANGFUSE_OBSERVATION_INPUT_ATTR: JSON.stringify({ code }),
tool: evaluateToolName,
code_length: code.length,
},
async () => {
const result = await evaluateTool({ code });
trace
.getActiveSpan()
?.setAttribute(
LANGFUSE_OBSERVATION_OUTPUT_ATTR,
JSON.stringify(result),
);
trace
.getActiveSpan()
?.setAttribute(
LANGFUSE_TRACE_OUTPUT_ATTR,
JSON.stringify(result),
);
const isError = result.error !== undefined;
const text = isError ? result.error : JSON.stringify(result, null, 2);
return {