create-message-kit 1.1.7-beta.9 → 1.1.8

package/templates/group/src/index.ts

@@ -1,34 +1,31 @@
 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");
-  }
-});
-async function handleReply(context: HandlerContext) {
-  const {
-    v2client,
-    getReplyChain,
-    version,
     message: {
-      content: { reference },
+      content: { text, params },
+      sender,
     },
   } = context;
 
-  const { chain, isSenderInChain } = await getReplyChain(
-    reference,
-    version,
-    v2client.address,
-  );
-  //await context.executeSkill(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,45 @@
+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");
+
+  let fineTunedPrompt = `
+## Example response
+
+1. If user wants to play a game, use the skill 'game' and specify the game type.
+  Hey! Sure let's do that.\n/game wordle
+  
+2. When user wants to pay a specific token:
+  I'll help you pay 1 USDC to 0x123...\n/pay 1 {TOKE}} 0x123456789...
+  *This will return a url to pay
+
+3. If the user wants to pay a eth domain:
+  I'll help you pay 1 USDC to vitalik.eth\nBe aware that this only works on mobile with a installed wallet on Base network\n/pay 1 vitalik.eth
+  *This will return a url to pay
+
+4. If the user wants to pay a username:
+  I'll help you pay 1 USDC to @fabri\nBe aware that this only works on mobile with a installed wallet on Base network\n/pay 1 @fabri
+  *This will return a url to pay
+  `;
+
+  systemPrompt += fineTunedPrompt;
+  // 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 payment } from "./handler/payment.js";
 import { handler as games } from "./handler/game.js";
-import { handler as loyalty } from "./handler/loyalty.js";
-import { handler as groupHelp } from "./handler/helpers.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 agent 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,20 +28,13 @@ export const skills: SkillGroup[] = [
           },
         },
       },
-    ],
-  },
-
-  {
-    name: "Transactions",
-    description: "Multipurpose transaction frame built onbase.",
-    skills: [
       {
-        command: "/send [amount] [token] [username]",
-        triggers: ["/send"],
-        examples: ["/send 10 usdc @vitalik"],
+        skill: "/pay [amount] [token] [username]",
+        triggers: ["/pay"],
+        examples: ["/pay 10 vitalik.eth"],
         description:
           "Send a specified amount of a cryptocurrency to a destination address.",
-        handler: transaction,
+        handler: payment,
         params: {
           amount: {
             default: 10,
@@ -60,47 +52,11 @@ export const skills: SkillGroup[] = [
         },
       },
       {
-        command: "/swap [amount] [token_from] [token_to]",
-        triggers: ["/swap"],
-        examples: ["/swap 10 usdc eth"],
-        description: "Exchange one type of cryptocurrency for another.",
-        handler: transaction,
-        params: {
-          amount: {
-            default: 10,
-            type: "number",
-          },
-          token_from: {
-            default: "usdc",
-            type: "string",
-            values: ["eth", "dai", "usdc", "degen"], // Accepted tokens
-          },
-          token_to: {
-            default: "eth",
-            type: "string",
-            values: ["eth", "dai", "usdc", "degen"], // Accepted tokenss
-          },
-        },
-      },
-      {
-        command: "/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 +65,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.executeSkill(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;
-}

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

@@ -1,174 +0,0 @@
-import dotenv from "dotenv";
-dotenv.config();
-import type { SkillGroup } from "@xmtp/message-kit";
-import OpenAI from "openai";
-const openai = new OpenAI({
-  apiKey: process.env.OPEN_AI_API_KEY,
-});
-
-export type ChatHistoryEntry = { role: string; content: string };
-export 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 = {};
-  }
-}
-
-export const clearMemory = () => {
-  chatHistories = {};
-};
-
-// Create singleton instance
-export const chatMemory = new ChatMemory();
-
-let chatHistories: ChatHistories = {};
-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 const PROMPT_SKILLS_AND_EXAMPLES = (skills: SkillGroup[]) => `
-Commands:
-${skills
-  .map((skill) => skill.skills.map((s) => s.command).join("\n"))
-  .join("\n")}
-
-Examples:
-${skills
-  .map((skill) => skill.skills.map((s) => s.examples).join("\n"))
-  .join("\n")}
-  `;
-
-export async function agentResponse(
-  sender: { address: string },
-  userPrompt: string,
-  systemPrompt: string,
-  context: any,
-) {
-  try {
-    const { reply } = await textGeneration(
-      sender.address,
-      userPrompt,
-      systemPrompt,
-    );
-    await processMultilineResponse(sender.address, reply, context);
-  } catch (error) {
-    console.error("Error during OpenAI call:", error);
-    await context.reply("An error occurred while processing your request.");
-  }
-}
-export async function textGeneration(
-  address: string,
-  userPrompt: string,
-  systemPrompt: string,
-) {
-  let messages = chatMemory.getHistory(address);
-  chatMemory.initializeWithSystem(address, 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(address, {
-      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(
-  address: string,
-  reply: string,
-  context: any,
-) {
-  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(address, {
-          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;
-}