Adapt scores to return proper Score objects

Chainlit · Apr 9, 2024 · 16be0b5 · 16be0b5
1 parent 1b209d4
commit 16be0b5
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/api.ts b/src/api.ts
@@ -949,11 +949,12 @@ export class API {
     return response;
   }
 
-  async createScores(scores: Score[]) {
+  async createScores(scores: Score[]): Promise<Score[]> {
     const query = createScoresQueryBuilder(scores);
     const variables = variablesBuilder(scores);
 
-    return this.makeGqlCall(query, variables);
+    const result = await this.makeGqlCall(query, variables);
+    return Object.values(result.data).map((x: any) => new Score(x));
   }
 
   async createScore(variables: OmitUtils<Score>) {