Code added for embeddings in hyde-example

arakoodev · Sep 25, 2023 · 9d9e2a2 · 9d9e2a2
1 parent 8ddac6e
commit 9d9e2a2
Show file tree

Hide file tree

Showing 6 changed files with 334 additions and 93 deletions.
diff --git a/JS/edgechains/examples/ormconfig.json b/JS/edgechains/examples/ormconfig.json
@@ -0,0 +1,12 @@
+{
+    "type": "postgres",
+    "host": "db.rmzqtepwnzoxgkkzjctt.supabase.co",
+    "port": 5432,
+    "username": "postgres",
+    "password": "xaX0MYcf1YiJlChK",
+    "database": "postgres",
+    "entities": ["dist/entities/**/*.js"],
+    "synchronize": false,
+    "logging": false
+  }
+
diff --git a/JS/edgechains/examples/package-lock.json b/JS/edgechains/examples/package-lock.json
diff --git a/JS/edgechains/examples/package.json b/JS/edgechains/examples/package.json
@@ -5,7 +5,6 @@
   "author": "",
   "private": true,
   "license": "UNLICENSED",
-
   "scripts": {
     "build": "nest build",
     "clean": "rm -rf coverage dist .nyc_output",
@@ -27,6 +26,7 @@
     "@nestjs/core": "^10.0.0",
     "@nestjs/platform-express": "^10.0.0",
     "@nestjs/typeorm": "^10.0.0",
+    "pg": "^8.11.3",
     "reflect-metadata": "^0.1.13",
     "rxjs": "^7.8.1"
   },

diff --git a/JS/edgechains/examples/src/app.controller.ts b/JS/edgechains/examples/src/app.controller.ts
@@ -14,13 +14,14 @@ export class AppController {
   }
 
   @Post("/hyde-search")
-  hydeSearch(@Query() params:any, @Body() query:string){
+  hydeSearch(@Query() params:any, @Body('query') query: string){
     const arkRequest = {
       tableName : params.table,
       nameSpace : params.namespace,
       query : query,
       topK : params.topK
     }
+
     hydeSearchAdaEmbedding(arkRequest);
 
   }