shortcutxl 0.3.77 → 0.3.78

package/dist/app/tools/web-search.js

@@ -174,7 +174,7 @@ function createStartRateLimiter(ops) {
         await waitForStartSlot();
     };
 }
-async function requestWebSearchResult(baseUrl, apiKey, query, outputSchema, ops, signal) {
+async function requestWebSearchResult(baseUrl, apiKey, query, outputSchema, attribution, ops, signal) {
     // Mirrors the embedded exa_web_search route call; api-v1 owns the Exa key and
     // forwards outputSchema to Exa when present.
     const response = await ops.fetch(`${baseUrl}${SHORTCUT_API_ENDPOINTS.webSearch}`, {
@@ -182,6 +182,10 @@ async function requestWebSearchResult(baseUrl, apiKey, query, outputSchema, ops,
         headers: {
             'Content-Type': 'application/json',
             Authorization: `Bearer ${apiKey}`,
+            'X-Client-Platform': 'shortcutxl',
+            ...(attribution.userMessageId ? { 'X-Turn-Id': attribution.userMessageId } : {}),
+            ...(attribution.conversationId ? { 'X-Session-Id': attribution.conversationId } : {}),
+            ...(attribution.agentRunId ? { 'X-Agent-Run-Id': attribution.agentRunId } : {}),
             'User-AgentController': `Mozilla/5.0 (compatible; ${APP_NAME}-coding-agent/1.0)`
         },
         body: JSON.stringify({
@@ -225,6 +229,12 @@ export function createWebSearchTool(baseUrl = SHORTCUT_API_URL, options) {
             if (!apiKey) {
                 throw new Error('Not authenticated. Please log in first.');
             }
+            const conversationId = extCtx?.sessionManager.getSessionId();
+            const attribution = {
+                ...(extCtx?.userMessageId ? { userMessageId: extCtx.userMessageId } : {}),
+                ...(conversationId ? { conversationId } : {}),
+                ...(extCtx?.continuationRunId ? { agentRunId: extCtx.continuationRunId } : {})
+            };
             const queries = await readQueries(normalizedInput, ops);
             if (queries.length === 0) {
                 throw new Error('Web search did not receive any non-empty queries.');
@@ -256,7 +266,7 @@ export function createWebSearchTool(baseUrl = SHORTCUT_API_URL, options) {
             await mapWithMaxInFlight(queries, MAX_IN_FLIGHT_WEB_SEARCH_REQUESTS, waitForStartSlot, async (query) => {
                 let result;
                 try {
-                    result = attachQuery(await requestWebSearchResult(baseUrl, apiKey, query, normalizedInput.outputSchema, ops, signal), query);
+                    result = attachQuery(await requestWebSearchResult(baseUrl, apiKey, query, normalizedInput.outputSchema, attribution, ops, signal), query);
                 }
                 catch (error) {
                     result = {