@ai.ntellect/core 0.1.96 → 0.1.98

package/dist/test.js

@@ -3,14 +3,112 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getRssNews = exports.getChainsTVL = exports.fetchMarkPrice = void 0;
+exports.getRssNews = exports.getChainsTVL = exports.fetchMarkPrice = exports.prepareEvmTransaction = exports.getNetworkProvider = exports.networkConfigs = void 0;
 const ccxt_1 = __importDefault(require("ccxt"));
+const ethers_1 = require("ethers");
 const rss_parser_1 = __importDefault(require("rss-parser"));
 const zod_1 = require("zod");
 const agent_1 = require("./agent");
 const orchestrator_1 = require("./llm/orchestrator");
 const cache_1 = require("./memory/cache");
 const persistent_1 = require("./memory/persistent");
+exports.networkConfigs = {
+    ethereum: {
+        name: "Ethereum Mainnet",
+        id: 1,
+        rpc: "https://eth.llamarpc.com",
+        explorerUrl: "https://etherscan.io",
+        nativeToken: "0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2",
+    },
+    polygon: {
+        name: "Polygon Mainnet",
+        id: 137,
+        rpc: "https://polygon.llamarpc.com",
+        explorerUrl: "https://polygonscan.com",
+        nativeToken: "0x0000000000000000000000000000000000001010",
+    },
+    arbitrum: {
+        name: "Arbitrum Mainnet",
+        id: 42161,
+        rpc: "https://arbitrum.llamarpc.com",
+        explorerUrl: "https://arbiscan.io",
+        nativeToken: "0x82af49447d8a07e3bd95bd0d56f35241523fbab1",
+    },
+    base: {
+        name: "Base Mainnet",
+        id: 8453,
+        rpc: "https://base.llamarpc.com",
+        explorerUrl: "https://basescan.org",
+        nativeToken: "0x4200000000000000000000000000000000000006",
+    },
+    solana: {
+        name: "Solana Mainnet",
+        rpc: "https://api.mainnet-beta.solana.com",
+        explorerUrl: "https://solscan.io",
+        nativeToken: "So11111111111111111111111111111111111111112",
+    },
+    sepolia: {
+        name: "Sepolia Testnet",
+        id: 11155111,
+        rpc: "https://sepolia.llamarpc.com",
+        explorerUrl: "https://sepolia.etherscan.io",
+        nativeToken: "0xfFf9976782d46CC05630D1f6eBAb18b2324d6B14",
+    },
+    baseSepolia: {
+        name: "Base Sepolia Testnet",
+        id: 84532,
+        rpc: "https://base-sepolia-rpc.publicnode.com",
+        explorerUrl: "https://sepolia.basescan.org",
+        nativeToken: "0x4200000000000000000000000000000000000006",
+    },
+};
+const getNetworkProvider = (networkName) => {
+    const config = exports.networkConfigs[networkName.toLowerCase()];
+    if (!config) {
+        throw new Error(`Network ${networkName} not supported`);
+    }
+    return { config };
+};
+exports.getNetworkProvider = getNetworkProvider;
+exports.prepareEvmTransaction = {
+    name: "prepare-evm-transaction",
+    description: "Prepare a transaction for the user to sign.",
+    parameters: zod_1.z.object({
+        walletAddress: zod_1.z.string(),
+        amount: zod_1.z
+            .string()
+            .describe("Ask the user for the amount to send, if not specified"),
+        network: zod_1.z
+            .string()
+            .describe("Examples networks: ethereum, arbitrum, base. IMPORTANT: You must respect the network name."),
+    }),
+    execute: async ({ walletAddress, amount, network, }) => {
+        try {
+            console.log("💰 Preparing transaction", {
+                to: walletAddress,
+                amount,
+                network,
+            });
+            const networkConfig = exports.networkConfigs[network.toLowerCase()];
+            if (!networkConfig) {
+                throw new Error(`Network ${network} not found`);
+            }
+            return {
+                to: walletAddress,
+                value: (0, ethers_1.parseEther)(amount).toString(),
+                chain: {
+                    id: networkConfig.id || 0,
+                    rpc: networkConfig.rpc,
+                },
+                type: "transfer",
+            };
+        }
+        catch (error) {
+            console.error("💰 Error sending transaction:", error);
+            throw new Error("An error occurred while sending the transaction");
+        }
+    },
+};
 exports.fetchMarkPrice = {
     name: "fetch-mark-price",
     description: "Fetches mark price for the market",
@@ -141,7 +239,10 @@ exports.getRssNews = {
         host: "http://localhost:7700",
         apiKey: "aSampleMasterKey",
     });
-    const orchestrator = new orchestrator_1.Orchestrator([exports.getRssNews, exports.getChainsTVL, exports.fetchMarkPrice], memory);
+    const orchestrator = new orchestrator_1.Orchestrator("1", [exports.getRssNews, exports.getChainsTVL, exports.fetchMarkPrice, exports.prepareEvmTransaction], {
+        persistent: memory,
+        cache: cacheMemory,
+    });
     const agent = new agent_1.Agent({
         user: {
             id: "1",
@@ -152,7 +253,7 @@ exports.getRssNews = {
         stream: false,
         maxEvaluatorIteration: 1,
     });
-    const prompt = "analyse le ai16z";
+    const prompt = "transfere 0.0001 eth to 0xf520cEd3b7FdA050a3A44486C160BEAb15ED3285 on ethereum";
     const context = prompt;
     // const save = await cacheMemory.createMemory({
     //   content: prompt,
@@ -160,8 +261,8 @@ exports.getRssNews = {
     //   scope: MemoryScope.GLOBAL,
     //   type: MemoryType.ACTION,
     // });
-    // console.log({ save });
+    // consoleil.log({ save });
     // const memo = await cacheMemory.getAllMemories();
     // console.log({ memo });
-    const result = await agent.process(prompt, context, {});
+    const result = await agent.process(prompt, {});
 })();

package/dist/types.d.ts

@@ -48,8 +48,15 @@ export interface ProcessPromptCallbacks {
 export interface ActionSchema {
     name: string;
     description: string;
-    parameters: z.ZodSchema;
+    parameters: z.ZodObject<{
+        [key: string]: z.ZodType;
+    }>;
     execute: (args: any) => Promise<any>;
+    examples?: {
+        role: string;
+        content: string;
+        parameters: Record<string, any>;
+    }[];
     confirmation?: {
         requireConfirmation: boolean;
         message: string;

package/dist/utils/inject-actions.js

@@ -5,7 +5,13 @@ const injectActions = (actions) => {
     return actions.map((action) => {
         const parameters = action.parameters;
         const schemaShape = Object.keys(parameters._def.shape()).join(", ");
-        const actionString = `Name: ${action.name}, Description: ${action.description}, Arguments (STRICTLY REQUIRED): { ${schemaShape} }`;
+        const actionString = `Name: ${action.name}, Description: ${action.description}, Arguments (STRICTLY REQUIRED): { ${schemaShape} } ${action.examples
+            ? `Examples: ${action.examples
+                .map((e) => {
+                return `Role: ${e.role}, Content: ${e.content}, Parameters: { ${Object.keys(e.parameters).join(", ")} }`;
+            })
+                .join("\n")}`
+            : ""}`;
         return actionString;
     });
 };

package/llm/evaluator/index.ts

@@ -44,7 +44,6 @@ export class Evaluator {
         actions: this.tools,
         results: results,
       });
-
       console.log("\n🔍 Evaluator processing");
       console.log("Goal:", prompt);
       console.log("Results to evaluate:", JSON.stringify(results, null, 2));

package/llm/orchestrator/context.ts

@@ -7,8 +7,8 @@ export const orchestratorContext = {
         "If there is no action to do, you must answer in the 'answer' field.",
         "If some parameters are not clear or missing, don't add the action, YOU MUST ask the user for them.",
         "ALWAYS use the same language as user request. (If it's English, use English, if it's French, use French, etc.)",
-        "For QUESTIONS or ANALYSIS, you CAN search in memory and internal knowledge base.",
         "For ON-CHAIN actions, just use the useful actions.",
+        "For QUESTIONS or ANALYSIS, you CAN search in memory and internal knowledge base.",
         "NEVER repeat same actions if the user doesn't ask for it.",
       ],
       warnings: [],

package/llm/synthesizer/index.ts

@@ -6,7 +6,7 @@ import { QueueResult } from "../../types";
 import { synthesizerContext } from "./context";
 
 export class Synthesizer {
-  private readonly model = openai("gpt-4-turbo");
+  private readonly model = openai("gpt-4o");
 
   composeContext(state: Partial<State>) {
     const { behavior, userRequest, results, examplesMessages } = state;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai.ntellect/core",
-  "version": "0.1.96",
+  "version": "0.1.98",
   "description": "",
   "main": "dist/index.js",
   "scripts": {

package/types.ts

@@ -63,6 +63,11 @@ export interface ActionSchema {
     [key: string]: z.ZodType;
   }>;
   execute: (args: any) => Promise<any>;
+  examples?: {
+    role: string;
+    content: string;
+    parameters: Record<string, any>;
+  }[];
   confirmation?: {
     requireConfirmation: boolean;
     message: string;

package/utils/inject-actions.ts

@@ -5,7 +5,19 @@ export const injectActions = (actions: ActionSchema[]) => {
   return actions.map((action) => {
     const parameters = action.parameters as z.ZodObject<any>;
     const schemaShape = Object.keys(parameters._def.shape()).join(", ");
-    const actionString = `Name: ${action.name}, Description: ${action.description}, Arguments (STRICTLY REQUIRED): { ${schemaShape} }`;
+    const actionString = `Name: ${action.name}, Description: ${
+      action.description
+    }, Arguments (STRICTLY REQUIRED): { ${schemaShape} } ${
+      action.examples
+        ? `Examples: ${action.examples
+            .map((e: any) => {
+              return `Role: ${e.role}, Content: ${
+                e.content
+              }, Parameters: { ${Object.keys(e.parameters).join(", ")} }`;
+            })
+            .join("\n")}`
+        : ""
+    }`;
     return actionString;
   });
 };