@langwatch/mcp-server 0.0.4 → 0.0.5

package/dist/index.js

@@ -6978,142 +6978,6 @@ var StdioServerTransport = class {
   }
 };
 
-// node_modules/date-fns/constants.js
-var daysInYear = 365.2425;
-var maxTime = Math.pow(10, 8) * 24 * 60 * 60 * 1e3;
-var minTime = -maxTime;
-var secondsInHour = 3600;
-var secondsInDay = secondsInHour * 24;
-var secondsInWeek = secondsInDay * 7;
-var secondsInYear = secondsInDay * daysInYear;
-var secondsInMonth = secondsInYear / 12;
-var secondsInQuarter = secondsInMonth * 3;
-var constructFromSymbol = Symbol.for("constructDateFrom");
-
-// node_modules/date-fns/constructFrom.js
-function constructFrom(date, value) {
-  if (typeof date === "function") return date(value);
-  if (date && typeof date === "object" && constructFromSymbol in date)
-    return date[constructFromSymbol](value);
-  if (date instanceof Date) return new date.constructor(value);
-  return new Date(value);
-}
-
-// node_modules/date-fns/toDate.js
-function toDate(argument, context) {
-  return constructFrom(context || argument, argument);
-}
-
-// node_modules/date-fns/addDays.js
-function addDays(date, amount, options) {
-  const _date = toDate(date, options?.in);
-  if (isNaN(amount)) return constructFrom(options?.in || date, NaN);
-  if (!amount) return _date;
-  _date.setDate(_date.getDate() + amount);
-  return _date;
-}
-
-// src/langwatch-api.ts
-var getLlmTraceById = async (authToken, id, opts) => {
-  const { langWatchEndpoint } = opts ?? {};
-  const endpoint2 = langWatchEndpoint ?? "https://app.langwatch.ai";
-  const url = new URL(`${endpoint2}/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();
-};
-var listLlmTraces = async (authToken, opts) => {
-  const {
-    pageSize = 10,
-    pageOffset = 0,
-    timeTravelDays = 1,
-    langWatchEndpoint
-  } = opts ?? {};
-  const endpoint2 = langWatchEndpoint ?? "https://app.langwatch.ai";
-  const response = await fetch(`${endpoint2}/api/trace/search`, {
-    method: "POST",
-    headers: {
-      "Content-Type": "application/json",
-      "X-Auth-Token": authToken
-    },
-    body: JSON.stringify({
-      startDate: addDays(/* @__PURE__ */ new Date(), -timeTravelDays).toISOString(),
-      endDate: addDays(/* @__PURE__ */ new Date(), 1).toISOString(),
-      llmMode: true,
-      pageOffset,
-      pageSize
-    })
-  });
-  return await response.json();
-};
-
-// package.json
-var package_default = {
-  name: "@langwatch/mcp-server",
-  version: "0.0.4",
-  description: "An MCP server for Langwatch.",
-  type: "module",
-  main: "./dist/index.js",
-  module: "./dist/index.mjs",
-  types: "./dist/index.d.ts",
-  exports: {
-    ".": {
-      import: "./dist/index.mjs",
-      require: "./dist/index.js"
-    }
-  },
-  repository: {
-    type: "git",
-    url: "https://github.com/langwatch/langwatch.git",
-    directory: "mcp-server"
-  },
-  scripts: {
-    start: "tsx src/index.ts",
-    build: "tsup && node build.js",
-    prepublish: "npm run build"
-  },
-  author: "",
-  license: "MIT",
-  engines: {
-    node: ">=18.0.0"
-  },
-  devDependencies: {
-    "@eslint/js": "^9.4.0",
-    "@types/debug": "^4.1.12",
-    "@types/eslint__js": "^8.42.3",
-    "@types/node": "^16.0.0",
-    "@types/yargs": "^17.0.33",
-    esbuild: "^0.21.5",
-    eslint: "^8.57.0",
-    tsup: "^8.1.0",
-    typescript: "^4.9.5",
-    "typescript-eslint": "^7.11.0"
-  },
-  dependencies: {
-    "@modelcontextprotocol/sdk": "^1.7.0",
-    "date-fns": "^4.1.0",
-    langwatch: "^0.1.6",
-    yargs: "^17.7.2",
-    zod: "^3.22.4",
-    "zod-validation-error": "^3.3.0"
-  },
-  bin: {
-    "langwatch-mcp-server": "./dist/index.js"
-  }
-};
-
 // node_modules/yargs/lib/platform-shims/esm.mjs
 import { notStrictEqual, strictEqual } from "assert";
 
@@ -11963,6 +11827,164 @@ function isYargsInstance(y) {
 var Yargs = YargsFactory(esm_default);
 var yargs_default = Yargs;
 
+// node_modules/date-fns/constants.js
+var daysInYear = 365.2425;
+var maxTime = Math.pow(10, 8) * 24 * 60 * 60 * 1e3;
+var minTime = -maxTime;
+var secondsInHour = 3600;
+var secondsInDay = secondsInHour * 24;
+var secondsInWeek = secondsInDay * 7;
+var secondsInYear = secondsInDay * daysInYear;
+var secondsInMonth = secondsInYear / 12;
+var secondsInQuarter = secondsInMonth * 3;
+var constructFromSymbol = Symbol.for("constructDateFrom");
+
+// node_modules/date-fns/constructFrom.js
+function constructFrom(date, value) {
+  if (typeof date === "function") return date(value);
+  if (date && typeof date === "object" && constructFromSymbol in date)
+    return date[constructFromSymbol](value);
+  if (date instanceof Date) return new date.constructor(value);
+  return new Date(value);
+}
+
+// node_modules/date-fns/toDate.js
+function toDate(argument, context) {
+  return constructFrom(context || argument, argument);
+}
+
+// node_modules/date-fns/addDays.js
+function addDays(date, amount, options) {
+  const _date = toDate(date, options?.in);
+  if (isNaN(amount)) return constructFrom(options?.in || date, NaN);
+  if (!amount) return _date;
+  _date.setDate(_date.getDate() + amount);
+  return _date;
+}
+
+// src/langwatch-api.ts
+var getLlmTraceById = async (authToken, id, opts) => {
+  const { endpoint: endpoint2 } = opts ?? {};
+  const url = new URL(`${endpoint2}/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();
+};
+var listLlmTraces = async (authToken, opts) => {
+  const {
+    pageSize = 10,
+    pageOffset = 0,
+    timeTravelDays = 1,
+    endpoint: endpoint2
+  } = opts ?? {};
+  const response = await fetch(`${endpoint2}/api/trace/search`, {
+    method: "POST",
+    headers: {
+      "Content-Type": "application/json",
+      "X-Auth-Token": authToken
+    },
+    body: JSON.stringify({
+      startDate: addDays(/* @__PURE__ */ new Date(), -timeTravelDays).toISOString(),
+      endDate: addDays(/* @__PURE__ */ new Date(), 1).toISOString(),
+      llmMode: true,
+      pageOffset,
+      pageSize
+    })
+  });
+  return await response.json();
+};
+var searchTraces = async (authToken, opts) => {
+  const {
+    pageSize = 10,
+    pageOffset = 0,
+    timeTravelDays = 1,
+    endpoint: endpoint2,
+    filters
+  } = opts;
+  const response = await fetch(`${endpoint2}/api/trace/search`, {
+    method: "POST",
+    headers: {
+      "Content-Type": "application/json",
+      "X-Auth-Token": authToken
+    },
+    body: JSON.stringify({
+      startDate: addDays(/* @__PURE__ */ new Date(), -timeTravelDays).toISOString(),
+      endDate: addDays(/* @__PURE__ */ new Date(), 1).toISOString(),
+      filters,
+      pageOffset,
+      pageSize
+    })
+  });
+  return await response.json();
+};
+
+// package.json
+var package_default = {
+  name: "@langwatch/mcp-server",
+  version: "0.0.5",
+  description: "An MCP server for Langwatch.",
+  type: "module",
+  main: "./dist/index.js",
+  module: "./dist/index.mjs",
+  types: "./dist/index.d.ts",
+  exports: {
+    ".": {
+      import: "./dist/index.mjs",
+      require: "./dist/index.js"
+    }
+  },
+  repository: {
+    type: "git",
+    url: "https://github.com/langwatch/langwatch.git",
+    directory: "mcp-server"
+  },
+  scripts: {
+    start: "tsx src/index.ts",
+    build: "tsup && node build.js",
+    prepublish: "npm run build"
+  },
+  author: "",
+  license: "MIT",
+  engines: {
+    node: ">=18.0.0"
+  },
+  devDependencies: {
+    "@eslint/js": "^9.4.0",
+    "@types/debug": "^4.1.12",
+    "@types/eslint__js": "^8.42.3",
+    "@types/node": "^16.0.0",
+    "@types/yargs": "^17.0.33",
+    esbuild: "^0.21.5",
+    eslint: "^8.57.0",
+    tsup: "^8.1.0",
+    typescript: "^4.9.5",
+    "typescript-eslint": "^7.11.0"
+  },
+  dependencies: {
+    "@modelcontextprotocol/sdk": "^1.7.0",
+    "date-fns": "^4.1.0",
+    langwatch: "^0.1.6",
+    yargs: "^17.7.2",
+    zod: "^3.22.4",
+    "zod-validation-error": "^3.3.0"
+  },
+  bin: {
+    "langwatch-mcp-server": "./dist/index.js"
+  }
+};
+
 // src/index.ts
 function loadAndValidateArgs() {
   const argv = yargs_default(hideBin(process.argv)).option("apiKey", {
@@ -11973,11 +11995,11 @@ function loadAndValidateArgs() {
     description: "LangWatch API endpoint",
     default: "https://app.langwatch.ai"
   }).help().alias("help", "h").parseSync();
-  const apiKey2 = argv.apiKey || process.env.LANGWATCH_API_KEY;
-  const endpoint2 = argv.endpoint || process.env.LANGWATCH_ENDPOINT || "https://app.langwatch.ai";
+  const apiKey2 = argv.apiKey ?? process.env.LANGWATCH_API_KEY;
+  const endpoint2 = argv.endpoint ?? process.env.LANGWATCH_ENDPOINT ?? "https://app.langwatch.ai";
   if (!apiKey2) {
     throw new Error(
-      "API key is required. Please provide it using --apiKey=<your_api_key> or set LANGWATCH_API_KEY environment variable"
+      "API key is required. Please provide it using --apiKey <your_api_key> or set LANGWATCH_API_KEY environment variable"
     );
   }
   return {
@@ -11986,6 +12008,7 @@ function loadAndValidateArgs() {
   };
 }
 var { apiKey, endpoint } = loadAndValidateArgs();
+var transport = new StdioServerTransport();
 var server = new McpServer({
   name: "LangWatch",
   version: package_default.version
@@ -12000,7 +12023,7 @@ server.tool(
     const response = await listLlmTraces(apiKey, {
       pageOffset,
       timeTravelDays: daysBackToSearch ?? 1,
-      langWatchEndpoint: endpoint
+      endpoint
     });
     return {
       content: [
@@ -12014,13 +12037,11 @@ server.tool(
 );
 server.tool(
   "get_trace_by_id",
-  {
-    id: z.string()
-  },
+  { id: z.string() },
   async ({ id }) => {
     try {
       const response = await getLlmTraceById(apiKey, id, {
-        langWatchEndpoint: endpoint
+        endpoint
       });
       return {
         content: [
@@ -12045,8 +12066,38 @@ server.tool(
     }
   }
 );
-var transport = new StdioServerTransport();
+createListTracesByMetadataTool("list_traces_by_user_id", "userId", "metadata.user_id");
+createListTracesByMetadataTool("list_traces_by_customer_id", "customerId", "metadata.customer_id");
+createListTracesByMetadataTool("list_traces_by_thread_id", "threadId", "metadata.thread_id");
+createListTracesByMetadataTool("list_traces_by_session_id", "sessionId", "metadata.thread_id");
 await server.connect(transport);
+function createListTracesByMetadataTool(name, argName, metadataKey) {
+  return server.tool(
+    name,
+    {
+      [argName]: z.string(),
+      pageSize: z.number().optional(),
+      pageOffset: z.number().optional(),
+      daysBackToSearch: z.number().optional()
+    },
+    async ({ pageSize, pageOffset, daysBackToSearch, ...restArgs }) => {
+      const response = await searchTraces(apiKey, {
+        endpoint,
+        pageSize,
+        pageOffset,
+        timeTravelDays: daysBackToSearch ?? 1,
+        filters: {
+          [metadataKey]: [restArgs[argName]]
+        }
+      });
+      return {
+        content: [
+          { type: "text", text: JSON.stringify(response, null, 2) }
+        ]
+      };
+    }
+  );
+}
 /*! Bundled license information:
 
 yargs-parser/build/lib/string-utils.js:

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 { getLlmTraceById, listLlmTraces } from \"./langwatch-api\";\nimport packageJson from \"../package.json\" assert { type: \"json\" };\nimport yargs from \"yargs\";\nimport { hideBin } from \"yargs/helpers\";\n\nfunction loadAndValidateArgs(): { apiKey: string; endpoint: string } {\n  // Parse command line arguments with yargs\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\n// Use the function to get apiKey and endpoint\nconst { apiKey, endpoint } = loadAndValidateArgs();\n\nconst server = new McpServer({\n  name: \"LangWatch\",\n  version: packageJson.version,\n});\n\nserver.tool(\n  \"get_latest_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      langWatchEndpoint: 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  {\n    id: z.string(),\n  },\n  async ({ id }) => {\n    try {\n      const response = await getLlmTraceById(apiKey, id, {\n        langWatchEndpoint: 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\nconst transport = new StdioServerTransport();\nawait server.connect(transport);\n","import { addDays } from \"date-fns\";\nimport type { LLMModeTrace } from \"langwatch\";\n\ninterface SearchTrace {\n\ttraces: LLMModeTrace[];\n}\n\ninterface GetLlmTraceByIdOptions {\n\tlangWatchEndpoint?: string;\n}\n\ninterface ListLLmTracesOptions {\n\tpageSize?: number;\n\tpageOffset?: number;\n\ttimeTravelDays?: number;\n\tlangWatchEndpoint?: string;\n}\n\nexport const getLlmTraceById = async (authToken: string, id: string, opts?: GetLlmTraceByIdOptions): Promise<LLMModeTrace> => {\n\tconst { langWatchEndpoint } = opts ?? {};\n\n\tconst endpoint = langWatchEndpoint ?? \"https://app.langwatch.ai\";\n\n\tconst url = new URL(`${endpoint}/api/trace/${id}`);\n\turl.searchParams.set(\"llmMode\", \"true\");\n\n\tconst response = await fetch(url.toString(), {\n\t\tmethod: \"GET\",\n\t\theaders: {\n\t\t\t\"Content-Type\": \"application/json\",\n\t\t\t\"X-Auth-Token\": authToken,\n\t\t},\n\t});\n\n\tif (!response.ok) {\n\t\tif (response.status === 404) {\n\t\t\tthrow new Error(\"Trace not found\");\n\t\t}\n\n\t\tthrow new Error(`Failed to get trace: ${response.statusText}`);\n\t}\n\n\treturn await response.json() as Promise<LLMModeTrace>;\n};\n\nexport const listLlmTraces = async (authToken: string, opts?: ListLLmTracesOptions): Promise<SearchTrace> => {\n\tconst {\n\t\tpageSize = 10,\n\t\tpageOffset = 0,\n\t\ttimeTravelDays = 1,\n\t\tlangWatchEndpoint,\n\t} = opts ?? {};\n\n\tconst endpoint = langWatchEndpoint ?? \"https://app.langwatch.ai\";\n\n\tconst response = await fetch(`${endpoint}/api/trace/search`, {\n\t\tmethod: \"POST\",\n\t\theaders: {\n\t\t\t\"Content-Type\": \"application/json\",\n\t\t\t\"X-Auth-Token\": authToken,\n\t\t},\n\t\tbody: JSON.stringify({\n\t\t\tstartDate: addDays(new Date(), -timeTravelDays).toISOString(),\n\t\t\tendDate: addDays(new Date(), 1).toISOString(),\n\t\t\tllmMode: true,\n\t\t\tpageOffset,\n\t\t\tpageSize,\n\t\t}),\n\t});\n\n\treturn await response.json() as Promise<SearchTrace>;\n}\n","{\n  \"name\": \"@langwatch/mcp-server\",\n  \"version\": \"0.0.4\",\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\": \"npm run build\"\n  },\n  \"author\": \"\",\n  \"license\": \"MIT\",\n  \"engines\": {\n    \"node\": \">=18.0.0\"\n  },\n  \"devDependencies\": {\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    \"esbuild\": \"^0.21.5\",\n    \"eslint\": \"^8.57.0\",\n    \"tsup\": \"^8.1.0\",\n    \"typescript\": \"^4.9.5\",\n    \"typescript-eslint\": \"^7.11.0\"\n  },\n  \"dependencies\": {\n    \"@modelcontextprotocol/sdk\": \"^1.7.0\",\n    \"date-fns\": \"^4.1.0\",\n    \"langwatch\": \"^0.1.6\",\n    \"yargs\": \"^17.7.2\",\n    \"zod\": \"^3.22.4\",\n    \"zod-validation-error\": \"^3.3.0\"\n  },\n  \"bin\": {\n    \"langwatch-mcp-server\": \"./dist/index.js\"\n  }\n}\n"],"mappings":";AAAA,SAAS,iBAAiB;AAC1B,SAAS,4BAA4B;AACrC,SAAS,SAAS;;;ACFlB,SAAS,eAAe;AAkBjB,IAAM,kBAAkB,OAAO,WAAmB,IAAY,SAAyD;AAC7H,QAAM,EAAE,kBAAkB,IAAI,QAAQ,CAAC;AAEvC,QAAMA,YAAW,qBAAqB;AAEtC,QAAM,MAAM,IAAI,IAAI,GAAGA,SAAQ,cAAc,EAAE,EAAE;AACjD,MAAI,aAAa,IAAI,WAAW,MAAM;AAEtC,QAAM,WAAW,MAAM,MAAM,IAAI,SAAS,GAAG;AAAA,IAC5C,QAAQ;AAAA,IACR,SAAS;AAAA,MACR,gBAAgB;AAAA,MAChB,gBAAgB;AAAA,IACjB;AAAA,EACD,CAAC;AAED,MAAI,CAAC,SAAS,IAAI;AACjB,QAAI,SAAS,WAAW,KAAK;AAC5B,YAAM,IAAI,MAAM,iBAAiB;AAAA,IAClC;AAEA,UAAM,IAAI,MAAM,wBAAwB,SAAS,UAAU,EAAE;AAAA,EAC9D;AAEA,SAAO,MAAM,SAAS,KAAK;AAC5B;AAEO,IAAM,gBAAgB,OAAO,WAAmB,SAAsD;AAC5G,QAAM;AAAA,IACL,WAAW;AAAA,IACX,aAAa;AAAA,IACb,iBAAiB;AAAA,IACjB;AAAA,EACD,IAAI,QAAQ,CAAC;AAEb,QAAMA,YAAW,qBAAqB;AAEtC,QAAM,WAAW,MAAM,MAAM,GAAGA,SAAQ,qBAAqB;AAAA,IAC5D,QAAQ;AAAA,IACR,SAAS;AAAA,MACR,gBAAgB;AAAA,MAChB,gBAAgB;AAAA,IACjB;AAAA,IACA,MAAM,KAAK,UAAU;AAAA,MACpB,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,IACD,CAAC;AAAA,EACF,CAAC;AAED,SAAO,MAAM,SAAS,KAAK;AAC5B;;;ACvEA;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,EAChB;AAAA,EACA,QAAU;AAAA,EACV,SAAW;AAAA,EACX,SAAW;AAAA,IACT,MAAQ;AAAA,EACV;AAAA,EACA,iBAAmB;AAAA,IACjB,cAAc;AAAA,IACd,gBAAgB;AAAA,IAChB,qBAAqB;AAAA,IACrB,eAAe;AAAA,IACf,gBAAgB;AAAA,IAChB,SAAW;AAAA,IACX,QAAU;AAAA,IACV,MAAQ;AAAA,IACR,YAAc;AAAA,IACd,qBAAqB;AAAA,EACvB;AAAA,EACA,cAAgB;AAAA,IACd,6BAA6B;AAAA,IAC7B,YAAY;AAAA,IACZ,WAAa;AAAA,IACb,OAAS;AAAA,IACT,KAAO;AAAA,IACP,wBAAwB;AAAA,EAC1B;AAAA,EACA,KAAO;AAAA,IACL,wBAAwB;AAAA,EAC1B;AACF;;;AF/CA,OAAO,WAAW;AAClB,SAAS,eAAe;AAExB,SAAS,sBAA4D;AAEnE,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;AAGA,IAAM,EAAE,QAAQ,SAAS,IAAI,oBAAoB;AAEjD,IAAM,SAAS,IAAI,UAAU;AAAA,EAC3B,MAAM;AAAA,EACN,SAAS,gBAAY;AACvB,CAAC;AAED,OAAO;AAAA,EACL;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,mBAAmB;AAAA,IACrB,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,IACE,IAAI,EAAE,OAAO;AAAA,EACf;AAAA,EACA,OAAO,EAAE,GAAG,MAAM;AAChB,QAAI;AACF,YAAM,WAAW,MAAM,gBAAgB,QAAQ,IAAI;AAAA,QACjD,mBAAmB;AAAA,MACrB,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,IAAM,YAAY,IAAI,qBAAqB;AAC3C,MAAM,OAAO,QAAQ,SAAS;","names":["endpoint","apiKey","endpoint"]}
+{"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 = argv.endpoint ?? process.env.LANGWATCH_ENDPOINT ?? \"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  \"get_latest_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  { 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\ncreateListTracesByMetadataTool(\"list_traces_by_user_id\", \"userId\", \"metadata.user_id\");\ncreateListTracesByMetadataTool(\"list_traces_by_customer_id\", \"customerId\", \"metadata.customer_id\");\ncreateListTracesByMetadataTool(\"list_traces_by_thread_id\", \"threadId\", \"metadata.thread_id\");\ncreateListTracesByMetadataTool(\"list_traces_by_session_id\", \"sessionId\", \"metadata.thread_id\"); // We access the thread_id in the metadata, as that is our name for the session_id\n\nawait server.connect(transport);\n\nfunction createListTracesByMetadataTool(name: string, argName: \"userId\" | \"customerId\" | \"threadId\" | \"sessionId\", metadataKey: string) {\n  return server.tool(\n    name,\n    {\n      [argName]: z.string(),\n      pageSize: z.number().optional(),\n      pageOffset: z.number().optional(),\n      daysBackToSearch: z.number().optional(),\n    },\n    async ({ pageSize, pageOffset, daysBackToSearch, ...restArgs }) => {\n      const response = await searchTraces(apiKey, {\n        endpoint,\n        pageSize: pageSize as number | undefined,\n        pageOffset: pageOffset as number | undefined,\n        timeTravelDays: (daysBackToSearch ?? 1) as number,\n        filters: {\n          [metadataKey]: [restArgs[argName] as string],\n        },\n      });\n  \n      return {\n        content: [\n          { type: \"text\", text: JSON.stringify(response, null, 2) },\n        ],\n      };\n    }\n  );\n}\n","import { addDays } from \"date-fns\";\nimport type { LLMModeTrace } from \"langwatch\";\n\ninterface SearchTrace {\n\ttraces: LLMModeTrace[];\n}\n\ninterface GetLlmTraceByIdOptions {\n\tendpoint: string;\n}\n\ninterface ListLLmTracesOptions {\n\tpageSize?: number;\n\tpageOffset?: number;\n\ttimeTravelDays?: number;\n\tendpoint: string;\n}\n\ninterface SearchTracesOptions {\n\tpageSize?: number;\n\tpageOffset?: number;\n\ttimeTravelDays?: number;\n\tendpoint: string;\n\n\tfilters: Record<string, string[] | Record<string, string[]>>;\n}\n\nexport const getLlmTraceById = async (authToken: string, id: string, opts?: GetLlmTraceByIdOptions): Promise<LLMModeTrace> => {\n\tconst { endpoint } = opts ?? {};\n\n\tconst url = new URL(`${endpoint}/api/trace/${id}`);\n\turl.searchParams.set(\"llmMode\", \"true\");\n\n\tconst response = await fetch(url.toString(), {\n\t\tmethod: \"GET\",\n\t\theaders: {\n\t\t\t\"Content-Type\": \"application/json\",\n\t\t\t\"X-Auth-Token\": authToken,\n\t\t},\n\t});\n\n\tif (!response.ok) {\n\t\tif (response.status === 404) {\n\t\t\tthrow new Error(\"Trace not found\");\n\t\t}\n\n\t\tthrow new Error(`Failed to get trace: ${response.statusText}`);\n\t}\n\n\treturn await response.json() as Promise<LLMModeTrace>;\n};\n\nexport const listLlmTraces = async (authToken: string, opts?: ListLLmTracesOptions): Promise<SearchTrace> => {\n\tconst {\n\t\tpageSize = 10,\n\t\tpageOffset = 0,\n\t\ttimeTravelDays = 1,\n\t\tendpoint,\n\t} = opts ?? {};\n\n\tconst response = await fetch(`${endpoint}/api/trace/search`, {\n\t\tmethod: \"POST\",\n\t\theaders: {\n\t\t\t\"Content-Type\": \"application/json\",\n\t\t\t\"X-Auth-Token\": authToken,\n\t\t},\n\t\tbody: JSON.stringify({\n\t\t\tstartDate: addDays(new Date(), -timeTravelDays).toISOString(),\n\t\t\tendDate: addDays(new Date(), 1).toISOString(),\n\t\t\tllmMode: true,\n\t\t\tpageOffset,\n\t\t\tpageSize,\n\t\t}),\n\t});\n\n\treturn await response.json() as Promise<SearchTrace>;\n}\n\nexport const searchTraces = async (authToken: string, opts: SearchTracesOptions): Promise<SearchTrace> => {\n\tconst {\n\t\tpageSize = 10,\n\t\tpageOffset = 0,\n\t\ttimeTravelDays = 1,\n\t\tendpoint,\n\t\tfilters,\n\t} = opts;\n\n\tconst response = await fetch(`${endpoint}/api/trace/search`, {\n\t\tmethod: \"POST\",\n\t\theaders: {\n\t\t\t\"Content-Type\": \"application/json\",\n\t\t\t\"X-Auth-Token\": authToken,\n\t\t},\n\t\tbody: JSON.stringify({\n\t\t\tstartDate: addDays(new Date(), -timeTravelDays).toISOString(),\n\t\t\tendDate: addDays(new Date(), 1).toISOString(),\n\t\t\tfilters: filters,\n\t\t\tpageOffset,\n\t\t\tpageSize,\n\t\t}),\n\t});\n\n\treturn await response.json() as Promise<SearchTrace>;\n}\n","{\n  \"name\": \"@langwatch/mcp-server\",\n  \"version\": \"0.0.5\",\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\": \"npm run build\"\n  },\n  \"author\": \"\",\n  \"license\": \"MIT\",\n  \"engines\": {\n    \"node\": \">=18.0.0\"\n  },\n  \"devDependencies\": {\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    \"esbuild\": \"^0.21.5\",\n    \"eslint\": \"^8.57.0\",\n    \"tsup\": \"^8.1.0\",\n    \"typescript\": \"^4.9.5\",\n    \"typescript-eslint\": \"^7.11.0\"\n  },\n  \"dependencies\": {\n    \"@modelcontextprotocol/sdk\": \"^1.7.0\",\n    \"date-fns\": \"^4.1.0\",\n    \"langwatch\": \"^0.1.6\",\n    \"yargs\": \"^17.7.2\",\n    \"zod\": \"^3.22.4\",\n    \"zod-validation-error\": \"^3.3.0\"\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,kBAAkB,OAAO,WAAmB,IAAY,SAAyD;AAC7H,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,IAC5C,QAAQ;AAAA,IACR,SAAS;AAAA,MACR,gBAAgB;AAAA,MAChB,gBAAgB;AAAA,IACjB;AAAA,EACD,CAAC;AAED,MAAI,CAAC,SAAS,IAAI;AACjB,QAAI,SAAS,WAAW,KAAK;AAC5B,YAAM,IAAI,MAAM,iBAAiB;AAAA,IAClC;AAEA,UAAM,IAAI,MAAM,wBAAwB,SAAS,UAAU,EAAE;AAAA,EAC9D;AAEA,SAAO,MAAM,SAAS,KAAK;AAC5B;AAEO,IAAM,gBAAgB,OAAO,WAAmB,SAAsD;AAC5G,QAAM;AAAA,IACL,WAAW;AAAA,IACX,aAAa;AAAA,IACb,iBAAiB;AAAA,IACjB,UAAAA;AAAA,EACD,IAAI,QAAQ,CAAC;AAEb,QAAM,WAAW,MAAM,MAAM,GAAGA,SAAQ,qBAAqB;AAAA,IAC5D,QAAQ;AAAA,IACR,SAAS;AAAA,MACR,gBAAgB;AAAA,MAChB,gBAAgB;AAAA,IACjB;AAAA,IACA,MAAM,KAAK,UAAU;AAAA,MACpB,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,IACD,CAAC;AAAA,EACF,CAAC;AAED,SAAO,MAAM,SAAS,KAAK;AAC5B;AAEO,IAAM,eAAe,OAAO,WAAmB,SAAoD;AACzG,QAAM;AAAA,IACL,WAAW;AAAA,IACX,aAAa;AAAA,IACb,iBAAiB;AAAA,IACjB,UAAAA;AAAA,IACA;AAAA,EACD,IAAI;AAEJ,QAAM,WAAW,MAAM,MAAM,GAAGA,SAAQ,qBAAqB;AAAA,IAC5D,QAAQ;AAAA,IACR,SAAS;AAAA,MACR,gBAAgB;AAAA,MAChB,gBAAgB;AAAA,IACjB;AAAA,IACA,MAAM,KAAK,UAAU;AAAA,MACpB,WAAW,QAAQ,oBAAI,KAAK,GAAG,CAAC,cAAc,EAAE,YAAY;AAAA,MAC5D,SAAS,QAAQ,oBAAI,KAAK,GAAG,CAAC,EAAE,YAAY;AAAA,MAC5C;AAAA,MACA;AAAA,MACA;AAAA,IACD,CAAC;AAAA,EACF,CAAC;AAED,SAAO,MAAM,SAAS,KAAK;AAC5B;;;ACvGA;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,EAChB;AAAA,EACA,QAAU;AAAA,EACV,SAAW;AAAA,EACX,SAAW;AAAA,IACT,MAAQ;AAAA,EACV;AAAA,EACA,iBAAmB;AAAA,IACjB,cAAc;AAAA,IACd,gBAAgB;AAAA,IAChB,qBAAqB;AAAA,IACrB,eAAe;AAAA,IACf,gBAAgB;AAAA,IAChB,SAAW;AAAA,IACX,QAAU;AAAA,IACV,MAAQ;AAAA,IACR,YAAc;AAAA,IACd,qBAAqB;AAAA,EACvB;AAAA,EACA,cAAgB;AAAA,IACd,6BAA6B;AAAA,IAC7B,YAAY;AAAA,IACZ,WAAa;AAAA,IACb,OAAS;AAAA,IACT,KAAO;AAAA,IACP,wBAAwB;AAAA,EAC1B;AAAA,EACA,KAAO;AAAA,IACL,wBAAwB;AAAA,EAC1B;AACF;;;AF3CA,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,YAAW,KAAK,YAAY,QAAQ,IAAI,sBAAsB;AAEpE,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,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,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,+BAA+B,0BAA0B,UAAU,kBAAkB;AACrF,+BAA+B,8BAA8B,cAAc,sBAAsB;AACjG,+BAA+B,4BAA4B,YAAY,oBAAoB;AAC3F,+BAA+B,6BAA6B,aAAa,oBAAoB;AAE7F,MAAM,OAAO,QAAQ,SAAS;AAE9B,SAAS,+BAA+B,MAAc,SAA6D,aAAqB;AACtI,SAAO,OAAO;AAAA,IACZ;AAAA,IACA;AAAA,MACE,CAAC,OAAO,GAAG,EAAE,OAAO;AAAA,MACpB,UAAU,EAAE,OAAO,EAAE,SAAS;AAAA,MAC9B,YAAY,EAAE,OAAO,EAAE,SAAS;AAAA,MAChC,kBAAkB,EAAE,OAAO,EAAE,SAAS;AAAA,IACxC;AAAA,IACA,OAAO,EAAE,UAAU,YAAY,kBAAkB,GAAG,SAAS,MAAM;AACjE,YAAM,WAAW,MAAM,aAAa,QAAQ;AAAA,QAC1C;AAAA,QACA;AAAA,QACA;AAAA,QACA,gBAAiB,oBAAoB;AAAA,QACrC,SAAS;AAAA,UACP,CAAC,WAAW,GAAG,CAAC,SAAS,OAAO,CAAW;AAAA,QAC7C;AAAA,MACF,CAAC;AAED,aAAO;AAAA,QACL,SAAS;AAAA,UACP,EAAE,MAAM,QAAQ,MAAM,KAAK,UAAU,UAAU,MAAM,CAAC,EAAE;AAAA,QAC1D;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;","names":["endpoint","apiKey","endpoint"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langwatch/mcp-server",
-  "version": "0.0.4",
+  "version": "0.0.5",
   "description": "An MCP server for Langwatch.",
   "type": "module",
   "main": "./dist/index.js",

package/src/index.ts

@@ -1,13 +1,13 @@
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 import { z } from "zod";
-import { getLlmTraceById, listLlmTraces } from "./langwatch-api";
-import packageJson from "../package.json" assert { type: "json" };
 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 } {
-  // Parse command line arguments with yargs
   const argv = yargs(hideBin(process.argv))
     .option("apiKey", {
       type: "string",
@@ -23,15 +23,12 @@ function loadAndValidateArgs(): { apiKey: string; endpoint: string } {
     .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";
+  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"
+      "API key is required. Please provide it using --apiKey <your_api_key> or set LANGWATCH_API_KEY environment variable"
     );
   }
 
@@ -41,9 +38,9 @@ function loadAndValidateArgs(): { apiKey: string; endpoint: string } {
   };
 }
 
-// Use the function to get apiKey and endpoint
 const { apiKey, endpoint } = loadAndValidateArgs();
 
+const transport = new StdioServerTransport();
 const server = new McpServer({
   name: "LangWatch",
   version: packageJson.version,
@@ -59,7 +56,7 @@ server.tool(
     const response = await listLlmTraces(apiKey, {
       pageOffset,
       timeTravelDays: daysBackToSearch ?? 1,
-      langWatchEndpoint: endpoint,
+      endpoint,
     });
 
     return {
@@ -75,13 +72,11 @@ server.tool(
 
 server.tool(
   "get_trace_by_id",
-  {
-    id: z.string(),
-  },
+  { id: z.string() },
   async ({ id }) => {
     try {
       const response = await getLlmTraceById(apiKey, id, {
-        langWatchEndpoint: endpoint,
+        endpoint,
       });
 
       return {
@@ -109,5 +104,38 @@ server.tool(
   }
 );
 
-const transport = new StdioServerTransport();
+createListTracesByMetadataTool("list_traces_by_user_id", "userId", "metadata.user_id");
+createListTracesByMetadataTool("list_traces_by_customer_id", "customerId", "metadata.customer_id");
+createListTracesByMetadataTool("list_traces_by_thread_id", "threadId", "metadata.thread_id");
+createListTracesByMetadataTool("list_traces_by_session_id", "sessionId", "metadata.thread_id"); // We access the thread_id in the metadata, as that is our name for the session_id
+
 await server.connect(transport);
+
+function createListTracesByMetadataTool(name: string, argName: "userId" | "customerId" | "threadId" | "sessionId", metadataKey: string) {
+  return server.tool(
+    name,
+    {
+      [argName]: z.string(),
+      pageSize: z.number().optional(),
+      pageOffset: z.number().optional(),
+      daysBackToSearch: z.number().optional(),
+    },
+    async ({ pageSize, pageOffset, daysBackToSearch, ...restArgs }) => {
+      const response = await searchTraces(apiKey, {
+        endpoint,
+        pageSize: pageSize as number | undefined,
+        pageOffset: pageOffset as number | undefined,
+        timeTravelDays: (daysBackToSearch ?? 1) as number,
+        filters: {
+          [metadataKey]: [restArgs[argName] as string],
+        },
+      });
+  
+      return {
+        content: [
+          { type: "text", text: JSON.stringify(response, null, 2) },
+        ],
+      };
+    }
+  );
+}

package/src/langwatch-api.ts

@@ -6,20 +6,27 @@ interface SearchTrace {
 }
 
 interface GetLlmTraceByIdOptions {
-	langWatchEndpoint?: string;
+	endpoint: string;
 }
 
 interface ListLLmTracesOptions {
 	pageSize?: number;
 	pageOffset?: number;
 	timeTravelDays?: number;
-	langWatchEndpoint?: string;
+	endpoint: string;
 }
 
-export const getLlmTraceById = async (authToken: string, id: string, opts?: GetLlmTraceByIdOptions): Promise<LLMModeTrace> => {
-	const { langWatchEndpoint } = opts ?? {};
+interface SearchTracesOptions {
+	pageSize?: number;
+	pageOffset?: number;
+	timeTravelDays?: number;
+	endpoint: string;
 
-	const endpoint = langWatchEndpoint ?? "https://app.langwatch.ai";
+	filters: Record<string, string[] | Record<string, string[]>>;
+}
+
+export const getLlmTraceById = async (authToken: string, id: string, opts?: GetLlmTraceByIdOptions): Promise<LLMModeTrace> => {
+	const { endpoint } = opts ?? {};
 
 	const url = new URL(`${endpoint}/api/trace/${id}`);
 	url.searchParams.set("llmMode", "true");
@@ -48,11 +55,9 @@ export const listLlmTraces = async (authToken: string, opts?: ListLLmTracesOptio
 		pageSize = 10,
 		pageOffset = 0,
 		timeTravelDays = 1,
-		langWatchEndpoint,
+		endpoint,
 	} = opts ?? {};
 
-	const endpoint = langWatchEndpoint ?? "https://app.langwatch.ai";
-
 	const response = await fetch(`${endpoint}/api/trace/search`, {
 		method: "POST",
 		headers: {
@@ -70,3 +75,30 @@ export const listLlmTraces = async (authToken: string, opts?: ListLLmTracesOptio
 
 	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>;
+}