Updates example scripts

galadriel-ai · Jul 2, 2024 · 5f66aff · 5f66aff
1 parent 40badf1
commit 5f66aff
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 16 deletions.
diff --git a/examples/agent.ts b/examples/agent.ts
@@ -112,16 +112,17 @@ async function getNewMessages(
   agentRunID: number,
   currentMessagesCount: number
 ): Promise<Message[]> {
-  const messages = await contract.getMessageHistoryContents(agentRunID)
-  const roles = await contract.getMessageHistoryRoles(agentRunID)
+  const messages = await contract.getMessageHistory(agentRunID)
 
   const newMessages: Message[] = []
   messages.forEach((message: any, i: number) => {
     if (i >= currentMessagesCount) {
-      newMessages.push({
-        role: roles[i],
-        content: messages[i]
-      })
+      newMessages.push(
+        {
+          role: message.role,
+          content: message.content[0].value,
+        }
+      );
     }
   })
   return newMessages;

diff --git a/examples/chat.ts b/examples/chat.ts
@@ -107,16 +107,17 @@ async function getNewMessages(
   chatId: number,
   currentMessagesCount: number
 ): Promise<Message[]> {
-  const messages = await contract.getMessageHistoryContents(chatId)
-  const roles = await contract.getMessageHistoryRoles(chatId)
+  const messages = await contract.getMessageHistory(chatId)
 
   const newMessages: Message[] = []
   messages.forEach((message: any, i: number) => {
     if (i >= currentMessagesCount) {
-      newMessages.push({
-        role: roles[i],
-        content: messages[i]
-      })
+      newMessages.push(
+        {
+          role: message[0],
+          content: message.content[0].value,
+        }
+      );
     }
   })
   return newMessages;

diff --git a/examples/chat_anthropic.ts b/examples/chat_anthropic.ts
@@ -121,8 +121,8 @@ async function getNewMessages(
     if (i >= currentMessagesCount) {
       newMessages.push(
         {
-          role: message[0],
-          content: message[1][0][1],
+          role: message.role,
+          content: message.content[0].value,
         }
       );
     }

diff --git a/examples/chat_vision.ts b/examples/chat_vision.ts
@@ -122,8 +122,8 @@ async function getNewMessages(
     if (i >= currentMessagesCount) {
       newMessages.push(
         {
-          role: message[0],
-          content: message[1][0][1],
+          role: message.role,
+          content: message.content[0].value,
         }
       );
     }