@langwatch/mcp-server 0.0.1

package/dist/index.js.map

@@ -0,0 +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\" };\n\nfunction loadAndValidateArgs() {\n  const args = process.argv.slice(2);\n\n  let apiKey = process.env.LANGWATCH_API_KEY;\n  let endpoint = process.env.LANGWATCH_ENDPOINT;\n\n  args.forEach(arg => {\n    const [key, value] = arg.split('=');\n    if (key === '--apiKey') {\n      apiKey = value;\n    } else if (key === '--endpoint') {\n      endpoint = value;\n    }\n  });\n\n  if (!apiKey) {\n    throw new Error(\"API key is required. Please provide it using --apiKey=<your_api_key>\");\n  }\n\n  return {\n    apiKey: String(apiKey),\n    endpoint: String(endpoint || 'https://app.langwatch.ai'),\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        type: \"text\",\n        text: JSON.stringify(response, null, 2),\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          type: \"text\",\n          text: JSON.stringify(response, null, 2),\n        }]\n      };\n    } catch (error) {\n      if (error instanceof Error && error.message === \"Trace not found\") {\n        return {\n          content: [{\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      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.1\",\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 && esbuild src/index.ts --bundle --platform=node --format=esm --outfile=dist/index.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    \"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    \"zod\": \"^3.22.4\",\n    \"zod-validation-error\": \"^3.3.0\"\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,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,KAAO;AAAA,IACP,wBAAwB;AAAA,EAC1B;AACF;;;AFzCA,SAAS,sBAAsB;AAC7B,QAAM,OAAO,QAAQ,KAAK,MAAM,CAAC;AAEjC,MAAIC,UAAS,QAAQ,IAAI;AACzB,MAAIC,YAAW,QAAQ,IAAI;AAE3B,OAAK,QAAQ,SAAO;AAClB,UAAM,CAAC,KAAK,KAAK,IAAI,IAAI,MAAM,GAAG;AAClC,QAAI,QAAQ,YAAY;AACtB,MAAAD,UAAS;AAAA,IACX,WAAW,QAAQ,cAAc;AAC/B,MAAAC,YAAW;AAAA,IACb;AAAA,EACF,CAAC;AAED,MAAI,CAACD,SAAQ;AACX,UAAM,IAAI,MAAM,sEAAsE;AAAA,EACxF;AAEA,SAAO;AAAA,IACL,QAAQ,OAAOA,OAAM;AAAA,IACrB,UAAU,OAAOC,aAAY,0BAA0B;AAAA,EACzD;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,CAAC;AAAA,QACR,MAAM;AAAA,QACN,MAAM,KAAK,UAAU,UAAU,MAAM,CAAC;AAAA,MACxC,CAAC;AAAA,IACH;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,CAAC;AAAA,UACR,MAAM;AAAA,UACN,MAAM,KAAK,UAAU,UAAU,MAAM,CAAC;AAAA,QACxC,CAAC;AAAA,MACH;AAAA,IACF,SAAS,OAAO;AACd,UAAI,iBAAiB,SAAS,MAAM,YAAY,mBAAmB;AACjE,eAAO;AAAA,UACL,SAAS,CAAC;AAAA,YACR,MAAM;AAAA,YACN,MAAM;AAAA,UACR,CAAC;AAAA,QACH;AAAA,MACF;AAEA,YAAM;AAAA,IACR;AAAA,EACF;AACF;AAEA,IAAM,YAAY,IAAI,qBAAqB;AAC3C,MAAM,OAAO,QAAQ,SAAS;","names":["endpoint","apiKey","endpoint"]}

package/package.json

@@ -0,0 +1,48 @@
+{
+  "name": "@langwatch/mcp-server",
+  "version": "0.0.1",
+  "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 && esbuild src/index.ts --bundle --platform=node --format=esm --outfile=dist/index.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",
+    "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",
+    "zod": "^3.22.4",
+    "zod-validation-error": "^3.3.0"
+  }
+}

package/src/augmentations.d.ts

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

package/src/index.ts

@@ -0,0 +1,95 @@
+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" };
+
+function loadAndValidateArgs() {
+  const args = process.argv.slice(2);
+
+  let apiKey = process.env.LANGWATCH_API_KEY;
+  let endpoint = process.env.LANGWATCH_ENDPOINT;
+
+  args.forEach(arg => {
+    const [key, value] = arg.split('=');
+    if (key === '--apiKey') {
+      apiKey = value;
+    } else if (key === '--endpoint') {
+      endpoint = value;
+    }
+  });
+
+  if (!apiKey) {
+    throw new Error("API key is required. Please provide it using --apiKey=<your_api_key>");
+  }
+
+  return {
+    apiKey: String(apiKey),
+    endpoint: String(endpoint || 'https://app.langwatch.ai'),
+  };
+}
+
+// Use the function to get apiKey and endpoint
+const { apiKey, endpoint } = loadAndValidateArgs();
+
+const server = new McpServer({
+  name: "LangWatch",
+  version: packageJson.version,
+});
+
+server.tool(
+  "get_latest_traces",
+  {
+    pageOffset: z.number().optional(),
+    daysBackToSearch: z.number().optional(),
+  },
+  async ({ pageOffset, daysBackToSearch }) => {
+    const response = await listLlmTraces(apiKey, {
+      pageOffset,
+      timeTravelDays: daysBackToSearch ?? 1,
+      langWatchEndpoint: endpoint,
+    });
+
+    return {
+      content: [{
+        type: "text",
+        text: JSON.stringify(response, null, 2),
+      }],
+    };
+  },
+);
+
+server.tool(
+  "get_trace_by_id",
+  {
+    id: z.string(),
+  },
+  async ({ id }) => {
+    try {
+      const response = await getLlmTraceById(apiKey, id, {
+        langWatchEndpoint: 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;
+    }
+  },
+);
+
+const transport = new StdioServerTransport();
+await server.connect(transport);

package/src/langwatch-api.ts

@@ -0,0 +1,72 @@
+import { addDays } from "date-fns";
+import type { LLMModeTrace } from "langwatch";
+
+interface SearchTrace {
+	traces: LLMModeTrace[];
+}
+
+interface GetLlmTraceByIdOptions {
+	langWatchEndpoint?: string;
+}
+
+interface ListLLmTracesOptions {
+	pageSize?: number;
+	pageOffset?: number;
+	timeTravelDays?: number;
+	langWatchEndpoint?: string;
+}
+
+export const getLlmTraceById = async (authToken: string, id: string, opts?: GetLlmTraceByIdOptions): Promise<LLMModeTrace> => {
+	const { langWatchEndpoint } = opts ?? {};
+
+	const endpoint = langWatchEndpoint ?? "https://app.langwatch.ai";
+
+	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,
+		langWatchEndpoint,
+	} = opts ?? {};
+
+	const endpoint = langWatchEndpoint ?? "https://app.langwatch.ai";
+
+	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>;
+}

package/tsconfig.json

@@ -0,0 +1,28 @@
+{
+  "compilerOptions": {
+    "target": "ES2022",
+    "module": "ESNext",
+    "declaration": true,
+    "declarationMap": true,
+    "outDir": "./dist",
+    "strict": true,
+    "esModuleInterop": true,
+    "resolveJsonModule": true,
+    "allowJs": true,
+    "checkJs": true,
+    "skipLibCheck": true,
+    "forceConsistentCasingInFileNames": true,
+    "moduleResolution": "node",
+    "isolatedModules": true,
+    "jsx": "preserve",
+    "incremental": true,
+    "noUncheckedIndexedAccess": true,
+    "baseUrl": ".",
+    "tsBuildInfoFile": "./tsconfig.tsbuildinfo"
+  },
+  "include": [
+    ".eslintrc.cjs",
+    "**/*.ts",
+  ],
+  "exclude": ["node_modules", "./dist/**/*", "./example/**/*"]
+}

package/tsup.config.ts

@@ -0,0 +1,10 @@
+import { defineConfig } from "tsup";
+
+export default defineConfig([
+  {
+    entry: ["src/index.ts"],
+    format: ["esm"],
+    dts: true,
+    sourcemap: true,
+  },
+]);