From 88dcdbfb2d12e1cc0cdee9f9708734e29ff25c48 Mon Sep 17 00:00:00 2001
From: Hongbo Wu <hongbo@omnivore.app>
Date: Mon, 27 May 2024 21:29:15 +0800
Subject: [PATCH] update score api client

---
 packages/api/src/jobs/score_library_item.ts | 3 ++-
 packages/api/src/jobs/update_home.ts        | 3 ++-
 packages/api/src/services/score.ts          | 5 +++--
 3 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/packages/api/src/jobs/score_library_item.ts b/packages/api/src/jobs/score_library_item.ts
index 3e001bf45..75f71f9dc 100644
--- a/packages/api/src/jobs/score_library_item.ts
+++ b/packages/api/src/jobs/score_library_item.ts
@@ -42,6 +42,7 @@ export const scoreLibraryItem = async (
 
   const itemFeatures = {
     [libraryItem.id]: {
+      library_item_id: libraryItem.id,
       title: libraryItem.title,
       has_thumbnail: !!libraryItem.thumbnail,
       has_site_icon: !!libraryItem.siteIcon,
@@ -59,7 +60,7 @@ export const scoreLibraryItem = async (
 
   const scores = await getScores({
     user_id: userId,
-    item_features: itemFeatures,
+    items: itemFeatures,
   })
 
   logger.info('Scores', scores)
diff --git a/packages/api/src/jobs/update_home.ts b/packages/api/src/jobs/update_home.ts
index 70d54f097..4189def79 100644
--- a/packages/api/src/jobs/update_home.ts
+++ b/packages/api/src/jobs/update_home.ts
@@ -166,8 +166,9 @@ const rankCandidates = async (
 
   const data = {
     user_id: userId,
-    item_features: unscoredCandidates.reduce((acc, item) => {
+    items: unscoredCandidates.reduce((acc, item) => {
       acc[item.id] = {
+        library_item_id: item.id,
         title: item.title,
         has_thumbnail: !!item.thumbnail,
         has_site_icon: !!item.siteIcon,
diff --git a/packages/api/src/services/score.ts b/packages/api/src/services/score.ts
index 4a4bbf6d5..9d41fead2 100644
--- a/packages/api/src/services/score.ts
+++ b/packages/api/src/services/score.ts
@@ -1,4 +1,5 @@
 export interface Feature {
+  library_item_id?: string
   title: string
   has_thumbnail: boolean
   has_site_icon: boolean
@@ -15,7 +16,7 @@ export interface Feature {
 
 export interface ScoreApiRequestBody {
   user_id: string
-  item_features: Record<string, Feature> // item_id -> feature
+  items: Record<string, Feature> // item_id -> feature
 }
 
 export type ScoreApiResponse = Record<string, number> // item_id -> score
@@ -23,7 +24,7 @@ export type ScoreApiResponse = Record<string, number> // item_id -> score
 export const getScores = async (
   data: ScoreApiRequestBody
 ): Promise<ScoreApiResponse> => {
-  const API_URL = 'http://127.0.0.1:5000/predictions'
+  const API_URL = 'http://digest-score/batch'
   // const token = process.env.SCORE_API_TOKEN
 
   // if (!token) {