@tryformation/querylight-cli 0.2.1 → 0.2.3

package/dist/types/models.d.ts

@@ -222,9 +222,44 @@ export type SearchResult = {
     lastChangedAt: string;
     metadata: Record<string, unknown>;
 };
+export type SearchHitSource = {
+    chunkId: string;
+    documentId: string;
+    sourceId: string;
+    sourceType: SourceType;
+    sourceName?: string;
+    title: string;
+    uri: string;
+    headingPath: string[];
+    text: string;
+    snippet?: string;
+    normalizedPath?: string;
+    publicationDate?: string | null;
+    crawledAt?: string;
+    firstSeenAt: string;
+    lastSeenAt: string;
+    lastChangedAt: string;
+    metadata: Record<string, unknown>;
+};
+export type SearchHit = {
+    _index: string;
+    _id: string;
+    _score: number;
+    _source: SearchHitSource;
+    highlight?: Record<string, string[]>;
+};
 export type SearchResponseData = {
     retrievalMode?: RetrievalMode;
-    results: SearchResult[];
+    took: number;
+    hits: {
+        total: {
+            value: number;
+            relation: "eq";
+        };
+        max_score: number | null;
+        hits: SearchHit[];
+    };
+    aggregations?: Record<string, unknown>;
 };
 export type RelatedDocumentResult = {
     documentId: string;

package/dist/vector/runtime.d.ts

@@ -1,4 +1,11 @@
 import type { SparseVectorModelConfig } from "../types/models.js";
+type SparseExecOptions = {
+    encoding: BufferEncoding;
+    maxBuffer: number;
+    env: NodeJS.ProcessEnv;
+};
+type SparseExecFileSync = (file: string, args: string[], options: SparseExecOptions) => string;
+export declare function setSparseExecFileSyncForTests(fn: SparseExecFileSync | null): void;
 export declare function resolveQliHomeDir(): string;
 export declare function resolveCacheDir(workspacePath: string, configuredPath: string): string;
 export declare function packageRootFromImportMeta(importMetaUrl: string): string;
@@ -18,3 +25,4 @@ export declare function getDenseTransformersRuntime(cacheDir: string): Promise<{
     };
     pipeline: typeof import("@huggingface/transformers").pipeline;
 }>;
+export {};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tryformation/querylight-cli",
-  "version": "0.2.1",
+  "version": "0.2.3",
   "description": "Querylight CLI for building and querying local knowledge bases.",
   "license": "MIT",
   "homepage": "https://github.com/formation-res/querylight-cli#readme",
@@ -40,7 +40,7 @@
   },
   "dependencies": {
     "@huggingface/transformers": "^3.8.1",
-    "@tryformation/querylight-ts": "^0.10.0",
+    "@tryformation/querylight-ts": "^0.11.0",
     "cheerio": "^1.2.0",
     "cli-table3": "^0.6.5",
     "commander": "^14.0.3",

package/scripts/sparse-encode.py

@@ -88,7 +88,11 @@ def encode_documents(model_id: str, top_tokens: int, documents):
 
 
 def main():
-    payload = json.load(sys.stdin)
+    if len(sys.argv) > 1:
+        with open(sys.argv[1], encoding="utf-8") as handle:
+            payload = json.load(handle)
+    else:
+        payload = json.load(sys.stdin)
     action = payload["action"]
     model_id = payload["model_id"]
     if action == "download_only":