create-message-kit 1.1.7-beta.2 → 1.1.7-beta.20

package/templates/group/src/handler/tipping.ts

@@ -1,49 +1,39 @@
-import { HandlerContext, AbstractedMember } from "@xmtp/message-kit";
-import { getUserInfo } from "../lib/resolver.js";
+import {
+  HandlerContext,
+  AbstractedMember,
+  SkillResponse,
+} from "@xmtp/message-kit";
+import { getUserInfo } from "@xmtp/message-kit";
 
-export async function handler(context: HandlerContext) {
+export async function handler(context: HandlerContext): Promise<SkillResponse> {
   const {
     members,
-    getMessageById,
-    message: { content, sender, typeId },
+    message: {
+      content: {
+        reference,
+        reply,
+        text,
+        skill,
+        params: { amount, username },
+      },
+      sender,
+    },
   } = context;
-  const msg = await getMessageById(content.reference);
-  const replyReceiver = members?.find(
-    (member) => member.inboxId === msg?.senderInboxId,
-  );
-  let amount: number = 0,
-    receivers: AbstractedMember[] = [];
-  // Handle different types of messages
-  if (typeId === "reply" && replyReceiver) {
-    const { content: reply } = content;
-
-    if (reply.includes("degen")) {
-      receivers = [replyReceiver];
-      const match = reply.match(/(\d+)/);
-      if (match)
-        amount = parseInt(match[0]); // Extract amount from reply
-      else amount = 10;
-    }
-  } else if (typeId === "text") {
-    const { content: text, params } = content;
-    if (text.startsWith("/tip") && params) {
-      // Process text skills starting with "/tip"
-      const {
-        params: { amount: extractedAmount, username },
-      } = content;
-      amount = extractedAmount || 10; // Default amount if not specified
+  let receivers: AbstractedMember[] = [];
 
-      receivers = await Promise.all(
-        username.map((username: string) => getUserInfo(username)),
-      );
-    }
+  if (skill === "tip") {
+    receivers = await Promise.all(
+      username.map((username: string) => getUserInfo(username)),
+    );
   }
   if (!sender || receivers.length === 0 || amount === 0) {
     context.reply("Sender or receiver or amount not found.");
-    return;
+    return {
+      code: 400,
+      message: "Sender or receiver or amount not found.",
+    };
   }
   const receiverAddresses = receivers.map((receiver) => receiver.address);
-  // Process sending tokens to each receiver
 
   context.sendTo(
     `You received ${amount} tokens from ${sender.address}.`,
@@ -55,4 +45,8 @@ export async function handler(context: HandlerContext) {
     `You sent ${amount * receiverAddresses.length} tokens in total.`,
     [sender.address],
   );
+  return {
+    code: 200,
+    message: "Success",
+  };
 }

package/templates/group/src/handler/transaction.ts

@@ -1,50 +1,69 @@
-import { HandlerContext } from "@xmtp/message-kit";
-import { getUserInfo } from "../lib/resolver.js";
+import { getUserInfo, HandlerContext, SkillResponse } from "@xmtp/message-kit";
 
-// Main handler function for processing commands
-export async function handler(context: HandlerContext) {
+// Main handler function for processing
+export async function handler(context: HandlerContext): Promise<SkillResponse> {
   const {
     message: {
-      content: { command, params },
+      content: { skill, params },
     },
   } = context;
   const baseUrl = "https://base-tx-frame.vercel.app/transaction";
 
-  switch (command) {
+  switch (skill) {
     case "send":
-      // Destructure and validate parameters for the send command
+      // Destructure and validate parameters for the send
       const { amount: amountSend, token: tokenSend, username } = params; // [!code hl] // [!code focus]
       let senderInfo = await getUserInfo(username);
       if (!amountSend || !tokenSend || !senderInfo) {
         context.reply(
           "Missing required parameters. Please provide amount, token, and username.",
         );
-        return;
+        return {
+          code: 400,
+          message:
+            "Missing required parameters. Please provide amount, token, and username.",
+        };
       }
 
       let sendUrl = `${baseUrl}/?transaction_type=send&amount=${amountSend}&token=${tokenSend}&receiver=${senderInfo.address}`;
       context.send(`${sendUrl}`);
-      break;
+      return {
+        code: 200,
+        message: `${sendUrl}`,
+      };
     case "swap":
-      // Destructure and validate parameters for the swap command
+      // Destructure and validate parameters for the swap
       const { amount, token_from, token_to } = params; // [!code hl] // [!code focus]
 
       if (!amount || !token_from || !token_to) {
         context.reply(
           "Missing required parameters. Please provide amount, token_from, and token_to.",
         );
-        return;
+        return {
+          code: 400,
+          message:
+            "Missing required parameters. Please provide amount, token_from, and token_to.",
+        };
       }
 
       let swapUrl = `${baseUrl}/?transaction_type=swap&token_from=${token_from}&token_to=${token_to}&amount=${amount}`;
       context.send(`${swapUrl}`);
-      break;
+      return {
+        code: 200,
+        message: `${swapUrl}`,
+      };
     case "show": // [!code hl] // [!code focus]
       // Show the base URL without the transaction path
       context.reply(`${baseUrl.replace("/transaction", "")}`);
-      break;
+      return {
+        code: 200,
+        message: `${baseUrl.replace("/transaction", "")}`,
+      };
     default:
-      // Handle unknown commands
-      context.reply("Unknown command. Use help to see all available commands.");
+      context.reply("Unknown skill. Use help to see all available skills.");
+      return {
+        code: 400,
+        message: "Unknown skill. Use help to see all available skills.",
+      };
   }
 }

package/templates/group/src/index.ts

@@ -1,37 +1,32 @@
 import { run, HandlerContext } from "@xmtp/message-kit";
+import { textGeneration, processMultilineResponse } from "@xmtp/message-kit";
+import { agent_prompt } from "./prompt.js";
+import { getUserInfo } from "@xmtp/message-kit";
 
-// Main function to run the app
-run(
-  async (context: HandlerContext) => {
-    const {
-      message: { typeId },
-      group,
-    } = context;
-    switch (typeId) {
-      case "reply":
-        handleReply(context);
-        break;
-    }
-    if (!group) {
-      context.send("This is a group bot, add this address to a group");
-    }
-  },
-  { attachments: true },
-);
-async function handleReply(context: HandlerContext) {
+run(async (context: HandlerContext) => {
   const {
-    v2client,
-    getReplyChain,
-    version,
     message: {
-      content: { reference },
+      content: { text, params },
+      sender,
     },
+    group,
   } = context;
 
-  const { chain, isSenderInChain } = await getReplyChain(
-    reference,
-    version,
-    v2client.address,
-  );
-  //await context.skill(chain);
-}
+  try {
+    let userPrompt = params?.prompt ?? text;
+    const userInfo = await getUserInfo(sender.address);
+    if (!userInfo) {
+      console.log("User info not found");
+      return;
+    }
+    const { reply } = await textGeneration(
+      sender.address,
+      userPrompt,
+      await agent_prompt(userInfo),
+    );
+    await processMultilineResponse(sender.address, reply, context);
+  } catch (error) {
+    console.error("Error during OpenAI call:", error);
+    await context.send("An error occurred while processing your request.");
+  }
+});

package/templates/group/src/prompt.ts

@@ -0,0 +1,25 @@
+import { skills } from "./skills.js";
+import {
+  UserInfo,
+  PROMPT_USER_CONTENT,
+  PROMPT_RULES,
+  PROMPT_SKILLS_AND_EXAMPLES,
+  PROMPT_REPLACE_VARIABLES,
+} from "@xmtp/message-kit";
+
+export async function agent_prompt(userInfo: UserInfo) {
+  let systemPrompt =
+    PROMPT_RULES +
+    PROMPT_USER_CONTENT(userInfo) +
+    PROMPT_SKILLS_AND_EXAMPLES(skills, "@bot");
+
+  // Replace the variables in the system prompt
+  systemPrompt = PROMPT_REPLACE_VARIABLES(
+    systemPrompt,
+    userInfo?.address ?? "",
+    userInfo,
+    "@bot",
+  );
+  console.log(systemPrompt);
+  return systemPrompt;
+}

package/templates/group/src/skills.ts

@@ -1,18 +1,17 @@
 import { handler as tipping } from "./handler/tipping.js";
-import { handler as agent } from "./handler/agent.js";
 import { handler as transaction } from "./handler/transaction.js";
 import { handler as games } from "./handler/game.js";
-import { handler as loyalty } from "./handler/loyalty.js";
-import { handler as groupHelp } from "./handler/group.js";
+import { handler as help } from "./handler/helpers.js";
 import type { SkillGroup } from "@xmtp/message-kit";
 
 export const skills: SkillGroup[] = [
   {
-    name: "Tipping",
-    description: "Tip tokens via emoji, replies or command.",
+    name: "Group bot",
+    tag: "@bot",
+    description: "Group bot for tipping and transactions.",
     skills: [
       {
-        command: "/tip [@usernames] [amount] [token]",
+        skill: "/tip [usernames] [amount] [token]",
         triggers: ["/tip"],
         examples: ["/tip @vitalik 10 usdc"],
         description: "Tip users in a specified token.",
@@ -29,15 +28,8 @@ export const skills: SkillGroup[] = [
           },
         },
       },
-    ],
-  },
-
-  {
-    name: "Transactions",
-    description: "Multipurpose transaction frame built onbase.",
-    skills: [
       {
-        command: "/send [amount] [token] [username]",
+        skill: "/send [amount] [token] [username]",
         triggers: ["/send"],
         examples: ["/send 10 usdc @vitalik"],
         description:
@@ -60,7 +52,7 @@ export const skills: SkillGroup[] = [
         },
       },
       {
-        command: "/swap [amount] [token_from] [token_to]",
+        skill: "/swap [amount] [token_from] [token_to]",
         triggers: ["/swap"],
         examples: ["/swap 10 usdc eth"],
         description: "Exchange one type of cryptocurrency for another.",
@@ -83,24 +75,19 @@ export const skills: SkillGroup[] = [
         },
       },
       {
-        command: "/show",
+        skill: "/show",
         triggers: ["/show"],
         examples: ["/show"],
         handler: transaction,
         description: "Show the whole frame.",
         params: {},
       },
-    ],
-  },
-  {
-    name: "Games",
-    description: "Provides various gaming experiences.",
-    skills: [
       {
-        command: "/game [game]",
+        skill: "/game [game]",
         triggers: ["/game", "🔎", "🔍"],
         handler: games,
         description: "Play a game.",
+        examples: ["/game wordle", "/game slot", "/game help"],
         params: {
           game: {
             default: "",
@@ -109,65 +96,19 @@ export const skills: SkillGroup[] = [
           },
         },
       },
-    ],
-  },
-  {
-    name: "Loyalty",
-    description: "Manage group members and metadata.",
-    skills: [
-      {
-        command: "/points",
-        triggers: ["/points"],
-        examples: ["/points"],
-        handler: loyalty,
-        description: "Check your points.",
-        params: {},
-      },
-      {
-        command: "/leaderboard",
-        triggers: ["/leaderboard"],
-        adminOnly: true,
-        handler: loyalty,
-        description: "Check the points of a user.",
-        params: {},
-      },
-    ],
-  },
-  {
-    name: "Agent",
-    description: "Manage agent commands.",
-    skills: [
-      {
-        command: "/agent [prompt]",
-        triggers: ["/agent", "@agent", "@bot"],
-        examples: ["/agent @vitalik"],
-        handler: agent,
-        description: "Manage agent commands.",
-        params: {
-          prompt: {
-            default: "",
-            type: "prompt",
-          },
-        },
-      },
-    ],
-  },
-  {
-    name: "Help",
-    description: "Get help with the bot.",
-    skills: [
       {
-        command: "/help",
+        skill: "/help",
         triggers: ["/help"],
         examples: ["/help"],
-        handler: groupHelp,
+        handler: help,
         description: "Get help with the bot.",
         params: {},
       },
       {
-        command: "/id",
+        skill: "/id",
         adminOnly: true,
-        handler: groupHelp,
+        examples: ["/id"],
+        handler: help,
         triggers: ["/id"],
         description: "Get the group ID.",
         params: {},

package/templates/agent/package.json

@@ -1,22 +0,0 @@
-{
-  "name": "agent",
-  "private": true,
-  "type": "module",
-  "scripts": {
-    "build": "tsc",
-    "dev": "tsc -w & sleep 1 && nodemon --quiet dist/index.js",
-    "start": "node dist/index.js"
-  },
-  "dependencies": {
-    "@xmtp/message-kit": "workspace:*",
-    "openai": "^4.65.0"
-  },
-  "devDependencies": {
-    "@types/node": "^20.14.2",
-    "nodemon": "^3.1.3",
-    "typescript": "^5.4.5"
-  },
-  "engines": {
-    "node": ">=20"
-  }
-}

package/templates/agent/src/lib/gpt.ts

@@ -1,161 +0,0 @@
-import "dotenv/config";
-import type { SkillGroup } from "@xmtp/message-kit";
-import OpenAI from "openai";
-const openai = new OpenAI({
-  apiKey: process.env.OPEN_AI_API_KEY,
-});
-
-type ChatHistoryEntry = { role: string; content: string };
-type ChatHistories = Record<string, ChatHistoryEntry[]>;
-// New ChatMemory class
-class ChatMemory {
-  private histories: ChatHistories = {};
-
-  getHistory(address: string): ChatHistoryEntry[] {
-    return this.histories[address] || [];
-  }
-
-  addEntry(address: string, entry: ChatHistoryEntry) {
-    if (!this.histories[address]) {
-      this.histories[address] = [];
-    }
-    this.histories[address].push(entry);
-  }
-
-  initializeWithSystem(address: string, systemPrompt: string) {
-    if (this.getHistory(address).length === 0) {
-      this.addEntry(address, {
-        role: "system",
-        content: systemPrompt,
-      });
-    }
-  }
-
-  clear() {
-    this.histories = {};
-  }
-}
-
-// Create singleton instance
-export const chatMemory = new ChatMemory();
-
-export const clearMemory = () => {
-  chatMemory.clear();
-};
-
-export const PROMPT_RULES = `You are a helpful and playful agent called {NAME} that lives inside a web3 messaging app called Converse.
-- You can respond with multiple messages if needed. Each message should be separated by a newline character.
-- You can trigger skills by only sending the command in a newline message.
-- Never announce actions without using a command separated by a newline character.
-- Dont answer in markdown format, just answer in plaintext.
-- Do not make guesses or assumptions
-- Only answer if the verified information is in the prompt.
-- Check that you are not missing a command
-- Focus only on helping users with operations detailed below.
-`;
-
-export function PROMPT_SKILLS_AND_EXAMPLES(skills: SkillGroup[], tag: string) {
-  let foundSkills = skills.filter(
-    (skill) => skill.tag == `@${tag.toLowerCase()}`,
-  );
-  if (!foundSkills.length || !foundSkills[0] || !foundSkills[0].skills)
-    return "";
-  let returnPrompt = `\nCommands:\n${foundSkills[0].skills
-    .map((skill) => skill.command)
-    .join("\n")}\n\nExamples:\n${foundSkills[0].skills
-    .map((skill) => skill.examples)
-    .join("\n")}`;
-  return returnPrompt;
-}
-
-export async function textGeneration(
-  memoryKey: string,
-  userPrompt: string,
-  systemPrompt: string,
-) {
-  if (!memoryKey) {
-    clearMemory();
-  }
-  let messages = chatMemory.getHistory(memoryKey);
-  chatMemory.initializeWithSystem(memoryKey, systemPrompt);
-  if (messages.length === 0) {
-    messages.push({
-      role: "system",
-      content: systemPrompt,
-    });
-  }
-  messages.push({
-    role: "user",
-    content: userPrompt,
-  });
-  try {
-    const response = await openai.chat.completions.create({
-      model: "gpt-4o",
-      messages: messages as any,
-    });
-    const reply = response.choices[0].message.content;
-    messages.push({
-      role: "assistant",
-      content: reply || "No response from OpenAI.",
-    });
-    const cleanedReply = parseMarkdown(reply as string);
-    chatMemory.addEntry(memoryKey, {
-      role: "assistant",
-      content: cleanedReply,
-    });
-    return { reply: cleanedReply, history: messages };
-  } catch (error) {
-    console.error("Failed to fetch from OpenAI:", error);
-    throw error;
-  }
-}
-
-export async function processMultilineResponse(
-  memoryKey: string,
-  reply: string,
-  context: any,
-) {
-  if (!memoryKey) {
-    clearMemory();
-  }
-  let messages = reply
-    .split("\n")
-    .map((message: string) => parseMarkdown(message))
-    .filter((message): message is string => message.length > 0);
-
-  console.log(messages);
-  for (const message of messages) {
-    if (message.startsWith("/")) {
-      const response = await context.skill(message);
-      if (response && typeof response.message === "string") {
-        let msg = parseMarkdown(response.message);
-        chatMemory.addEntry(memoryKey, {
-          role: "system",
-          content: msg,
-        });
-        await context.send(response.message);
-      }
-    } else {
-      await context.send(message);
-    }
-  }
-}
-export function parseMarkdown(message: string) {
-  let trimmedMessage = message;
-  // Remove bold and underline markdown
-  trimmedMessage = trimmedMessage?.replace(/(\*\*|__)(.*?)\1/g, "$2");
-  // Remove markdown links, keeping only the URL
-  trimmedMessage = trimmedMessage?.replace(/\[([^\]]+)\]\(([^)]+)\)/g, "$2");
-  // Remove markdown headers
-  trimmedMessage = trimmedMessage?.replace(/^#+\s*(.*)$/gm, "$1");
-  // Remove inline code formatting
-  trimmedMessage = trimmedMessage?.replace(/`([^`]+)`/g, "$1");
-  // Remove single backticks at the start or end of the message
-  trimmedMessage = trimmedMessage?.replace(/^`|`$/g, "");
-  // Remove leading and trailing whitespace
-  trimmedMessage = trimmedMessage?.replace(/^\s+|\s+$/g, "");
-  // Remove any remaining leading or trailing whitespace
-  trimmedMessage = trimmedMessage.trim();
-
-  return trimmedMessage;
-}