searchsocket 0.3.2 → 0.4.0

package/dist/client.cjs

@@ -1,5 +1,59 @@
 'use strict';
 
+// src/merge.ts
+function mergeSearchResults(initial, reranked, options) {
+  const maxDisplacement = options?.maxDisplacement ?? 3;
+  const initialUrls = initial.results.map((r) => r.url);
+  const rerankedUrls = reranked.results.map((r) => r.url);
+  const initialPos = /* @__PURE__ */ new Map();
+  for (let i = 0; i < initialUrls.length; i++) {
+    initialPos.set(initialUrls[i], i);
+  }
+  const rerankedPos = /* @__PURE__ */ new Map();
+  for (let i = 0; i < rerankedUrls.length; i++) {
+    rerankedPos.set(rerankedUrls[i], i);
+  }
+  const displacements = [];
+  for (const url of initialUrls) {
+    const iPos = initialPos.get(url);
+    const rPos = rerankedPos.get(url);
+    const displacement = rPos !== void 0 ? Math.abs(iPos - rPos) : 0;
+    displacements.push({ url, displacement });
+  }
+  const totalResults = displacements.length;
+  if (totalResults === 0) {
+    return {
+      response: reranked,
+      usedRerankedOrder: true,
+      displacements
+    };
+  }
+  const hasLargeDisplacement = displacements.some((d) => d.displacement > maxDisplacement);
+  if (hasLargeDisplacement) {
+    return {
+      response: reranked,
+      usedRerankedOrder: true,
+      displacements
+    };
+  }
+  const rerankedScoreMap = /* @__PURE__ */ new Map();
+  for (const result of reranked.results) {
+    rerankedScoreMap.set(result.url, result.score);
+  }
+  const mergedResults = initial.results.map((result) => ({
+    ...result,
+    score: rerankedScoreMap.get(result.url) ?? result.score
+  }));
+  return {
+    response: {
+      ...reranked,
+      results: mergedResults
+    },
+    usedRerankedOrder: false,
+    displacements
+  };
+}
+
 // src/client.ts
 function createSearchClient(options = {}) {
   const endpoint = options.endpoint ?? "/api/search";
@@ -27,10 +81,77 @@ function createSearchClient(options = {}) {
         throw new Error(message);
       }
       return payload;
+    },
+    async streamSearch(request, onPhase) {
+      const response = await fetchImpl(endpoint, {
+        method: "POST",
+        headers: {
+          "content-type": "application/json"
+        },
+        body: JSON.stringify(request)
+      });
+      if (!response.ok) {
+        let payload;
+        try {
+          payload = await response.json();
+        } catch {
+          throw new Error("Search failed");
+        }
+        const message = payload.error?.message ?? "Search failed";
+        throw new Error(message);
+      }
+      const contentType = response.headers.get("content-type") ?? "";
+      if (contentType.includes("application/json")) {
+        const data = await response.json();
+        onPhase({ phase: "initial", data });
+        return data;
+      }
+      if (!response.body) {
+        throw new Error("Response body is not readable");
+      }
+      const reader = response.body.getReader();
+      const decoder = new TextDecoder();
+      let buffer = "";
+      let lastResponse = null;
+      for (; ; ) {
+        const { done, value } = await reader.read();
+        if (done) break;
+        buffer += decoder.decode(value, { stream: true });
+        let newlineIdx;
+        while ((newlineIdx = buffer.indexOf("\n")) !== -1) {
+          const line = buffer.slice(0, newlineIdx).trim();
+          buffer = buffer.slice(newlineIdx + 1);
+          if (line.length === 0) continue;
+          const event = JSON.parse(line);
+          if (event.phase === "error") {
+            const errData = event.data;
+            throw new Error(errData.error.message ?? "Streaming search error");
+          }
+          const searchEvent = event;
+          onPhase(searchEvent);
+          lastResponse = searchEvent.data;
+        }
+      }
+      const remaining = buffer.trim();
+      if (remaining.length > 0) {
+        const event = JSON.parse(remaining);
+        if (event.phase === "error") {
+          const errData = event.data;
+          throw new Error(errData.error.message ?? "Streaming search error");
+        }
+        const searchEvent = event;
+        onPhase(searchEvent);
+        lastResponse = searchEvent.data;
+      }
+      if (!lastResponse) {
+        throw new Error("No search results received");
+      }
+      return lastResponse;
     }
   };
 }
 
 exports.createSearchClient = createSearchClient;
+exports.mergeSearchResults = mergeSearchResults;
 //# sourceMappingURL=client.cjs.map
 //# sourceMappingURL=client.cjs.map

package/dist/client.d.cts

@@ -1,4 +1,15 @@
-import { S as SearchRequest, a as SearchResponse } from './types-BrG6XTUU.cjs';
+import { S as SearchResponse, M as MergeSearchOptions, a as MergeSearchResult, b as SearchRequest, c as StreamSearchEvent } from './types-z2dw3H6E.cjs';
+
+/**
+ * Smart merge of initial (pre-rerank) and reranked search results.
+ *
+ * If the reranker barely changed the ordering, we keep the initial order
+ * (which the user already saw) and just update scores from the reranked response.
+ * If the reranker moved any single result by more than `maxDisplacement`
+ * positions, we adopt the reranked order — the reranker is semantic and
+ * expensive, so if it strongly disagrees on even one result, trust it.
+ */
+declare function mergeSearchResults(initial: SearchResponse, reranked: SearchResponse, options?: MergeSearchOptions): MergeSearchResult;
 
 interface SearchClientOptions {
     endpoint?: string;
@@ -6,6 +17,10 @@ interface SearchClientOptions {
 }
 declare function createSearchClient(options?: SearchClientOptions): {
     search(request: SearchRequest): Promise<SearchResponse>;
+    streamSearch(request: SearchRequest & {
+        stream: true;
+        rerank: true;
+    }, onPhase: (event: StreamSearchEvent) => void): Promise<SearchResponse>;
 };
 
-export { type SearchClientOptions, createSearchClient };
+export { type SearchClientOptions, createSearchClient, mergeSearchResults };

package/dist/client.d.ts

@@ -1,4 +1,15 @@
-import { S as SearchRequest, a as SearchResponse } from './types-BrG6XTUU.js';
+import { S as SearchResponse, M as MergeSearchOptions, a as MergeSearchResult, b as SearchRequest, c as StreamSearchEvent } from './types-z2dw3H6E.js';
+
+/**
+ * Smart merge of initial (pre-rerank) and reranked search results.
+ *
+ * If the reranker barely changed the ordering, we keep the initial order
+ * (which the user already saw) and just update scores from the reranked response.
+ * If the reranker moved any single result by more than `maxDisplacement`
+ * positions, we adopt the reranked order — the reranker is semantic and
+ * expensive, so if it strongly disagrees on even one result, trust it.
+ */
+declare function mergeSearchResults(initial: SearchResponse, reranked: SearchResponse, options?: MergeSearchOptions): MergeSearchResult;
 
 interface SearchClientOptions {
     endpoint?: string;
@@ -6,6 +17,10 @@ interface SearchClientOptions {
 }
 declare function createSearchClient(options?: SearchClientOptions): {
     search(request: SearchRequest): Promise<SearchResponse>;
+    streamSearch(request: SearchRequest & {
+        stream: true;
+        rerank: true;
+    }, onPhase: (event: StreamSearchEvent) => void): Promise<SearchResponse>;
 };
 
-export { type SearchClientOptions, createSearchClient };
+export { type SearchClientOptions, createSearchClient, mergeSearchResults };

package/dist/client.js

@@ -1,3 +1,57 @@
+// src/merge.ts
+function mergeSearchResults(initial, reranked, options) {
+  const maxDisplacement = options?.maxDisplacement ?? 3;
+  const initialUrls = initial.results.map((r) => r.url);
+  const rerankedUrls = reranked.results.map((r) => r.url);
+  const initialPos = /* @__PURE__ */ new Map();
+  for (let i = 0; i < initialUrls.length; i++) {
+    initialPos.set(initialUrls[i], i);
+  }
+  const rerankedPos = /* @__PURE__ */ new Map();
+  for (let i = 0; i < rerankedUrls.length; i++) {
+    rerankedPos.set(rerankedUrls[i], i);
+  }
+  const displacements = [];
+  for (const url of initialUrls) {
+    const iPos = initialPos.get(url);
+    const rPos = rerankedPos.get(url);
+    const displacement = rPos !== void 0 ? Math.abs(iPos - rPos) : 0;
+    displacements.push({ url, displacement });
+  }
+  const totalResults = displacements.length;
+  if (totalResults === 0) {
+    return {
+      response: reranked,
+      usedRerankedOrder: true,
+      displacements
+    };
+  }
+  const hasLargeDisplacement = displacements.some((d) => d.displacement > maxDisplacement);
+  if (hasLargeDisplacement) {
+    return {
+      response: reranked,
+      usedRerankedOrder: true,
+      displacements
+    };
+  }
+  const rerankedScoreMap = /* @__PURE__ */ new Map();
+  for (const result of reranked.results) {
+    rerankedScoreMap.set(result.url, result.score);
+  }
+  const mergedResults = initial.results.map((result) => ({
+    ...result,
+    score: rerankedScoreMap.get(result.url) ?? result.score
+  }));
+  return {
+    response: {
+      ...reranked,
+      results: mergedResults
+    },
+    usedRerankedOrder: false,
+    displacements
+  };
+}
+
 // src/client.ts
 function createSearchClient(options = {}) {
   const endpoint = options.endpoint ?? "/api/search";
@@ -25,10 +79,76 @@ function createSearchClient(options = {}) {
         throw new Error(message);
       }
       return payload;
+    },
+    async streamSearch(request, onPhase) {
+      const response = await fetchImpl(endpoint, {
+        method: "POST",
+        headers: {
+          "content-type": "application/json"
+        },
+        body: JSON.stringify(request)
+      });
+      if (!response.ok) {
+        let payload;
+        try {
+          payload = await response.json();
+        } catch {
+          throw new Error("Search failed");
+        }
+        const message = payload.error?.message ?? "Search failed";
+        throw new Error(message);
+      }
+      const contentType = response.headers.get("content-type") ?? "";
+      if (contentType.includes("application/json")) {
+        const data = await response.json();
+        onPhase({ phase: "initial", data });
+        return data;
+      }
+      if (!response.body) {
+        throw new Error("Response body is not readable");
+      }
+      const reader = response.body.getReader();
+      const decoder = new TextDecoder();
+      let buffer = "";
+      let lastResponse = null;
+      for (; ; ) {
+        const { done, value } = await reader.read();
+        if (done) break;
+        buffer += decoder.decode(value, { stream: true });
+        let newlineIdx;
+        while ((newlineIdx = buffer.indexOf("\n")) !== -1) {
+          const line = buffer.slice(0, newlineIdx).trim();
+          buffer = buffer.slice(newlineIdx + 1);
+          if (line.length === 0) continue;
+          const event = JSON.parse(line);
+          if (event.phase === "error") {
+            const errData = event.data;
+            throw new Error(errData.error.message ?? "Streaming search error");
+          }
+          const searchEvent = event;
+          onPhase(searchEvent);
+          lastResponse = searchEvent.data;
+        }
+      }
+      const remaining = buffer.trim();
+      if (remaining.length > 0) {
+        const event = JSON.parse(remaining);
+        if (event.phase === "error") {
+          const errData = event.data;
+          throw new Error(errData.error.message ?? "Streaming search error");
+        }
+        const searchEvent = event;
+        onPhase(searchEvent);
+        lastResponse = searchEvent.data;
+      }
+      if (!lastResponse) {
+        throw new Error("No search results received");
+      }
+      return lastResponse;
     }
   };
 }
 
-export { createSearchClient };
+export { createSearchClient, mergeSearchResults };
 //# sourceMappingURL=client.js.map
 //# sourceMappingURL=client.js.map