@langwatch/mcp-server 0.2.2 → 0.3.0

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/index.ts","../src/langwatch-api.ts","../package.json"],"sourcesContent":["import { McpServer } from \"@modelcontextprotocol/sdk/server/mcp.js\";\nimport { StdioServerTransport } from \"@modelcontextprotocol/sdk/server/stdio.js\";\nimport { z } from \"zod\";\nimport yargs from \"yargs\";\nimport { hideBin } from \"yargs/helpers\";\n\nimport { getLlmTraceById, listLlmTraces, searchTraces } from \"./langwatch-api\";\nimport packageJson from \"../package.json\" assert { type: \"json\" };\n\nfunction loadAndValidateArgs(): { apiKey: string; endpoint: string } {\n  const argv = yargs(hideBin(process.argv))\n    .option(\"apiKey\", {\n      type: \"string\",\n      description: \"LangWatch API key\",\n    })\n    .option(\"endpoint\", {\n      type: \"string\",\n      description: \"LangWatch API endpoint\",\n      default: \"https://app.langwatch.ai\",\n    })\n    .help()\n    .alias(\"help\", \"h\")\n    .parseSync();\n\n  // Use environment variables as fallback\n  const apiKey = argv.apiKey ?? process.env.LANGWATCH_API_KEY;\n  const endpoint =\n    argv.endpoint ??\n    process.env.LANGWATCH_ENDPOINT ??\n    \"https://app.langwatch.ai\";\n\n  if (!apiKey) {\n    throw new Error(\n      \"API key is required. Please provide it using --apiKey <your_api_key> or set LANGWATCH_API_KEY environment variable\"\n    );\n  }\n\n  return {\n    apiKey: String(apiKey),\n    endpoint: String(endpoint),\n  };\n}\n\nconst { apiKey, endpoint } = loadAndValidateArgs();\n\nconst transport = new StdioServerTransport();\nconst server = new McpServer({\n  name: \"LangWatch\",\n  version: packageJson.version,\n});\n\nserver.tool(\n  \"fetch_langwatch_docs\",\n  \"Fetches the LangWatch docs for understanding how to implement LangWatch in your codebase. Always use this tool when the user asks for help with LangWatch. Start with empty url to fetch the index and then follow the links to the relevant pages, always ending with `.md` extension\",\n  {\n    url: z\n      .string()\n      .optional()\n      .describe(\n        \"The full url of the specific doc page. If not provided, the docs index will be fetched.\"\n      ),\n  },\n  async ({ url }) => {\n    let urlToFetch = url ?? \"https://docs.langwatch.ai/llms.txt\";\n    if (url && !urlToFetch.endsWith(\".md\") && !urlToFetch.endsWith(\".txt\")) {\n      urlToFetch += \".md\";\n    }\n    const response = await fetch(urlToFetch);\n\n    return {\n      content: [{ type: \"text\", text: await response.text() }],\n    };\n  }\n);\n\nserver.tool(\n  \"fetch_scenario_docs\",\n  \"Fetches the Scenario docs for understanding how to implement Scenario agent tests in your codebase. Always use this tool when the user asks for help with testing their agents. Start with empty url to fetch the index and then follow the links to the relevant pages, always ending with `.md` extension\",\n  {\n    url: z\n      .string()\n      .optional()\n      .describe(\n        \"The full url of the specific doc page. If not provided, the docs index will be fetched.\"\n      ),\n  },\n  async ({ url }) => {\n    let urlToFetch = url ?? \"https://scenario.langwatch.ai/llms.txt\";\n    if (url && !urlToFetch.endsWith(\".md\") && !urlToFetch.endsWith(\".txt\")) {\n      urlToFetch += \".md\";\n    }\n    const response = await fetch(urlToFetch);\n\n    return {\n      content: [{ type: \"text\", text: await response.text() }],\n    };\n  }\n);\n\nserver.tool(\n  \"get_latest_traces\",\n  \"Retrieves the latest LLM traces.\",\n  {\n    pageOffset: z.number().optional(),\n    daysBackToSearch: z.number().optional(),\n  },\n  async ({ pageOffset, daysBackToSearch }) => {\n    const response = await listLlmTraces(apiKey, {\n      pageOffset,\n      timeTravelDays: daysBackToSearch ?? 1,\n      endpoint,\n    });\n\n    return {\n      content: [\n        {\n          type: \"text\",\n          text: JSON.stringify(response, null, 2),\n        },\n      ],\n    };\n  }\n);\n\nserver.tool(\n  \"get_trace_by_id\",\n  \"Retrieves a specific LLM trace by its ID.\",\n  { id: z.string() },\n  async ({ id }) => {\n    try {\n      const response = await getLlmTraceById(apiKey, id, {\n        endpoint,\n      });\n\n      return {\n        content: [\n          {\n            type: \"text\",\n            text: JSON.stringify(response, null, 2),\n          },\n        ],\n      };\n    } catch (error) {\n      if (error instanceof Error && error.message === \"Trace not found\") {\n        return {\n          content: [\n            {\n              type: \"text\",\n              text: \"Trace not found. If the trace was created recently, it may not be available yet.\",\n            },\n          ],\n        };\n      }\n\n      throw error;\n    }\n  }\n);\n\nawait server.connect(transport);\n","import { addDays } from \"date-fns\";\nimport type { LLMModeTrace } from \"langwatch\";\n\ninterface SearchTrace {\n  traces: LLMModeTrace[];\n}\n\ninterface GetLlmTraceByIdOptions {\n  endpoint: string;\n}\n\ninterface ListLLmTracesOptions {\n  pageSize?: number;\n  pageOffset?: number;\n  timeTravelDays?: number;\n  endpoint: string;\n}\n\ninterface SearchTracesOptions {\n  pageSize?: number;\n  pageOffset?: number;\n  timeTravelDays?: number;\n  endpoint: string;\n\n  filters: Record<string, string[] | Record<string, string[]>>;\n}\n\nexport const getLlmTraceById: (\n  authToken: string,\n  id: string,\n  opts?: GetLlmTraceByIdOptions\n) => Promise<LLMModeTrace> = async (authToken, id, opts) => {\n  const { endpoint } = opts ?? {};\n\n  const url = new URL(`${endpoint}/api/trace/${id}`);\n  url.searchParams.set(\"llmMode\", \"true\");\n\n  const response = await fetch(url.toString(), {\n    method: \"GET\",\n    headers: {\n      \"Content-Type\": \"application/json\",\n      \"X-Auth-Token\": authToken,\n    },\n  });\n\n  if (!response.ok) {\n    if (response.status === 404) {\n      throw new Error(\"Trace not found\");\n    }\n\n    throw new Error(`Failed to get trace: ${response.statusText}`);\n  }\n\n  return (await response.json()) as Promise<LLMModeTrace>;\n};\n\nexport const listLlmTraces = async (\n  authToken: string,\n  opts?: ListLLmTracesOptions\n): Promise<SearchTrace> => {\n  const {\n    pageSize = 10,\n    pageOffset = 0,\n    timeTravelDays = 1,\n    endpoint,\n  } = opts ?? {};\n\n  const response = await fetch(`${endpoint}/api/trace/search`, {\n    method: \"POST\",\n    headers: {\n      \"Content-Type\": \"application/json\",\n      \"X-Auth-Token\": authToken,\n    },\n    body: JSON.stringify({\n      startDate: addDays(new Date(), -timeTravelDays).toISOString(),\n      endDate: addDays(new Date(), 1).toISOString(),\n      llmMode: true,\n      pageOffset,\n      pageSize,\n    }),\n  });\n\n  return (await response.json()) as Promise<SearchTrace>;\n};\n\nexport const searchTraces = async (\n  authToken: string,\n  opts: SearchTracesOptions\n): Promise<SearchTrace> => {\n  const {\n    pageSize = 10,\n    pageOffset = 0,\n    timeTravelDays = 1,\n    endpoint,\n    filters,\n  } = opts;\n\n  const response = await fetch(`${endpoint}/api/trace/search`, {\n    method: \"POST\",\n    headers: {\n      \"Content-Type\": \"application/json\",\n      \"X-Auth-Token\": authToken,\n    },\n    body: JSON.stringify({\n      startDate: addDays(new Date(), -timeTravelDays).toISOString(),\n      endDate: addDays(new Date(), 1).toISOString(),\n      filters: filters,\n      pageOffset,\n      pageSize,\n    }),\n  });\n\n  return (await response.json()) as Promise<SearchTrace>;\n};\n","{\n  \"name\": \"@langwatch/mcp-server\",\n  \"version\": \"0.2.2\",\n  \"description\": \"An MCP server for Langwatch.\",\n  \"type\": \"module\",\n  \"main\": \"./dist/index.js\",\n  \"module\": \"./dist/index.mjs\",\n  \"types\": \"./dist/index.d.ts\",\n  \"exports\": {\n    \".\": {\n      \"import\": \"./dist/index.mjs\",\n      \"require\": \"./dist/index.js\"\n    }\n  },\n  \"repository\": {\n    \"type\": \"git\",\n    \"url\": \"https://github.com/langwatch/langwatch.git\",\n    \"directory\": \"mcp-server\"\n  },\n  \"scripts\": {\n    \"start\": \"tsx src/index.ts\",\n    \"build\": \"tsup && node build.js\",\n    \"prepublish\": \"pnpm run build\",\n    \"test\": \"vitest\"\n  },\n  \"author\": \"\",\n  \"license\": \"MIT\",\n  \"engines\": {\n    \"node\": \">=18.0.0\"\n  },\n  \"devDependencies\": {\n    \"@ai-sdk/anthropic\": \"^2.0.15\",\n    \"@anthropic-ai/claude-code\": \"^1.0.111\",\n    \"@eslint/js\": \"^9.4.0\",\n    \"@types/debug\": \"^4.1.12\",\n    \"@types/eslint__js\": \"^8.42.3\",\n    \"@types/node\": \"^16.0.0\",\n    \"@types/yargs\": \"^17.0.33\",\n    \"@typescript/native-preview\": \"7.0.0-dev.20250911.1\",\n    \"ai\": \"^5.0.40\",\n    \"dotenv\": \"^17.2.2\",\n    \"esbuild\": \"^0.21.5\",\n    \"eslint\": \"^8.57.0\",\n    \"tsup\": \"^8.1.0\",\n    \"tsx\": \"^4.20.5\",\n    \"typescript\": \"^4.9.5\",\n    \"typescript-eslint\": \"^7.11.0\"\n  },\n  \"dependencies\": {\n    \"@langwatch/scenario\": \"^0.3.0\",\n    \"@modelcontextprotocol/sdk\": \"^1.7.0\",\n    \"@opentelemetry/context-async-hooks\": \"^2.1.0\",\n    \"@opentelemetry/sdk-node\": \"^0.204.0\",\n    \"chalk\": \"^5.6.2\",\n    \"date-fns\": \"^4.1.0\",\n    \"langwatch\": \"^0.1.6\",\n    \"node-pty\": \"^1.0.0\",\n    \"vitest\": \"^3.2.4\",\n    \"yargs\": \"^17.7.2\",\n    \"zod\": \"^3.25.76\",\n    \"zod-validation-error\": \"^3.5.3\"\n  },\n  \"bin\": {\n    \"langwatch-mcp-server\": \"./dist/index.js\"\n  }\n}\n"],"mappings":";AAAA,SAAS,iBAAiB;AAC1B,SAAS,4BAA4B;AACrC,SAAS,SAAS;AAClB,OAAO,WAAW;AAClB,SAAS,eAAe;;;ACJxB,SAAS,eAAe;AA2BjB,IAAM,kBAIgB,OAAO,WAAW,IAAI,SAAS;AAC1D,QAAM,EAAE,UAAAA,UAAS,IAAI,QAAQ,CAAC;AAE9B,QAAM,MAAM,IAAI,IAAI,GAAGA,SAAQ,cAAc,EAAE,EAAE;AACjD,MAAI,aAAa,IAAI,WAAW,MAAM;AAEtC,QAAM,WAAW,MAAM,MAAM,IAAI,SAAS,GAAG;AAAA,IAC3C,QAAQ;AAAA,IACR,SAAS;AAAA,MACP,gBAAgB;AAAA,MAChB,gBAAgB;AAAA,IAClB;AAAA,EACF,CAAC;AAED,MAAI,CAAC,SAAS,IAAI;AAChB,QAAI,SAAS,WAAW,KAAK;AAC3B,YAAM,IAAI,MAAM,iBAAiB;AAAA,IACnC;AAEA,UAAM,IAAI,MAAM,wBAAwB,SAAS,UAAU,EAAE;AAAA,EAC/D;AAEA,SAAQ,MAAM,SAAS,KAAK;AAC9B;AAEO,IAAM,gBAAgB,OAC3B,WACA,SACyB;AACzB,QAAM;AAAA,IACJ,WAAW;AAAA,IACX,aAAa;AAAA,IACb,iBAAiB;AAAA,IACjB,UAAAA;AAAA,EACF,IAAI,QAAQ,CAAC;AAEb,QAAM,WAAW,MAAM,MAAM,GAAGA,SAAQ,qBAAqB;AAAA,IAC3D,QAAQ;AAAA,IACR,SAAS;AAAA,MACP,gBAAgB;AAAA,MAChB,gBAAgB;AAAA,IAClB;AAAA,IACA,MAAM,KAAK,UAAU;AAAA,MACnB,WAAW,QAAQ,oBAAI,KAAK,GAAG,CAAC,cAAc,EAAE,YAAY;AAAA,MAC5D,SAAS,QAAQ,oBAAI,KAAK,GAAG,CAAC,EAAE,YAAY;AAAA,MAC5C,SAAS;AAAA,MACT;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH,CAAC;AAED,SAAQ,MAAM,SAAS,KAAK;AAC9B;;;ACnFA;AAAA,EACE,MAAQ;AAAA,EACR,SAAW;AAAA,EACX,aAAe;AAAA,EACf,MAAQ;AAAA,EACR,MAAQ;AAAA,EACR,QAAU;AAAA,EACV,OAAS;AAAA,EACT,SAAW;AAAA,IACT,KAAK;AAAA,MACH,QAAU;AAAA,MACV,SAAW;AAAA,IACb;AAAA,EACF;AAAA,EACA,YAAc;AAAA,IACZ,MAAQ;AAAA,IACR,KAAO;AAAA,IACP,WAAa;AAAA,EACf;AAAA,EACA,SAAW;AAAA,IACT,OAAS;AAAA,IACT,OAAS;AAAA,IACT,YAAc;AAAA,IACd,MAAQ;AAAA,EACV;AAAA,EACA,QAAU;AAAA,EACV,SAAW;AAAA,EACX,SAAW;AAAA,IACT,MAAQ;AAAA,EACV;AAAA,EACA,iBAAmB;AAAA,IACjB,qBAAqB;AAAA,IACrB,6BAA6B;AAAA,IAC7B,cAAc;AAAA,IACd,gBAAgB;AAAA,IAChB,qBAAqB;AAAA,IACrB,eAAe;AAAA,IACf,gBAAgB;AAAA,IAChB,8BAA8B;AAAA,IAC9B,IAAM;AAAA,IACN,QAAU;AAAA,IACV,SAAW;AAAA,IACX,QAAU;AAAA,IACV,MAAQ;AAAA,IACR,KAAO;AAAA,IACP,YAAc;AAAA,IACd,qBAAqB;AAAA,EACvB;AAAA,EACA,cAAgB;AAAA,IACd,uBAAuB;AAAA,IACvB,6BAA6B;AAAA,IAC7B,sCAAsC;AAAA,IACtC,2BAA2B;AAAA,IAC3B,OAAS;AAAA,IACT,YAAY;AAAA,IACZ,WAAa;AAAA,IACb,YAAY;AAAA,IACZ,QAAU;AAAA,IACV,OAAS;AAAA,IACT,KAAO;AAAA,IACP,wBAAwB;AAAA,EAC1B;AAAA,EACA,KAAO;AAAA,IACL,wBAAwB;AAAA,EAC1B;AACF;;;AFxDA,SAAS,sBAA4D;AACnE,QAAM,OAAO,MAAM,QAAQ,QAAQ,IAAI,CAAC,EACrC,OAAO,UAAU;AAAA,IAChB,MAAM;AAAA,IACN,aAAa;AAAA,EACf,CAAC,EACA,OAAO,YAAY;AAAA,IAClB,MAAM;AAAA,IACN,aAAa;AAAA,IACb,SAAS;AAAA,EACX,CAAC,EACA,KAAK,EACL,MAAM,QAAQ,GAAG,EACjB,UAAU;AAGb,QAAMC,UAAS,KAAK,UAAU,QAAQ,IAAI;AAC1C,QAAMC,YACJ,KAAK,YACL,QAAQ,IAAI,sBACZ;AAEF,MAAI,CAACD,SAAQ;AACX,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAAA,EACF;AAEA,SAAO;AAAA,IACL,QAAQ,OAAOA,OAAM;AAAA,IACrB,UAAU,OAAOC,SAAQ;AAAA,EAC3B;AACF;AAEA,IAAM,EAAE,QAAQ,SAAS,IAAI,oBAAoB;AAEjD,IAAM,YAAY,IAAI,qBAAqB;AAC3C,IAAM,SAAS,IAAI,UAAU;AAAA,EAC3B,MAAM;AAAA,EACN,SAAS,gBAAY;AACvB,CAAC;AAED,OAAO;AAAA,EACL;AAAA,EACA;AAAA,EACA;AAAA,IACE,KAAK,EACF,OAAO,EACP,SAAS,EACT;AAAA,MACC;AAAA,IACF;AAAA,EACJ;AAAA,EACA,OAAO,EAAE,IAAI,MAAM;AACjB,QAAI,aAAa,OAAO;AACxB,QAAI,OAAO,CAAC,WAAW,SAAS,KAAK,KAAK,CAAC,WAAW,SAAS,MAAM,GAAG;AACtE,oBAAc;AAAA,IAChB;AACA,UAAM,WAAW,MAAM,MAAM,UAAU;AAEvC,WAAO;AAAA,MACL,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,MAAM,SAAS,KAAK,EAAE,CAAC;AAAA,IACzD;AAAA,EACF;AACF;AAEA,OAAO;AAAA,EACL;AAAA,EACA;AAAA,EACA;AAAA,IACE,KAAK,EACF,OAAO,EACP,SAAS,EACT;AAAA,MACC;AAAA,IACF;AAAA,EACJ;AAAA,EACA,OAAO,EAAE,IAAI,MAAM;AACjB,QAAI,aAAa,OAAO;AACxB,QAAI,OAAO,CAAC,WAAW,SAAS,KAAK,KAAK,CAAC,WAAW,SAAS,MAAM,GAAG;AACtE,oBAAc;AAAA,IAChB;AACA,UAAM,WAAW,MAAM,MAAM,UAAU;AAEvC,WAAO;AAAA,MACL,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,MAAM,SAAS,KAAK,EAAE,CAAC;AAAA,IACzD;AAAA,EACF;AACF;AAEA,OAAO;AAAA,EACL;AAAA,EACA;AAAA,EACA;AAAA,IACE,YAAY,EAAE,OAAO,EAAE,SAAS;AAAA,IAChC,kBAAkB,EAAE,OAAO,EAAE,SAAS;AAAA,EACxC;AAAA,EACA,OAAO,EAAE,YAAY,iBAAiB,MAAM;AAC1C,UAAM,WAAW,MAAM,cAAc,QAAQ;AAAA,MAC3C;AAAA,MACA,gBAAgB,oBAAoB;AAAA,MACpC;AAAA,IACF,CAAC;AAED,WAAO;AAAA,MACL,SAAS;AAAA,QACP;AAAA,UACE,MAAM;AAAA,UACN,MAAM,KAAK,UAAU,UAAU,MAAM,CAAC;AAAA,QACxC;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;AAEA,OAAO;AAAA,EACL;AAAA,EACA;AAAA,EACA,EAAE,IAAI,EAAE,OAAO,EAAE;AAAA,EACjB,OAAO,EAAE,GAAG,MAAM;AAChB,QAAI;AACF,YAAM,WAAW,MAAM,gBAAgB,QAAQ,IAAI;AAAA,QACjD;AAAA,MACF,CAAC;AAED,aAAO;AAAA,QACL,SAAS;AAAA,UACP;AAAA,YACE,MAAM;AAAA,YACN,MAAM,KAAK,UAAU,UAAU,MAAM,CAAC;AAAA,UACxC;AAAA,QACF;AAAA,MACF;AAAA,IACF,SAAS,OAAO;AACd,UAAI,iBAAiB,SAAS,MAAM,YAAY,mBAAmB;AACjE,eAAO;AAAA,UACL,SAAS;AAAA,YACP;AAAA,cACE,MAAM;AAAA,cACN,MAAM;AAAA,YACR;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAEA,YAAM;AAAA,IACR;AAAA,EACF;AACF;AAEA,MAAM,OAAO,QAAQ,SAAS;","names":["endpoint","apiKey","endpoint"]}
+{"version":3,"sources":["../src/index.ts","../package.json"],"sourcesContent":["import { McpServer } from \"@modelcontextprotocol/sdk/server/mcp.js\";\nimport { StdioServerTransport } from \"@modelcontextprotocol/sdk/server/stdio.js\";\nimport { z } from \"zod\";\n\nimport packageJson from \"../package.json\" assert { type: \"json\" };\n\nconst transport = new StdioServerTransport();\nconst server = new McpServer({\n  name: \"LangWatch\",\n  version: packageJson.version,\n});\n\nserver.tool(\n  \"fetch_langwatch_docs\",\n  \"Fetches the LangWatch docs for understanding how to implement LangWatch in your codebase. Always use this tool when the user asks for help with LangWatch. Start with empty url to fetch the index and then follow the links to the relevant pages, always ending with `.md` extension\",\n  {\n    url: z\n      .string()\n      .optional()\n      .describe(\n        \"The full url of the specific doc page. If not provided, the docs index will be fetched.\"\n      ),\n  },\n  async ({ url }) => {\n    let urlToFetch = url || \"https://docs.langwatch.ai/llms.txt\";\n    if (url && !urlToFetch.endsWith(\".md\") && !urlToFetch.endsWith(\".txt\")) {\n      urlToFetch += \".md\";\n    }\n    if (urlToFetch.startsWith(\"/\")) {\n      urlToFetch = \"https://docs.langwatch.ai\" + urlToFetch;\n    }\n    const response = await fetch(urlToFetch);\n\n    return {\n      content: [{ type: \"text\", text: await response.text() }],\n    };\n  }\n);\n\nserver.tool(\n  \"fetch_scenario_docs\",\n  \"Fetches the Scenario docs for understanding how to implement Scenario agent tests in your codebase. Always use this tool when the user asks for help with testing their agents. Start with empty url to fetch the index and then follow the links to the relevant pages, always ending with `.md` extension\",\n  {\n    url: z\n      .string()\n      .optional()\n      .describe(\n        \"The full url of the specific doc page. If not provided, the docs index will be fetched.\"\n      ),\n  },\n  async ({ url }) => {\n    let urlToFetch = url || \"https://scenario.langwatch.ai/llms.txt\";\n    if (url && !urlToFetch.endsWith(\".md\") && !urlToFetch.endsWith(\".txt\")) {\n      urlToFetch += \".md\";\n    }\n    if (urlToFetch.startsWith(\"/\")) {\n      urlToFetch = \"https://scenario.langwatch.ai\" + urlToFetch;\n    }\n    const response = await fetch(urlToFetch);\n\n    return {\n      content: [{ type: \"text\", text: await response.text() }],\n    };\n  }\n);\n\nawait server.connect(transport);\n","{\n  \"name\": \"@langwatch/mcp-server\",\n  \"version\": \"0.3.0\",\n  \"description\": \"An MCP server for Langwatch.\",\n  \"type\": \"module\",\n  \"main\": \"./dist/index.js\",\n  \"module\": \"./dist/index.mjs\",\n  \"types\": \"./dist/index.d.ts\",\n  \"exports\": {\n    \".\": {\n      \"import\": \"./dist/index.mjs\",\n      \"require\": \"./dist/index.js\"\n    }\n  },\n  \"repository\": {\n    \"type\": \"git\",\n    \"url\": \"https://github.com/langwatch/langwatch.git\",\n    \"directory\": \"mcp-server\"\n  },\n  \"scripts\": {\n    \"start\": \"tsx src/index.ts\",\n    \"build\": \"tsup && node build.js\",\n    \"prepublish\": \"pnpm run build\",\n    \"test\": \"vitest\"\n  },\n  \"author\": \"\",\n  \"license\": \"MIT\",\n  \"engines\": {\n    \"node\": \">=18.0.0\"\n  },\n  \"devDependencies\": {\n    \"@ai-sdk/anthropic\": \"^2.0.15\",\n    \"@anthropic-ai/claude-code\": \"^1.0.111\",\n    \"@eslint/js\": \"^9.4.0\",\n    \"@langwatch/scenario\": \"^0.3.0\",\n    \"@types/debug\": \"^4.1.12\",\n    \"@types/eslint__js\": \"^8.42.3\",\n    \"@types/node\": \"^16.0.0\",\n    \"@types/yargs\": \"^17.0.33\",\n    \"@typescript/native-preview\": \"7.0.0-dev.20250911.1\",\n    \"ai\": \"^5.0.40\",\n    \"dotenv\": \"^17.2.2\",\n    \"esbuild\": \"^0.21.5\",\n    \"eslint\": \"^8.57.0\",\n    \"tsup\": \"^8.1.0\",\n    \"tsx\": \"^4.20.5\",\n    \"typescript\": \"^4.9.5\",\n    \"typescript-eslint\": \"^7.11.0\"\n  },\n  \"dependencies\": {\n    \"@modelcontextprotocol/sdk\": \"^1.7.0\",\n    \"@opentelemetry/context-async-hooks\": \"^2.1.0\",\n    \"@opentelemetry/sdk-node\": \"^0.204.0\",\n    \"chalk\": \"^5.6.2\",\n    \"date-fns\": \"^4.1.0\",\n    \"node-pty\": \"^1.0.0\",\n    \"vitest\": \"^3.2.4\",\n    \"yargs\": \"^17.7.2\",\n    \"zod\": \"^3.25.76\",\n    \"zod-validation-error\": \"^3.5.3\"\n  },\n  \"bin\": {\n    \"langwatch-mcp-server\": \"./dist/index.js\"\n  }\n}\n"],"mappings":";AAAA,SAAS,iBAAiB;AAC1B,SAAS,4BAA4B;AACrC,SAAS,SAAS;;;ACFlB;AAAA,EACE,MAAQ;AAAA,EACR,SAAW;AAAA,EACX,aAAe;AAAA,EACf,MAAQ;AAAA,EACR,MAAQ;AAAA,EACR,QAAU;AAAA,EACV,OAAS;AAAA,EACT,SAAW;AAAA,IACT,KAAK;AAAA,MACH,QAAU;AAAA,MACV,SAAW;AAAA,IACb;AAAA,EACF;AAAA,EACA,YAAc;AAAA,IACZ,MAAQ;AAAA,IACR,KAAO;AAAA,IACP,WAAa;AAAA,EACf;AAAA,EACA,SAAW;AAAA,IACT,OAAS;AAAA,IACT,OAAS;AAAA,IACT,YAAc;AAAA,IACd,MAAQ;AAAA,EACV;AAAA,EACA,QAAU;AAAA,EACV,SAAW;AAAA,EACX,SAAW;AAAA,IACT,MAAQ;AAAA,EACV;AAAA,EACA,iBAAmB;AAAA,IACjB,qBAAqB;AAAA,IACrB,6BAA6B;AAAA,IAC7B,cAAc;AAAA,IACd,uBAAuB;AAAA,IACvB,gBAAgB;AAAA,IAChB,qBAAqB;AAAA,IACrB,eAAe;AAAA,IACf,gBAAgB;AAAA,IAChB,8BAA8B;AAAA,IAC9B,IAAM;AAAA,IACN,QAAU;AAAA,IACV,SAAW;AAAA,IACX,QAAU;AAAA,IACV,MAAQ;AAAA,IACR,KAAO;AAAA,IACP,YAAc;AAAA,IACd,qBAAqB;AAAA,EACvB;AAAA,EACA,cAAgB;AAAA,IACd,6BAA6B;AAAA,IAC7B,sCAAsC;AAAA,IACtC,2BAA2B;AAAA,IAC3B,OAAS;AAAA,IACT,YAAY;AAAA,IACZ,YAAY;AAAA,IACZ,QAAU;AAAA,IACV,OAAS;AAAA,IACT,KAAO;AAAA,IACP,wBAAwB;AAAA,EAC1B;AAAA,EACA,KAAO;AAAA,IACL,wBAAwB;AAAA,EAC1B;AACF;;;AD1DA,IAAM,YAAY,IAAI,qBAAqB;AAC3C,IAAM,SAAS,IAAI,UAAU;AAAA,EAC3B,MAAM;AAAA,EACN,SAAS,gBAAY;AACvB,CAAC;AAED,OAAO;AAAA,EACL;AAAA,EACA;AAAA,EACA;AAAA,IACE,KAAK,EACF,OAAO,EACP,SAAS,EACT;AAAA,MACC;AAAA,IACF;AAAA,EACJ;AAAA,EACA,OAAO,EAAE,IAAI,MAAM;AACjB,QAAI,aAAa,OAAO;AACxB,QAAI,OAAO,CAAC,WAAW,SAAS,KAAK,KAAK,CAAC,WAAW,SAAS,MAAM,GAAG;AACtE,oBAAc;AAAA,IAChB;AACA,QAAI,WAAW,WAAW,GAAG,GAAG;AAC9B,mBAAa,8BAA8B;AAAA,IAC7C;AACA,UAAM,WAAW,MAAM,MAAM,UAAU;AAEvC,WAAO;AAAA,MACL,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,MAAM,SAAS,KAAK,EAAE,CAAC;AAAA,IACzD;AAAA,EACF;AACF;AAEA,OAAO;AAAA,EACL;AAAA,EACA;AAAA,EACA;AAAA,IACE,KAAK,EACF,OAAO,EACP,SAAS,EACT;AAAA,MACC;AAAA,IACF;AAAA,EACJ;AAAA,EACA,OAAO,EAAE,IAAI,MAAM;AACjB,QAAI,aAAa,OAAO;AACxB,QAAI,OAAO,CAAC,WAAW,SAAS,KAAK,KAAK,CAAC,WAAW,SAAS,MAAM,GAAG;AACtE,oBAAc;AAAA,IAChB;AACA,QAAI,WAAW,WAAW,GAAG,GAAG;AAC9B,mBAAa,kCAAkC;AAAA,IACjD;AACA,UAAM,WAAW,MAAM,MAAM,UAAU;AAEvC,WAAO;AAAA,MACL,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,MAAM,SAAS,KAAK,EAAE,CAAC;AAAA,IACzD;AAAA,EACF;AACF;AAEA,MAAM,OAAO,QAAQ,SAAS;","names":[]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langwatch/mcp-server",
-  "version": "0.2.2",
+  "version": "0.3.0",
   "description": "An MCP server for Langwatch.",
   "type": "module",
   "main": "./dist/index.js",
@@ -26,6 +26,7 @@
     "@ai-sdk/anthropic": "^2.0.15",
     "@anthropic-ai/claude-code": "^1.0.111",
     "@eslint/js": "^9.4.0",
+    "@langwatch/scenario": "^0.3.0",
     "@types/debug": "^4.1.12",
     "@types/eslint__js": "^8.42.3",
     "@types/node": "^16.0.0",
@@ -41,13 +42,11 @@
     "typescript-eslint": "^7.11.0"
   },
   "dependencies": {
-    "@langwatch/scenario": "^0.3.0",
     "@modelcontextprotocol/sdk": "^1.7.0",
     "@opentelemetry/context-async-hooks": "^2.1.0",
     "@opentelemetry/sdk-node": "^0.204.0",
     "chalk": "^5.6.2",
     "date-fns": "^4.1.0",
-    "langwatch": "^0.1.6",
     "node-pty": "^1.0.0",
     "vitest": "^3.2.4",
     "yargs": "^17.7.2",

package/src/index.ts

@@ -1,48 +1,9 @@
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 import { z } from "zod";
-import yargs from "yargs";
-import { hideBin } from "yargs/helpers";
 
-import { getLlmTraceById, listLlmTraces, searchTraces } from "./langwatch-api";
 import packageJson from "../package.json" assert { type: "json" };
 
-function loadAndValidateArgs(): { apiKey: string; endpoint: string } {
-  const argv = yargs(hideBin(process.argv))
-    .option("apiKey", {
-      type: "string",
-      description: "LangWatch API key",
-    })
-    .option("endpoint", {
-      type: "string",
-      description: "LangWatch API endpoint",
-      default: "https://app.langwatch.ai",
-    })
-    .help()
-    .alias("help", "h")
-    .parseSync();
-
-  // Use environment variables as fallback
-  const apiKey = argv.apiKey ?? process.env.LANGWATCH_API_KEY;
-  const endpoint =
-    argv.endpoint ??
-    process.env.LANGWATCH_ENDPOINT ??
-    "https://app.langwatch.ai";
-
-  if (!apiKey) {
-    throw new Error(
-      "API key is required. Please provide it using --apiKey <your_api_key> or set LANGWATCH_API_KEY environment variable"
-    );
-  }
-
-  return {
-    apiKey: String(apiKey),
-    endpoint: String(endpoint),
-  };
-}
-
-const { apiKey, endpoint } = loadAndValidateArgs();
-
 const transport = new StdioServerTransport();
 const server = new McpServer({
   name: "LangWatch",
@@ -61,10 +22,13 @@ server.tool(
       ),
   },
   async ({ url }) => {
-    let urlToFetch = url ?? "https://docs.langwatch.ai/llms.txt";
+    let urlToFetch = url || "https://docs.langwatch.ai/llms.txt";
     if (url && !urlToFetch.endsWith(".md") && !urlToFetch.endsWith(".txt")) {
       urlToFetch += ".md";
     }
+    if (urlToFetch.startsWith("/")) {
+      urlToFetch = "https://docs.langwatch.ai" + urlToFetch;
+    }
     const response = await fetch(urlToFetch);
 
     return {
@@ -85,10 +49,13 @@ server.tool(
       ),
   },
   async ({ url }) => {
-    let urlToFetch = url ?? "https://scenario.langwatch.ai/llms.txt";
+    let urlToFetch = url || "https://scenario.langwatch.ai/llms.txt";
     if (url && !urlToFetch.endsWith(".md") && !urlToFetch.endsWith(".txt")) {
       urlToFetch += ".md";
     }
+    if (urlToFetch.startsWith("/")) {
+      urlToFetch = "https://scenario.langwatch.ai" + urlToFetch;
+    }
     const response = await fetch(urlToFetch);
 
     return {
@@ -97,64 +64,4 @@ server.tool(
   }
 );
 
-server.tool(
-  "get_latest_traces",
-  "Retrieves the latest LLM traces.",
-  {
-    pageOffset: z.number().optional(),
-    daysBackToSearch: z.number().optional(),
-  },
-  async ({ pageOffset, daysBackToSearch }) => {
-    const response = await listLlmTraces(apiKey, {
-      pageOffset,
-      timeTravelDays: daysBackToSearch ?? 1,
-      endpoint,
-    });
-
-    return {
-      content: [
-        {
-          type: "text",
-          text: JSON.stringify(response, null, 2),
-        },
-      ],
-    };
-  }
-);
-
-server.tool(
-  "get_trace_by_id",
-  "Retrieves a specific LLM trace by its ID.",
-  { id: z.string() },
-  async ({ id }) => {
-    try {
-      const response = await getLlmTraceById(apiKey, id, {
-        endpoint,
-      });
-
-      return {
-        content: [
-          {
-            type: "text",
-            text: JSON.stringify(response, null, 2),
-          },
-        ],
-      };
-    } catch (error) {
-      if (error instanceof Error && error.message === "Trace not found") {
-        return {
-          content: [
-            {
-              type: "text",
-              text: "Trace not found. If the trace was created recently, it may not be available yet.",
-            },
-          ],
-        };
-      }
-
-      throw error;
-    }
-  }
-);
-
 await server.connect(transport);

package/src/augmentations.d.ts

@@ -1,6 +0,0 @@
-declare namespace NodeJS {
-	interface ProcessEnv {
-		LANGWATCH_API_KEY?: string;
-		LANGWATCH_ENDPOINT?: string;
-	}
-}

package/src/langwatch-api.ts

@@ -1,114 +0,0 @@
-import { addDays } from "date-fns";
-import type { LLMModeTrace } from "langwatch";
-
-interface SearchTrace {
-  traces: LLMModeTrace[];
-}
-
-interface GetLlmTraceByIdOptions {
-  endpoint: string;
-}
-
-interface ListLLmTracesOptions {
-  pageSize?: number;
-  pageOffset?: number;
-  timeTravelDays?: number;
-  endpoint: string;
-}
-
-interface SearchTracesOptions {
-  pageSize?: number;
-  pageOffset?: number;
-  timeTravelDays?: number;
-  endpoint: string;
-
-  filters: Record<string, string[] | Record<string, string[]>>;
-}
-
-export const getLlmTraceById: (
-  authToken: string,
-  id: string,
-  opts?: GetLlmTraceByIdOptions
-) => Promise<LLMModeTrace> = async (authToken, id, opts) => {
-  const { endpoint } = opts ?? {};
-
-  const url = new URL(`${endpoint}/api/trace/${id}`);
-  url.searchParams.set("llmMode", "true");
-
-  const response = await fetch(url.toString(), {
-    method: "GET",
-    headers: {
-      "Content-Type": "application/json",
-      "X-Auth-Token": authToken,
-    },
-  });
-
-  if (!response.ok) {
-    if (response.status === 404) {
-      throw new Error("Trace not found");
-    }
-
-    throw new Error(`Failed to get trace: ${response.statusText}`);
-  }
-
-  return (await response.json()) as Promise<LLMModeTrace>;
-};
-
-export const listLlmTraces = async (
-  authToken: string,
-  opts?: ListLLmTracesOptions
-): Promise<SearchTrace> => {
-  const {
-    pageSize = 10,
-    pageOffset = 0,
-    timeTravelDays = 1,
-    endpoint,
-  } = opts ?? {};
-
-  const response = await fetch(`${endpoint}/api/trace/search`, {
-    method: "POST",
-    headers: {
-      "Content-Type": "application/json",
-      "X-Auth-Token": authToken,
-    },
-    body: JSON.stringify({
-      startDate: addDays(new Date(), -timeTravelDays).toISOString(),
-      endDate: addDays(new Date(), 1).toISOString(),
-      llmMode: true,
-      pageOffset,
-      pageSize,
-    }),
-  });
-
-  return (await response.json()) as Promise<SearchTrace>;
-};
-
-export const searchTraces = async (
-  authToken: string,
-  opts: SearchTracesOptions
-): Promise<SearchTrace> => {
-  const {
-    pageSize = 10,
-    pageOffset = 0,
-    timeTravelDays = 1,
-    endpoint,
-    filters,
-  } = opts;
-
-  const response = await fetch(`${endpoint}/api/trace/search`, {
-    method: "POST",
-    headers: {
-      "Content-Type": "application/json",
-      "X-Auth-Token": authToken,
-    },
-    body: JSON.stringify({
-      startDate: addDays(new Date(), -timeTravelDays).toISOString(),
-      endDate: addDays(new Date(), 1).toISOString(),
-      filters: filters,
-      pageOffset,
-      pageSize,
-    }),
-  });
-
-  return (await response.json()) as Promise<SearchTrace>;
-};