Predictions

docker-compose.yaml

@@ -1,19 +1,24 @@
 services:
     tee:
+        image: bino:latest
         command: ["pnpm", "start", "--character=${CHARACTER_FILE}"]
-        build:
-            context: .
-            dockerfile: Dockerfile
         stdin_open: true
         tty: true
         volumes:
             - /var/run/tappd.sock:/var/run/tappd.sock
             - ./characters:/app/characters
+            # If we want to mount a local directory into the container
+            # - type: bind
+            #   source: /root/iotex2-docs
+            #   target: /app/characters/knowledge/iotex2-docs
+            #   bind:
+            #     create_host_path: true
         env_file:
             - .env
         environment:
             - CHARACTER_FILE=${CHARACTER_FILE}
             - NODE_OPTIONS=--max-old-space-size=8192
+            - PORT=${PORT}
         ports:
-            - "3000"
-        restart: always
+            - "${PORT}:3000"
+        restart: always

packages/adapter-postgres/src/index.ts

@@ -1,7 +1,7 @@
 import { v4 } from "uuid";
 
 // Import the entire module as default
-import pg, { PoolConfig} from "pg";
+import pg, { PoolConfig } from "pg";
 type Pool = pg.Pool;
 
 import {
@@ -11,6 +11,7 @@ import {
     EmbeddingProvider,
     GoalStatus,
     Participant,
+    Prediction,
     RAGKnowledgeItem,
     elizaLogger,
     getEmbeddingConfig,
@@ -1783,6 +1784,50 @@ export class PostgresDatabaseAdapter
             ]
         );
     }
+
+    async createPrediction(prediction: Prediction): Promise<void> {
+        await this.pool.query(
+            `INSERT INTO predictions (id, creator, statement, deadline, contract_address, status, smartcontract_id) VALUES ($1, $2, $3, $4, $5, $6, $7)`,
+            [
+                prediction.id,
+                prediction.creator,
+                prediction.statement,
+                prediction.deadline,
+                prediction.contract_address,
+                prediction.status,
+                prediction.smartcontract_id,
+            ]
+        );
+    }
+
+    async getPredictions(params: {
+        status: "OPEN" | "RESOLVED" | "CLOSED";
+    }): Promise<Prediction[]> {
+        const { rows } = await this.pool.query(
+            `SELECT * FROM predictions WHERE "status" = $1`,
+            [params.status]
+        );
+        return rows;
+    }
+
+    async getReadyActivePredictions(): Promise<Prediction[]> {
+        const currentTime = new Date();
+        const { rows } = await this.pool.query(
+            `SELECT * FROM predictions WHERE "status" = 'OPEN' AND "deadline" < $1`,
+            [currentTime]
+        );
+        return rows;
+    }
+
+    async resolvePrediction(
+        predictionId: string,
+        outcome: boolean
+    ): Promise<void> {
+        await this.pool.query(
+            `UPDATE predictions SET "status" = 'RESOLVED', "outcome" = $2 WHERE "id" = $1`,
+            [predictionId, outcome]
+        );
+    }
 }
 
-export default PostgresDatabaseAdapter;
+export default PostgresDatabaseAdapter;

packages/client-auto/src/index.ts

@@ -1,4 +1,4 @@
-import { Client, IAgentRuntime, elizaLogger } from "@elizaos/core";
+import { Client, elizaLogger, IAgentRuntime } from "@elizaos/core";
 
 export class AutoClient {
     interval: NodeJS.Timeout;
@@ -10,7 +10,7 @@ export class AutoClient {
         // start a loop that runs every x seconds
         this.interval = setInterval(
             async () => {
-                elizaLogger.log("running auto client...");
+                elizaLogger.debug("running auto client...");
             },
             60 * 60 * 1000
         ); // 1 hour in milliseconds

packages/client-direct/src/index.ts

@@ -253,7 +253,7 @@ export class DirectClient {
                 const response = await generateMessageResponse({
                     runtime: runtime,
                     context,
-                    modelClass: ModelClass.LARGE,
+                    modelClass: ModelClass.SMALL,
                 });
 
                 if (!response) {
@@ -519,7 +519,7 @@ export class DirectClient {
                 const response = await generateMessageResponse({
                     runtime: runtime,
                     context,
-                    modelClass: ModelClass.LARGE,
+                    modelClass: ModelClass.SMALL,
                 });
 
                 // save response to memory

packages/client-discord/src/actions/chat_with_attachments.ts

@@ -1,4 +1,4 @@
-import { composeContext, getModelSettings } from "@elizaos/core";
+import { composeContext, getModelSettings, parseTagContent } from "@elizaos/core";
 import { generateText, trimTokens } from "@elizaos/core";
 import { models } from "@elizaos/core";
 import { parseJSONObjectFromText } from "@elizaos/core";
@@ -32,12 +32,12 @@ The "objective" is a detailed description of what the user wants to summarize ba
 The "attachmentIds" is an array of attachment IDs that the user wants to summarize. If not specified, default to including all attachments from the conversation.
 
 Your response must be formatted as a JSON block with this structure:
-\`\`\`json
+<response>
 {
   "objective": "<What the user wants to summarize>",
   "attachmentIds": ["<Attachment ID 1>", "<Attachment ID 2>", ...]
 }
-\`\`\`
+</response>
 `;
 
 const getAttachmentIds = async (
@@ -60,7 +60,8 @@ const getAttachmentIds = async (
         });
         console.log("response", response);
         // try parsing to a json object
-        const parsedResponse = parseJSONObjectFromText(response) as {
+        const extractedResponse = parseTagContent(response, "response");
+        const parsedResponse = parseJSONObjectFromText(extractedResponse) as {
             objective: string;
             attachmentIds: string[];
         } | null;

packages/client-discord/src/actions/download_media.ts

@@ -1,5 +1,5 @@
 import path from "path";
-import { composeContext } from "@elizaos/core";
+import { composeContext, parseTagContent } from "@elizaos/core";
 import { parseJSONObjectFromText } from "@elizaos/core";
 import {
     Action,
@@ -22,11 +22,11 @@ export const mediaUrlTemplate = `# Messages we are searching for a media URL
 The "mediaUrl" is the URL of the media file that the user wants downloaded. If not specified, return null.
 
 Your response must be formatted as a JSON block with this structure:
-\`\`\`json
+<response>
 {
   "mediaUrl": "<Media URL>"
 }
-\`\`\`
+</response>
 `;
 
 const getMediaUrl = async (
@@ -50,7 +50,8 @@ const getMediaUrl = async (
             modelClass: ModelClass.SMALL,
         });
 
-        const parsedResponse = parseJSONObjectFromText(response) as {
+        const extractedResponse = parseTagContent(response, "response");
+        const parsedResponse = parseJSONObjectFromText(extractedResponse) as {
             mediaUrl: string;
         } | null;
 

packages/client-discord/src/actions/summarize_conversation.ts

@@ -1,4 +1,4 @@
-import { composeContext, getModelSettings } from "@elizaos/core";
+import { composeContext, getModelSettings, parseTagContent } from "@elizaos/core";
 import { generateText, splitChunks, trimTokens } from "@elizaos/core";
 import { getActorDetails } from "@elizaos/core";
 import { parseJSONObjectFromText } from "@elizaos/core";
@@ -33,13 +33,13 @@ The "start" and "end" are the range of dates that the user wants to summarize, r
 If you aren't sure, you can use a default range of "0 minutes ago" to "2 hours ago" or more. Better to err on the side of including too much than too little.
 
 Your response must be formatted as a JSON block with this structure:
-\`\`\`json
+<response>
 {
   "objective": "<What the user wants to summarize>",
   "start": "0 minutes ago",
   "end": "2 hours ago"
 }
-\`\`\`
+</response>
 `;
 
 const getDateRange = async (
@@ -62,7 +62,8 @@ const getDateRange = async (
         });
         console.log("response", response);
         // try parsing to a json object
-        const parsedResponse = parseJSONObjectFromText(response) as {
+        const extractedResponse = parseTagContent(response, "response");
+        const parsedResponse = parseJSONObjectFromText(extractedResponse) as {
             objective: string;
             start: string | number;
             end: string | number;

packages/client-discord/src/actions/transcribe_media.ts

@@ -1,4 +1,4 @@
-import { composeContext } from "@elizaos/core";
+import { composeContext, parseTagContent } from "@elizaos/core";
 import { generateText } from "@elizaos/core";
 import { parseJSONObjectFromText } from "@elizaos/core";
 import {
@@ -24,10 +24,11 @@ export const mediaAttachmentIdTemplate = `# Messages we are transcribing
 The "attachmentId" is the ID of the media file attachment that the user wants transcribed. If not specified, return null.
 
 Your response must be formatted as a JSON block with this structure:
-\`\`\`json
+<response>
 {
   "attachmentId": "<Attachment ID>"
 }
+</response>
 \`\`\`
 `;
 
@@ -50,8 +51,8 @@ const getMediaAttachmentId = async (
             modelClass: ModelClass.SMALL,
         });
         console.log("response", response);
-
-        const parsedResponse = parseJSONObjectFromText(response) as {
+        const extractedResponse = parseTagContent(response, "response");
+        const parsedResponse = parseJSONObjectFromText(extractedResponse) as {
             attachmentId: string;
         } | null;
 

packages/client-discord/src/attachments.ts

@@ -1,4 +1,4 @@
-import { generateText, trimTokens } from "@elizaos/core";
+import { generateText, parseTagContent, trimTokens } from "@elizaos/core";
 import { parseJSONObjectFromText } from "@elizaos/core";
 import {
     IAgentRuntime,
@@ -28,11 +28,12 @@ async function generateSummary(
   """
 
   Respond with a JSON object in the following format:
-  \`\`\`json
+  <response>
   {
     "title": "Generated Title",
     "summary": "Generated summary and/or description of the text"
   }
+  </response>
   \`\`\``;
 
     const response = await generateText({
@@ -41,7 +42,8 @@ async function generateSummary(
         modelClass: ModelClass.SMALL,
     });
 
-    const parsedResponse = parseJSONObjectFromText(response);
+    const extractedResponse = parseTagContent(response, "response");
+    const parsedResponse = parseJSONObjectFromText(extractedResponse);
 
     if (parsedResponse) {
         return {

packages/client-discord/src/index.ts

@@ -93,14 +93,14 @@ export class DiscordClient extends EventEmitter {
         // );
 
         // Handle voice events with the voice manager
-        this.client.on(
-            "voiceStateUpdate",
-            this.voiceManager.handleVoiceStateUpdate.bind(this.voiceManager)
-        );
-        this.client.on(
-            "userStream",
-            this.voiceManager.handleUserStream.bind(this.voiceManager)
-        );
+        // this.client.on(
+        //     "voiceStateUpdate",
+        //     this.voiceManager.handleVoiceStateUpdate.bind(this.voiceManager)
+        // );
+        // this.client.on(
+        //     "userStream",
+        //     this.voiceManager.handleUserStream.bind(this.voiceManager)
+        // );
 
         // Handle a new message with the message manager
         this.client.on(

packages/client-discord/src/messages.ts

@@ -1,4 +1,8 @@
-import { composeContext, composeRandomUser } from "@elizaos/core";
+import {
+    composeContext,
+    composeRandomUser,
+    parseTagContent,
+} from "@elizaos/core";
 import { generateMessageResponse, generateShouldRespond } from "@elizaos/core";
 import {
     Content,
@@ -1242,13 +1246,15 @@ export class MessageManager {
                 composeRandomUser(discordShouldRespondTemplate, 2),
         });
 
-        const response = await generateShouldRespond({
+        const rawResponse = await generateShouldRespond({
             runtime: this.runtime,
             context: shouldRespondContext,
             modelClass: ModelClass.SMALL,
         });
 
-        if (response === "RESPOND") {
+        const parsedResponse = parseTagContent(rawResponse, "response");
+
+        if (parsedResponse === "RESPOND") {
             if (channelState) {
                 channelState.previousContext = {
                     content: message.content,
@@ -1257,23 +1263,23 @@ export class MessageManager {
             }
 
             return true;
-        } else if (response === "IGNORE") {
+        } else if (parsedResponse === "IGNORE") {
             return false;
-        } else if (response === "STOP") {
+        } else if (parsedResponse === "STOP") {
             delete this.interestChannels[message.channelId];
             return false;
         } else {
             console.error(
                 "Invalid response from response generateText:",
-                response
+                parsedResponse
             );
             return false;
         }
     }
 
     private async _generateResponse(
         message: Memory,
-        state: State,
+        _state: State,
         context: string
     ): Promise<Content> {
         const { userId, roomId } = message;