feat: asterai plugin

Author: rellfy

packages/plugin-asterai/package.json

@@ -32,15 +32,13 @@
     "devDependencies": {
         "@types/elliptic": "6.4.18",
         "@types/uuid": "10.0.0",
-        "tsup": "8.3.5",
-        "vitest": "2.1.4"
+        "tsup": "8.3.5"
     },
     "scripts": {
         "lines": "find . \\( -name '*.cdc' -o -name '*.ts' \\) -not -path '*/node_modules/*' -not -path '*/tests/*' -not -path '*/deps/*' -not -path '*/dist/*' -not -path '*/imports*' | xargs wc -l",
         "build": "tsup --format esm --dts",
         "dev": "tsup --format esm --dts --watch",
-        "lint": "eslint --fix  --cache .",
-        "test": "vitest run"
+        "lint": "eslint --fix  --cache ."
     },
     "peerDependencies": {
         "whatwg-url": "7.1.0"

packages/plugin-asterai/src/actions/query.ts

@@ -8,9 +8,7 @@ import {
     type State,
 } from "@elizaos/core";
 import { validateAsteraiConfig } from "../environment";
-import {AsteraiClient} from "@asterai/client";
-
-let asteraiClient: AsteraiClient | null = null;
+import {getInitAsteraiClient} from "../index.ts";
 
 export const queryAction = {
     name: "QUERY_ASTERAI_AGENT",
@@ -26,25 +24,24 @@ export const queryAction = {
         "the user you are assisting, to help perform a workflow task, etc.",
     validate: async (runtime: IAgentRuntime, _message: Memory) => {
         const config = await validateAsteraiConfig(runtime);
-        if (!asteraiClient) {
-            asteraiClient = new AsteraiClient({
-                appId: config.ASTERAI_AGENT_ID,
-                queryKey: config.ASTERAI_PUBLIC_QUERY_KEY,
-            })
-        }
+        getInitAsteraiClient(
+          config.ASTERAI_AGENT_ID,
+          config.ASTERAI_PUBLIC_QUERY_KEY
+        );
         return true;
     },
     handler: async (
         runtime: IAgentRuntime,
         message: Memory,
-        state: State,
+        _state: State,
         _options: { [key: string]: unknown },
         callback?: HandlerCallback
     ): Promise<boolean> => {
-        if (asteraiClient === null) {
-            elizaLogger.warn("asterai client is not initialised; ignoring");
-            return null;
-        }
+        const config = await validateAsteraiConfig(runtime);
+        const asteraiClient = getInitAsteraiClient(
+          config.ASTERAI_AGENT_ID,
+          config.ASTERAI_PUBLIC_QUERY_KEY
+        );
         elizaLogger.debug("called QUERY_ASTERAI_AGENT action with message:", message.content);
         const response = await asteraiClient.query({
             query: message.content.text

packages/plugin-asterai/src/index.ts

@@ -1,13 +1,30 @@
+import {asteraiProvider} from "./providers/asterai.provider.ts";
+import type { Plugin } from "@elizaos/core";
+import { queryAction } from "./actions/query";
+import { AsteraiClient } from "@asterai/client";
+
 export * from "./environment";
 export * from "./providers/asterai.provider";
 
-import type { Plugin } from "@elizaos/core";
-import { queryAction } from "./actions/query";
+let asteraiClient: AsteraiClient | null = null;
+
+export const getInitAsteraiClient = (
+  agentId: string,
+  publicQueryKey: string
+): AsteraiClient => {
+    if (!asteraiClient) {
+        asteraiClient = new AsteraiClient({
+            appId: agentId,
+            queryKey: publicQueryKey,
+        })
+    }
+    return asteraiClient;
+};
 
 export const asteraiPlugin: Plugin = {
     name: "asterai",
     description: "asterai Plugin for Eliza",
-    providers: [/*todo*/],
+    providers: [asteraiProvider],
     actions: [queryAction],
     evaluators: [],
     services: [],

packages/plugin-asterai/src/providers/asterai.provider.ts

@@ -0,0 +1,63 @@
+import {
+    elizaLogger,
+    IAgentRuntime,
+    Memory,
+    Provider,
+    State, UUID,
+} from "@elizaos/core";
+import {validateAsteraiConfig} from "../environment.ts";
+import {getInitAsteraiClient} from "../index.ts";
+
+const asteraiProvider: Provider = {
+    get: async (
+        runtime: IAgentRuntime,
+        message: Memory,
+        _state?: State
+    ): Promise<string | null> => {
+        const hasConfiguredEnv =
+          !!runtime.getSetting("ASTERAI_AGENT_ID") &&
+          !!runtime.getSetting("ASTERAI_PUBLIC_QUERY_KEY");
+        if (!hasConfiguredEnv) {
+            elizaLogger.error(
+                "ASTERAI_AGENT_ID or ASTERAI_PUBLIC_QUERY_KEY " +
+                "not configured, skipping provider"
+            );
+            return null;
+        }
+        const config = await validateAsteraiConfig(runtime);
+        const asteraiClient = getInitAsteraiClient(
+          config.ASTERAI_AGENT_ID,
+          config.ASTERAI_PUBLIC_QUERY_KEY
+        );
+        if (!asteraiClient) {
+            elizaLogger.error("asteraiClient is not initialised");
+            return null;
+        }
+        const agentId = runtime.getSetting("ASTERAI_AGENT_ID") as UUID;
+        let agentSummaryMemory = await runtime.knowledgeManager.getMemoryById(agentId);
+        if (!agentSummaryMemory) {
+            // Fetch & set summary memory.
+            const summary = await asteraiClient.fetchSummary();
+            elizaLogger.debug("asterai agent summary fetched:", summary);
+            await runtime.knowledgeManager.createMemory({
+                id: agentId,
+                userId: message.userId,
+                agentId: message.agentId,
+                roomId: message.roomId,
+                createdAt: Date.now(),
+                content: {
+                    text: summary
+                }
+            });
+            agentSummaryMemory = await runtime.knowledgeManager.getMemoryById(agentId);
+        }
+        if (!agentSummaryMemory) {
+            elizaLogger.error("failed to initialise agent's summary memory");
+            return null;
+        }
+        return agentSummaryMemory.content.text;
+    },
+};
+
+// Module exports
+export { asteraiProvider };