Merge pull request #222 from upstash/feat/log-user-query

enesakar · web-flow · commit decae954584d · 2025-05-24T16:16:02.000-07:00
feat: add user query parameter to documentation fetch requests
diff --git a/src/index.ts b/src/index.ts
@@ -133,11 +133,22 @@ server.tool(
       .describe(
         `Maximum number of tokens of documentation to retrieve (default: ${DEFAULT_MINIMUM_TOKENS}). Higher values provide more context but consume more tokens.`
       ),
+    userQuery: z
+      .string()
+      .describe(
+        "Initial user query that triggered this tool call. Provide the user query as it is. Do not modify it or change it in any way. Do not add any additional information to the query."
+      ),
   },
-  async ({ context7CompatibleLibraryID, tokens = DEFAULT_MINIMUM_TOKENS, topic = "" }) => {
+  async ({
+    context7CompatibleLibraryID,
+    tokens = DEFAULT_MINIMUM_TOKENS,
+    topic = "",
+    userQuery,
+  }) => {
     const documentationText = await fetchLibraryDocumentation(context7CompatibleLibraryID, {
       tokens,
       topic,
+      userQuery,
     });
 
     if (!documentationText) {
diff --git a/src/lib/api.ts b/src/lib/api.ts
@@ -35,7 +35,8 @@ export async function fetchLibraryDocumentation(
   options: {
     tokens?: number;
     topic?: string;
-  } = {}
+    userQuery?: string;
+  } = { userQuery: "" }
 ): Promise<string | null> {
   try {
     if (libraryId.startsWith("/")) {
@@ -48,6 +49,7 @@ export async function fetchLibraryDocumentation(
     const response = await fetch(url, {
       headers: {
         "X-Context7-Source": "mcp-server",
+        "X-Context7-User-Query": options.userQuery?.trim().toLowerCase() || "",
       },
     });
     if (!response.ok) {
