searchsocket 0.4.0 → 0.5.0

package/dist/index.d.cts

@@ -1,7 +1,8 @@
-import { R as ResolvedSearchSocketConfig, d as SearchSocketConfig, e as Scope, E as EmbeddingsProvider, f as Reranker, g as RerankCandidate, V as VectorStore, I as IndexOptions, h as IndexStats, b as SearchRequest, S as SearchResponse, c as StreamSearchEvent } from './types-z2dw3H6E.cjs';
-export { C as Chunk, M as MergeSearchOptions, a as MergeSearchResult, Q as QueryOpts, i as StreamEvent, j as StreamSearchErrorEvent, k as VectorHit, l as VectorRecord } from './types-z2dw3H6E.cjs';
-export { searchsocketHandle, searchsocketVitePlugin } from './sveltekit.cjs';
-export { createSearchClient, mergeSearchResults } from './client.cjs';
+import { R as ResolvedSearchSocketConfig, c as SearchSocketConfig, d as Scope, V as VectorHit, P as PageHit, e as ScopeInfo, f as PageRecord, I as IndexOptions, g as IndexStats, S as SearchRequest, a as SearchResponse } from './types-Dk43uz25.cjs';
+export { C as Chunk } from './types-Dk43uz25.cjs';
+import { Search } from '@upstash/search';
+export { s as searchsocketHandle, b as searchsocketVitePlugin } from './plugin-B_npJSux.cjs';
+export { createSearchClient } from './client.cjs';
 
 interface LoadConfigOptions {
     cwd?: string;
@@ -28,26 +29,71 @@ declare function isServerless(): boolean;
 
 declare function resolveScope(config: ResolvedSearchSocketConfig, override?: string): Scope;
 
-declare function createEmbeddingsProvider(config: ResolvedSearchSocketConfig): EmbeddingsProvider;
-
-interface JinaRerankerOptions {
-    apiKey: string;
-    model: string;
-    maxRetries?: number;
+/** Content fields stored in Upstash Search (indexed and searchable) */
+interface ChunkContent {
+    title: string;
+    sectionTitle: string;
+    text: string;
+    url: string;
+    tags: string;
+    headingPath: string;
+    [key: string]: unknown;
+}
+/** Metadata fields stored in Upstash Search (returned but not searchable) */
+interface ChunkMetadata {
+    projectId: string;
+    scopeName: string;
+    path: string;
+    snippet: string;
+    ordinal: number;
+    contentHash: string;
+    depth: number;
+    incomingLinks: number;
+    routeFile: string;
+    description: string;
+    keywords: string;
+    [key: string]: unknown;
 }
-declare class JinaReranker implements Reranker {
-    private readonly apiKey;
-    private readonly model;
-    private readonly maxRetries;
-    constructor(options: JinaRerankerOptions);
-    rerank(query: string, candidates: RerankCandidate[], topN?: number): Promise<Array<{
+interface UpstashSearchStoreOptions {
+    client: Search;
+}
+declare class UpstashSearchStore {
+    private readonly client;
+    constructor(opts: UpstashSearchStoreOptions);
+    private chunkIndex;
+    private pageIndex;
+    upsertChunks(chunks: Array<{
         id: string;
-        score: number;
-    }>>;
+        content: ChunkContent;
+        metadata: ChunkMetadata;
+    }>, scope: Scope): Promise<void>;
+    search(query: string, opts: {
+        limit: number;
+        semanticWeight?: number;
+        inputEnrichment?: boolean;
+        reranking?: boolean;
+        filter?: string;
+    }, scope: Scope): Promise<VectorHit[]>;
+    searchPages(query: string, opts: {
+        limit: number;
+        semanticWeight?: number;
+        inputEnrichment?: boolean;
+        filter?: string;
+    }, scope: Scope): Promise<PageHit[]>;
+    deleteByIds(ids: string[], scope: Scope): Promise<void>;
+    deleteScope(scope: Scope): Promise<void>;
+    listScopes(projectId: string): Promise<ScopeInfo[]>;
+    getContentHashes(scope: Scope): Promise<Map<string, string>>;
+    upsertPages(pages: PageRecord[], scope: Scope): Promise<void>;
+    getPage(url: string, scope: Scope): Promise<PageRecord | null>;
+    deletePages(scope: Scope): Promise<void>;
+    health(): Promise<{
+        ok: boolean;
+        details?: string;
+    }>;
+    dropAllIndexes(projectId: string): Promise<void>;
 }
 
-declare function createReranker(config: ResolvedSearchSocketConfig): Reranker | null;
-
 interface LoggerOptions {
     json?: boolean;
     verbose?: boolean;
@@ -74,15 +120,13 @@ interface IndexPipelineOptions {
     cwd?: string;
     configPath?: string;
     config?: ResolvedSearchSocketConfig;
-    embeddingsProvider?: EmbeddingsProvider;
-    vectorStore?: VectorStore;
+    store?: UpstashSearchStore;
     logger?: Logger;
 }
 declare class IndexPipeline {
     private readonly cwd;
     private readonly config;
-    private readonly embeddings;
-    private readonly vectorStore;
+    private readonly store;
     private readonly logger;
     private constructor();
     static create(options?: IndexPipelineOptions): Promise<IndexPipeline>;
@@ -94,21 +138,17 @@ interface SearchEngineOptions {
     cwd?: string;
     configPath?: string;
     config?: ResolvedSearchSocketConfig;
-    embeddingsProvider?: EmbeddingsProvider;
-    vectorStore?: VectorStore;
-    reranker?: Reranker | null;
+    store?: UpstashSearchStore;
 }
 declare class SearchEngine {
     private readonly cwd;
     private readonly config;
-    private readonly embeddings;
-    private readonly vectorStore;
-    private readonly reranker;
+    private readonly store;
     private constructor();
     static create(options?: SearchEngineOptions): Promise<SearchEngine>;
     getConfig(): ResolvedSearchSocketConfig;
     search(request: SearchRequest): Promise<SearchResponse>;
-    searchStreaming(request: SearchRequest): AsyncGenerator<StreamSearchEvent>;
+    private ensureSnippet;
     private buildResults;
     getPage(pathOrUrl: string, scope?: string): Promise<{
         url: string;
@@ -120,11 +160,9 @@ declare class SearchEngine {
         details?: string;
     }>;
     private resolveInputPath;
-    private assertModelCompatibility;
-    private rerankHits;
 }
 
-declare function createVectorStore(config: ResolvedSearchSocketConfig, cwd: string): Promise<VectorStore>;
+declare function createUpstashStore(config: ResolvedSearchSocketConfig): Promise<UpstashSearchStore>;
 
 interface McpServerOptions {
     cwd?: string;
@@ -135,4 +173,4 @@ interface McpServerOptions {
 }
 declare function runMcpServer(options?: McpServerOptions): Promise<void>;
 
-export { EmbeddingsProvider, IndexOptions, IndexPipeline, IndexStats, JinaReranker, RerankCandidate, Reranker, ResolvedSearchSocketConfig, Scope, SearchEngine, SearchRequest, SearchResponse, SearchSocketConfig, StreamSearchEvent, VectorStore, createEmbeddingsProvider, createReranker, createVectorStore, isServerless, loadConfig, mergeConfig, mergeConfigServerless, resolveScope, runMcpServer };
+export { IndexOptions, IndexPipeline, IndexStats, ResolvedSearchSocketConfig, Scope, SearchEngine, SearchRequest, SearchResponse, SearchSocketConfig, UpstashSearchStore, VectorHit, createUpstashStore, isServerless, loadConfig, mergeConfig, mergeConfigServerless, resolveScope, runMcpServer };

package/dist/index.d.ts

@@ -1,7 +1,8 @@
-import { R as ResolvedSearchSocketConfig, d as SearchSocketConfig, e as Scope, E as EmbeddingsProvider, f as Reranker, g as RerankCandidate, V as VectorStore, I as IndexOptions, h as IndexStats, b as SearchRequest, S as SearchResponse, c as StreamSearchEvent } from './types-z2dw3H6E.js';
-export { C as Chunk, M as MergeSearchOptions, a as MergeSearchResult, Q as QueryOpts, i as StreamEvent, j as StreamSearchErrorEvent, k as VectorHit, l as VectorRecord } from './types-z2dw3H6E.js';
-export { searchsocketHandle, searchsocketVitePlugin } from './sveltekit.js';
-export { createSearchClient, mergeSearchResults } from './client.js';
+import { R as ResolvedSearchSocketConfig, c as SearchSocketConfig, d as Scope, V as VectorHit, P as PageHit, e as ScopeInfo, f as PageRecord, I as IndexOptions, g as IndexStats, S as SearchRequest, a as SearchResponse } from './types-Dk43uz25.js';
+export { C as Chunk } from './types-Dk43uz25.js';
+import { Search } from '@upstash/search';
+export { s as searchsocketHandle, b as searchsocketVitePlugin } from './plugin-M-aW0ev6.js';
+export { createSearchClient } from './client.js';
 
 interface LoadConfigOptions {
     cwd?: string;
@@ -28,26 +29,71 @@ declare function isServerless(): boolean;
 
 declare function resolveScope(config: ResolvedSearchSocketConfig, override?: string): Scope;
 
-declare function createEmbeddingsProvider(config: ResolvedSearchSocketConfig): EmbeddingsProvider;
-
-interface JinaRerankerOptions {
-    apiKey: string;
-    model: string;
-    maxRetries?: number;
+/** Content fields stored in Upstash Search (indexed and searchable) */
+interface ChunkContent {
+    title: string;
+    sectionTitle: string;
+    text: string;
+    url: string;
+    tags: string;
+    headingPath: string;
+    [key: string]: unknown;
+}
+/** Metadata fields stored in Upstash Search (returned but not searchable) */
+interface ChunkMetadata {
+    projectId: string;
+    scopeName: string;
+    path: string;
+    snippet: string;
+    ordinal: number;
+    contentHash: string;
+    depth: number;
+    incomingLinks: number;
+    routeFile: string;
+    description: string;
+    keywords: string;
+    [key: string]: unknown;
 }
-declare class JinaReranker implements Reranker {
-    private readonly apiKey;
-    private readonly model;
-    private readonly maxRetries;
-    constructor(options: JinaRerankerOptions);
-    rerank(query: string, candidates: RerankCandidate[], topN?: number): Promise<Array<{
+interface UpstashSearchStoreOptions {
+    client: Search;
+}
+declare class UpstashSearchStore {
+    private readonly client;
+    constructor(opts: UpstashSearchStoreOptions);
+    private chunkIndex;
+    private pageIndex;
+    upsertChunks(chunks: Array<{
         id: string;
-        score: number;
-    }>>;
+        content: ChunkContent;
+        metadata: ChunkMetadata;
+    }>, scope: Scope): Promise<void>;
+    search(query: string, opts: {
+        limit: number;
+        semanticWeight?: number;
+        inputEnrichment?: boolean;
+        reranking?: boolean;
+        filter?: string;
+    }, scope: Scope): Promise<VectorHit[]>;
+    searchPages(query: string, opts: {
+        limit: number;
+        semanticWeight?: number;
+        inputEnrichment?: boolean;
+        filter?: string;
+    }, scope: Scope): Promise<PageHit[]>;
+    deleteByIds(ids: string[], scope: Scope): Promise<void>;
+    deleteScope(scope: Scope): Promise<void>;
+    listScopes(projectId: string): Promise<ScopeInfo[]>;
+    getContentHashes(scope: Scope): Promise<Map<string, string>>;
+    upsertPages(pages: PageRecord[], scope: Scope): Promise<void>;
+    getPage(url: string, scope: Scope): Promise<PageRecord | null>;
+    deletePages(scope: Scope): Promise<void>;
+    health(): Promise<{
+        ok: boolean;
+        details?: string;
+    }>;
+    dropAllIndexes(projectId: string): Promise<void>;
 }
 
-declare function createReranker(config: ResolvedSearchSocketConfig): Reranker | null;
-
 interface LoggerOptions {
     json?: boolean;
     verbose?: boolean;
@@ -74,15 +120,13 @@ interface IndexPipelineOptions {
     cwd?: string;
     configPath?: string;
     config?: ResolvedSearchSocketConfig;
-    embeddingsProvider?: EmbeddingsProvider;
-    vectorStore?: VectorStore;
+    store?: UpstashSearchStore;
     logger?: Logger;
 }
 declare class IndexPipeline {
     private readonly cwd;
     private readonly config;
-    private readonly embeddings;
-    private readonly vectorStore;
+    private readonly store;
     private readonly logger;
     private constructor();
     static create(options?: IndexPipelineOptions): Promise<IndexPipeline>;
@@ -94,21 +138,17 @@ interface SearchEngineOptions {
     cwd?: string;
     configPath?: string;
     config?: ResolvedSearchSocketConfig;
-    embeddingsProvider?: EmbeddingsProvider;
-    vectorStore?: VectorStore;
-    reranker?: Reranker | null;
+    store?: UpstashSearchStore;
 }
 declare class SearchEngine {
     private readonly cwd;
     private readonly config;
-    private readonly embeddings;
-    private readonly vectorStore;
-    private readonly reranker;
+    private readonly store;
     private constructor();
     static create(options?: SearchEngineOptions): Promise<SearchEngine>;
     getConfig(): ResolvedSearchSocketConfig;
     search(request: SearchRequest): Promise<SearchResponse>;
-    searchStreaming(request: SearchRequest): AsyncGenerator<StreamSearchEvent>;
+    private ensureSnippet;
     private buildResults;
     getPage(pathOrUrl: string, scope?: string): Promise<{
         url: string;
@@ -120,11 +160,9 @@ declare class SearchEngine {
         details?: string;
     }>;
     private resolveInputPath;
-    private assertModelCompatibility;
-    private rerankHits;
 }
 
-declare function createVectorStore(config: ResolvedSearchSocketConfig, cwd: string): Promise<VectorStore>;
+declare function createUpstashStore(config: ResolvedSearchSocketConfig): Promise<UpstashSearchStore>;
 
 interface McpServerOptions {
     cwd?: string;
@@ -135,4 +173,4 @@ interface McpServerOptions {
 }
 declare function runMcpServer(options?: McpServerOptions): Promise<void>;
 
-export { EmbeddingsProvider, IndexOptions, IndexPipeline, IndexStats, JinaReranker, RerankCandidate, Reranker, ResolvedSearchSocketConfig, Scope, SearchEngine, SearchRequest, SearchResponse, SearchSocketConfig, StreamSearchEvent, VectorStore, createEmbeddingsProvider, createReranker, createVectorStore, isServerless, loadConfig, mergeConfig, mergeConfigServerless, resolveScope, runMcpServer };
+export { IndexOptions, IndexPipeline, IndexStats, ResolvedSearchSocketConfig, Scope, SearchEngine, SearchRequest, SearchResponse, SearchSocketConfig, UpstashSearchStore, VectorHit, createUpstashStore, isServerless, loadConfig, mergeConfig, mergeConfigServerless, resolveScope, runMcpServer };