@langwatch/mcp-server 0.0.4 → 0.1.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 { 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 =\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 the index page and follow the links to the relevant pages.\",\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    const response = await fetch(url ?? \"https://docs.langwatch.ai/llms.txt\");\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\ncreateListTracesByMetadataTool(\n  \"list_traces_by_user_id\",\n  \"userId\",\n  \"metadata.user_id\"\n);\ncreateListTracesByMetadataTool(\n  \"list_traces_by_customer_id\",\n  \"customerId\",\n  \"metadata.customer_id\"\n);\ncreateListTracesByMetadataTool(\n  \"list_traces_by_thread_id\",\n  \"threadId\",\n  \"metadata.thread_id\"\n);\ncreateListTracesByMetadataTool(\n  \"list_traces_by_session_id\",\n  \"sessionId\",\n  \"metadata.thread_id\"\n); // 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(\n  name: string,\n  argName: \"userId\" | \"customerId\" | \"threadId\" | \"sessionId\",\n  metadataKey: string\n) {\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: [{ type: \"text\", text: JSON.stringify(response, null, 2) }],\n      };\n    }\n  );\n}\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.1.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    \"@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;AAEO,IAAM,eAAe,OAC1B,WACA,SACyB;AACzB,QAAM;AAAA,IACJ,WAAW;AAAA,IACX,aAAa;AAAA,IACb,iBAAiB;AAAA,IACjB,UAAAA;AAAA,IACA;AAAA,EACF,IAAI;AAEJ,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;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH,CAAC;AAED,SAAQ,MAAM,SAAS,KAAK;AAC9B;;;ACjHA;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,UAAM,WAAW,MAAM,MAAM,OAAO,oCAAoC;AAExE,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;AAAA,EACE;AAAA,EACA;AAAA,EACA;AACF;AACA;AAAA,EACE;AAAA,EACA;AAAA,EACA;AACF;AACA;AAAA,EACE;AAAA,EACA;AAAA,EACA;AACF;AACA;AAAA,EACE;AAAA,EACA;AAAA,EACA;AACF;AAEA,MAAM,OAAO,QAAQ,SAAS;AAE9B,SAAS,+BACP,MACA,SACA,aACA;AACA,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,CAAC,EAAE,MAAM,QAAQ,MAAM,KAAK,UAAU,UAAU,MAAM,CAAC,EAAE,CAAC;AAAA,MACrE;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.1.0",
   "description": "An MCP server for Langwatch.",
   "type": "module",
   "main": "./dist/index.js",
@@ -17,37 +17,50 @@
     "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": {
+    "@ai-sdk/anthropic": "^2.0.15",
+    "@anthropic-ai/claude-code": "^1.0.111",
     "@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",
+    "@typescript/native-preview": "7.0.0-dev.20250911.1",
+    "ai": "^5.0.40",
+    "dotenv": "^17.2.2",
     "esbuild": "^0.21.5",
     "eslint": "^8.57.0",
     "tsup": "^8.1.0",
+    "tsx": "^4.20.5",
     "typescript": "^4.9.5",
     "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",
-    "zod": "^3.22.4",
-    "zod-validation-error": "^3.3.0"
+    "zod": "^3.25.76",
+    "zod-validation-error": "^3.5.3"
   },
   "bin": {
     "langwatch-mcp-server": "./dist/index.js"
+  },
+  "scripts": {
+    "start": "tsx src/index.ts",
+    "build": "tsup && node build.js",
+    "prepublish": "pnpm run build",
+    "test": "vitest"
   }
-}
+}

package/pnpm-workspace.yaml

@@ -0,0 +1,2 @@
+onlyBuiltDependencies:
+  - node-pty

package/pyproject.toml

@@ -0,0 +1,17 @@
+[project]
+name = "mcp-server"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.11"
+dependencies = [
+    "ipykernel>=6.30.1",
+    "langwatch",
+    "litellm>=1.77.0",
+    "mcp[cli]>=1.13.1",
+    "pandas>=2.3.2",
+    "tenacity>=9.1.2",
+]
+
+[tool.uv.sources]
+langwatch = { path = "../python-sdk", editable = true }

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,15 @@ function loadAndValidateArgs(): { apiKey: string; endpoint: string } {
     .parseSync();
 
   // Use environment variables as fallback
-  const apiKey = argv.apiKey || process.env.LANGWATCH_API_KEY;
+  const apiKey = argv.apiKey ?? process.env.LANGWATCH_API_KEY;
   const endpoint =
-    argv.endpoint ||
-    process.env.LANGWATCH_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,16 +41,37 @@ 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,
 });
 
+server.tool(
+  "fetch_langwatch_docs",
+  "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 the index page and follow the links to the relevant pages.",
+  {
+    url: z
+      .string()
+      .optional()
+      .describe(
+        "The full url of the specific doc page. If not provided, the docs index will be fetched."
+      ),
+  },
+  async ({ url }) => {
+    const response = await fetch(url ?? "https://docs.langwatch.ai/llms.txt");
+
+    return {
+      content: [{ type: "text", text: await response.text() }],
+    };
+  }
+);
+
 server.tool(
   "get_latest_traces",
+  "Retrieves the latest LLM traces.",
   {
     pageOffset: z.number().optional(),
     daysBackToSearch: z.number().optional(),
@@ -59,7 +80,7 @@ server.tool(
     const response = await listLlmTraces(apiKey, {
       pageOffset,
       timeTravelDays: daysBackToSearch ?? 1,
-      langWatchEndpoint: endpoint,
+      endpoint,
     });
 
     return {
@@ -75,13 +96,12 @@ server.tool(
 
 server.tool(
   "get_trace_by_id",
-  {
-    id: z.string(),
-  },
+  "Retrieves a specific LLM trace by its ID.",
+  { id: z.string() },
   async ({ id }) => {
     try {
       const response = await getLlmTraceById(apiKey, id, {
-        langWatchEndpoint: endpoint,
+        endpoint,
       });
 
       return {
@@ -98,7 +118,7 @@ server.tool(
           content: [
             {
               type: "text",
-              text: "Trace not found 😭😭😭😭. If the trace was created recently, it may not be available yet.",
+              text: "Trace not found. If the trace was created recently, it may not be available yet.",
             },
           ],
         };
@@ -109,5 +129,56 @@ 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

@@ -2,71 +2,113 @@ import { addDays } from "date-fns";
 import type { LLMModeTrace } from "langwatch";
 
 interface SearchTrace {
-	traces: LLMModeTrace[];
+  traces: LLMModeTrace[];
 }
 
 interface GetLlmTraceByIdOptions {
-	langWatchEndpoint?: string;
+  endpoint: string;
 }
 
 interface ListLLmTracesOptions {
-	pageSize?: number;
-	pageOffset?: number;
-	timeTravelDays?: number;
-	langWatchEndpoint?: string;
+  pageSize?: number;
+  pageOffset?: number;
+  timeTravelDays?: number;
+  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: (
+  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 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,
-		},
-	});
+  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");
-		}
+  if (!response.ok) {
+    if (response.status === 404) {
+      throw new Error("Trace not found");
+    }
 
-		throw new Error(`Failed to get trace: ${response.statusText}`);
-	}
+    throw new Error(`Failed to get trace: ${response.statusText}`);
+  }
 
-	return await response.json() as Promise<LLMModeTrace>;
+  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>;
-}
+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>;
+};