diff --git a/src/index.test.ts b/src/index.test.ts index bdf1d65..6c05141 100644 --- a/src/index.test.ts +++ b/src/index.test.ts @@ -124,11 +124,13 @@ describe("Perplexity MCP Server", () => { AGENT_URL, expect.objectContaining({ method: "POST", - headers: expect.objectContaining({ + headers: { "Content-Type": "application/json", Authorization: "Bearer test-api-key", + "User-Agent": "perplexity-mcp/1.2.0", "X-Source": "pplx-mcp-server", - }), + "X-Pplx-Integration": "perplexity-mcp/1.2.0", + }, body: JSON.stringify({ preset: "fast", input: [{ type: "message", role: "user", content: "test question" }], diff --git a/src/server.ts b/src/server.ts index 56c0be1..ad11f35 100644 --- a/src/server.ts +++ b/src/server.ts @@ -111,6 +111,7 @@ async function makeApiRequest( "Authorization": `Bearer ${resolvedApiKey}`, "User-Agent": `perplexity-mcp/${VERSION}`, "X-Source": "pplx-mcp-server", + "X-Pplx-Integration": `perplexity-mcp/${VERSION}`, }; if (serviceOrigin) { headers["X-Service"] = serviceOrigin;