start Common client on start-up

hicommonwealth · Jan 3, 2025 · e33b6c5 · e33b6c5
1 parent 7cf2ff1
commit e33b6c5
Show file tree

Hide file tree

Showing 4 changed files with 25 additions and 4 deletions.
diff --git a/agent/src/index.ts b/agent/src/index.ts
@@ -7,6 +7,7 @@ import { LensAgentClient } from "@elizaos/client-lens";
 import { SlackClientInterface } from "@elizaos/client-slack";
 import { TelegramClientInterface } from "@elizaos/client-telegram";
 import { TwitterClientInterface } from "@elizaos/client-twitter";
+import { CommonClientInterface } from '@elizaos/client-common';
 import {
     AgentRuntime,
     CacheManager,
@@ -416,6 +417,13 @@ export async function initializeClients(
         }
     }
 
+    if (clientTypes.includes(Clients.COMMON)) {
+        const commonClient = await CommonClientInterface.start(runtime);
+        if (commonClient) {
+            clients.common = commonClient;
+        }
+    }
+
     if (clientTypes.includes(Clients.FARCASTER)) {
         // why is this one different :(
         const farcasterClient = new FarcasterAgentClient(runtime);

diff --git a/packages/client-common/src/index.ts b/packages/client-common/src/index.ts
@@ -1,4 +1,4 @@
-import { Character, elizaLogger, IAgentRuntime } from "@elizaos/core";
+import { Character, elizaLogger, IAgentRuntime, Client } from "@elizaos/core";
 import { validateCommonConfig } from "./environment.ts";
 import express, { Request, Response, NextFunction } from "express";
 import { CommonApiClient } from "@commonxyz/api-client";
@@ -156,3 +156,14 @@ export class CommonClient {
         }
     }
 }
+
+// TODO: add error handling
+export const CommonClientInterface: Client = {
+    start: async (runtime: IAgentRuntime) => {
+        const client = new CommonClient(runtime);
+        await client.start();
+    },
+    stop: async (runtime: IAgentRuntime) => {
+        await runtime.clients.common.stop();
+    }
+}
diff --git a/packages/client-common/src/messages.ts b/packages/client-common/src/messages.ts
@@ -1,5 +1,6 @@
 import {CommonApiClient} from "@commonxyz/api-client";
 import {
+    Clients,
     composeContext, Content,
     elizaLogger, generateMessageResponse,
     generateShouldRespond,
@@ -131,7 +132,7 @@ export class MessageManager {
             roomId,
             undefined,
             message.profile_name,
-            "Common"
+            Clients.COMMON
         );
 
         const memory: Memory = {
@@ -141,7 +142,7 @@ export class MessageManager {
             agentId: this.runtime.agentId,
             content: {
                 text: message.full_object_text,
-                source: "Common",
+                source: Clients.COMMON,
                 url: message.object_url,
                 // TODO: inReplyTo: undefined,
             },
@@ -188,7 +189,7 @@ export class MessageManager {
             agentId: this.runtime.agentId,
             content: {
                 text: res.body,
-                source: "Common",
+                source: Clients.COMMON,
                 content_url: res.content_url,
             },
             createdAt: new Date(res.created_at).getTime(),

diff --git a/packages/core/src/types.ts b/packages/core/src/types.ts
@@ -614,6 +614,7 @@ export type Plugin = {
  * Available client platforms
  */
 export enum Clients {
+    COMMON = "common",
     DISCORD = "discord",
     DIRECT = "direct",
     TWITTER = "twitter",