fix: convert milliseconds to seconds.

ainft-team · Oct 23, 2024 · b65b86d · b65b86d
1 parent ff8916f
commit b65b86d
Show file tree

Hide file tree

Showing 2 changed files with 64 additions and 47 deletions.
diff --git a/src/ai/thread.ts b/src/ai/thread.ts
@@ -15,7 +15,7 @@ import {
 } from '../types';
 import { Path } from '../utils/path';
 import { buildSetValueOp, buildSetTxBody, sendTx } from '../utils/transaction';
-import { getAssistant, getValue } from '../utils/util';
+import { getAssistant, getValue, isMillisecond, toSecond } from '../utils/util';
 import {
   validateAssistant,
   validateObject,
@@ -203,14 +203,13 @@ export class Threads extends FactoryBase {
       .history(address)
       .thread(threadId)
       .value();
-    const data = await this.ain.db.ref(threadPath).getValue();
-    const thread = {
-      id: data.id,
-      metadata: data.metadata || {},
-      createdAt: data.createdAt,
-    };
+    const thread = await this.ain.db.ref(threadPath).getValue();
 
-    return thread;
+    return {
+      id: thread.id,
+      createdAt: isMillisecond(thread.createdAt) ? toSecond(thread.createdAt) : thread.createdAt,
+      metadata: thread.metadata || {},
+    };
   }
 
   /**
@@ -307,44 +306,48 @@ export class Threads extends FactoryBase {
 
   private async flattenThreads(objectId: string, tokens: any) {
     const flatten: any = [];
-    await Promise.all(_.map(tokens, async (token, tokenId) => {
-      if (!token.ai) {
-        return;
-      }
-      const assistantId = token.ai.id;
-      const assistant = await getAssistant(this.ain, objectId, tokenId, assistantId);
-      const histories = token.ai.history;
-      if (typeof histories !== 'object' || histories === true) {
-        return;
-      }
-      _.forEach(histories, (history, address) => {
-        const threads = _.get(history, 'threads');
-        _.forEach(threads, (thread) => {
-          let updatedAt = thread.createdAt;
-          if (typeof thread.messages === 'object' && thread.messages !== null) {
-            const keys = Object.keys(thread.messages);
-            updatedAt = Number(keys[keys.length - 1]);
-          }
-          flatten.push({
-            id: thread.id,
-            metadata: thread.metadata || {},
-            createdAt: thread.createdAt,
-            updatedAt,
-            assistant: {
-              id: assistant.id,
-              createdAt: assistant.createdAt,
-              objectId,
-              tokenId,
-              tokenOwner: token.owner,
-              model: assistant.model,
-              name: assistant.name,
-              description: assistant.description || null,
-              instructions: assistant.instructions || null,
-              metadata: assistant.metadata || {},
-              metrics: assistant.metrics || {},
-            },
-            author: { address },
-          });
+    await Promise.all(
+      _.map(tokens, async (token, tokenId) => {
+        if (!token.ai) {
+          return;
+        }
+        const assistantId = token.ai.id;
+        const assistant = await getAssistant(this.ain, objectId, tokenId, assistantId);
+        const histories = token.ai.history;
+        if (typeof histories !== 'object' || histories === true) {
+          return;
+        }
+        _.forEach(histories, (history, address) => {
+          const threads = _.get(history, 'threads');
+          _.forEach(threads, (thread) => {
+            const createdAt = isMillisecond(thread.createdAt)
+              ? toSecond(thread.createdAt)
+              : thread.createdAt;
+            let updatedAt = createdAt;
+            if (typeof thread.messages === 'object' && thread.messages !== null) {
+              const keys = Object.keys(thread.messages);
+              updatedAt = Number(keys[keys.length - 1]);
+            }
+            flatten.push({
+              id: thread.id,
+              metadata: thread.metadata || {},
+              createdAt: createdAt,
+              updatedAt,
+              assistant: {
+                id: assistant.id,
+                createdAt: assistant.createdAt,
+                objectId,
+                tokenId,
+                tokenOwner: token.owner,
+                model: assistant.model,
+                name: assistant.name,
+                description: assistant.description || null,
+                instructions: assistant.instructions || null,
+                metadata: assistant.metadata || {},
+                metrics: assistant.metrics || {},
+              },
+              author: { address },
+            });
           });
         });
       })

diff --git a/src/utils/util.ts b/src/utils/util.ts
@@ -67,7 +67,13 @@ export const getAssistant = async (
   if (!assistant) {
     return null;
   }
-  return { ...assistant, tokenOwner: token.owner };
+  return {
+    ...assistant,
+    createdAt: isMillisecond(assistant.createdAt)
+      ? toSecond(assistant.createdAt)
+      : assistant.createdAt,
+    tokenOwner: token.owner,
+  };
 };
 
 export const getToken = async (ain: Ain, objectId: string, tokenId: string) => {
@@ -99,3 +105,11 @@ export const arrayToObject = <T>(array: T[]): { [key: string]: T } => {
   });
   return result;
 };
+
+export const isMillisecond = (timestamp: number) => {
+  return timestamp.toString().length === 13;
+};
+
+export const toSecond = (millisecond: number) => {
+  return Math.floor(millisecond / 1000);
+};