Fix graph search result enum

SciPhi-AI · Dec 5, 2024 · f05456c · f05456c
1 parent b916121
commit f05456c
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 10 deletions.
diff --git a/js/sdk/package-lock.json b/js/sdk/package-lock.json
diff --git a/js/sdk/package.json b/js/sdk/package.json
@@ -1,6 +1,6 @@
 {
   "name": "r2r-js",
-  "version": "0.4.1",
+  "version": "0.4.2",
   "description": "",
   "main": "dist/index.js",
   "browser": "dist/index.browser.js",

diff --git a/js/sdk/src/types.ts b/js/sdk/src/types.ts
@@ -212,17 +212,19 @@ export interface VectorSearchResult {
   metadata?: Record<string, any>;
 }
 
-export interface KGSearchResult {
-  method: string;
-  content: string;
-  result_type?: string;
-  chunks_ids?: string[];
-  metadata?: Record<string, any>;
+type KGSearchResultType = "entity" | "relationship" | "community" | "global";
+
+interface GraphSearchResult {
+  content: any;
+  result_type?: KGSearchResultType;
+  chunk_ids?: string[];
+  metadata: Record<string, any>;
+  score?: number;
 }
 
 export interface CombinedSearchResponse {
-  vector_search_results: VectorSearchResult[];
-  graph_search_results?: KGSearchResult[];
+  chunk_search_results: VectorSearchResult[];
+  graph_search_results?: GraphSearchResult[];
 }
 
 // System types