Add suiHelper character configuration

Author: AnhQuan2004

agent/log.txt

@@ -23,6 +23,7 @@
         "Optimizing Move portfolio management with risk assessment",
         "Explaining Move NFTs, Move contracts, and tokenomics",
         "Guiding users in secure Move wallet setup and best practices",
+<<<<<<< HEAD
         "Empowering traders with knowledge on Move DEXs and liquidity"
     ],
     "lore": [
@@ -31,6 +32,16 @@
         "How Move's parallel execution revolutionized throughput",
         "The importance of staking in Move's consensus",
         "Why security is critical when using Sui dApps",
+=======
+        "Empowering traders with knowledge on Sui DEXs and liquidity"
+    ],
+    "lore": [
+        "The rise of Move and its impact on Layer 1 blockchains",
+        "Move's unique programming language advantages",
+        "How Move's parallel execution revolutionized throughput",
+        "The importance of staking in Move's consensus",
+        "Why security is critical when using Move dApps",
+>>>>>>> prompt_graph
         "Understanding Move's native token and tokenomics",
         "How institutional adoption is growing the Move ecosystem",
         "The evolution of Move from launch to mainstream adoption",
@@ -42,18 +53,27 @@
         "Knows how to securely store and transfer Move tokens",
         "Can explain Move gas fees, staking, and liquidity pools",
         "Analyzes Move tokenomics and project fundamentals",
+<<<<<<< HEAD
         "Understands risk management in move trading",
         "Can guide users on Move wallet options and security",
         "Explains Move DeFi protocols and their mechanics",
         "Knows Move development updates and roadmap",
         "Understands move security best practices",
+=======
+        "Understands risk management in Move trading",
+        "Can guide users on Move wallet options and security",
+        "Explains Move DeFi protocols and their mechanics",
+        "Knows Move development updates and roadmap",
+        "Understands Move security best practices",
+>>>>>>> prompt_graph
         "Follows Move ecosystem trends and major protocol launches"
     ],
     "messageExamples": [
         []
     ],
     "postExamples": [],
     "topics": [
+<<<<<<< HEAD
         "move token market analysis",
         "move DeFi yield opportunities",
         "move  security best practices",
@@ -64,6 +84,18 @@
         "How to secure a move wallet",
         "Comparing move staking options",
         "The future of move interoperability"
+=======
+        "Move token market analysis",
+        "Move DeFi yield opportunities",
+        "Move security best practices",
+        "Move NFTs and gaming",
+        "Move Move development",
+        "Move regulatory compliance",
+        "Move DEX trading strategies",
+        "How to secure a Move wallet",
+        "Comparing Move staking options",
+        "The future of Move interoperability"
+>>>>>>> prompt_graph
     ],
     "style": {
         "all": [
@@ -116,6 +148,10 @@
         "TRUSTED",
         "DIVERSIFIED"
     ],
+<<<<<<< HEAD
     "systemPrompt": "You are a professional Move blockchain expert focused on providing accurate, technical assistance. Your responses follow this structure: 1) For market queries: Analyze market sentiment using relevant data and provide clear indicators (Bullish/Bearish/Neutral) with supporting evidence. 2) For transaction requests (swaps/deposits/transfers): Verify wallet addresses and execute with appropriate security checks. 3) For general queries: Provide comprehensive technical information about the move ecosystem, focusing on accuracy and educational value. Always maintain objectivity and back claims with verifiable data.",
+=======
+    "systemPrompt": "You are a professional Move blockchain expert focused on providing accurate, technical assistance. Your responses follow this structure: 1) For market queries: Analyze market sentiment using relevant data and provide clear indicators (Bullish/Bearish/Neutral) with supporting evidence. 2) For transaction requests (swaps/deposits/transfers): Verify wallet addresses and execute with appropriate security checks. 3) For general queries: Provide comprehensive technical information about the Move ecosystem, focusing on accuracy and educational value. Always maintain objectivity and back claims with verifiable data.",
+>>>>>>> prompt_graph
     "providers": []
 }

characters/suiHelper.character.json

@@ -21,9 +21,10 @@
 	"dependencies": {
 		"@elizaos/core": "workspace:*",
 		"@elizaos/plugin-image-generation": "workspace:*",
-		"@elizaos/plugin-tee-verifiable-log": "workspace:*",
-		"@elizaos/plugin-tee-log": "workspace:*",
 		"@elizaos/plugin-sui": "workspace:*",
+		"@elizaos/plugin-tee-log": "workspace:*",
+		"@elizaos/plugin-tee-verifiable-log": "workspace:*",
+		"@google/generative-ai": "^0.21.0",
 		"@types/body-parser": "1.19.5",
 		"@types/cors": "2.8.17",
 		"@types/express": "5.0.0",

packages/client-direct/package.json

@@ -4,6 +4,7 @@ import cors from "cors";
 import path from "path";
 import fs from "fs";
 import { getFolderByUserAddress } from "./tusky";
+import {getFilesByParentId} from "./tusky";
 
 import {
     type AgentRuntime,
@@ -15,10 +16,11 @@ import {
     type Character,
 } from "@elizaos/core";
 
-import type { TeeLogQuery, TeeLogService } from "@elizaos/plugin-tee-log";
+import { TeeLogService, type TeeLogQuery } from "@elizaos/plugin-tee-log";
 import { REST, Routes } from "discord.js";
 import type { DirectClient } from ".";
 import { validateUuid } from "@elizaos/core";
+import { GoogleGenerativeAI } from "@google/generative-ai";
 
 interface UUIDParams {
     agentId: UUID;
@@ -51,6 +53,26 @@ function validateUUIDParams(
     return { agentId };
 }
 
+async function callGemini(prompt: string) {
+    try {
+        const genAI = new GoogleGenerativeAI(process.env.GOOGLE_GENERATIVE_AI_API_KEY || '');
+        const model = genAI.getGenerativeModel({ model: "gemini-2.0-flash" });
+        
+        const result = await model.generateContent(prompt);
+        const response = await result.response;
+        const text = response.text();
+        
+        // Strip markdown and return raw text
+        return text.replace(/```json\n?/g, '').replace(/```\n?/g, '').trim();
+    } catch (error) {
+        console.error('Gemini API error:', error);
+        return { 
+            error: "Failed to get Gemini response",
+            details: error.message
+        };
+    }
+}
+
 export function createApiRouter(
     agents: Map<string, AgentRuntime>,
     directClient: DirectClient
@@ -86,62 +108,114 @@ export function createApiRouter(
         res.header("Access-Control-Allow-Headers", "Content-Type");
 
         try {
-            const userAddress = "0xb4b291607e91da4654cab88e5e35ba2921ef68f1b43725ef2faeae045bf5915d";
-            const rawData = await getFolderByUserAddress(userAddress);
+            const parentId = String(req.query.parentId || "45c6c728-6e0d-4260-8c2e-1bb25d285874");
+            
+            // 🔥 Lấy dữ liệu từ database
+            let rawData = await getFilesByParentId(parentId);
 
             if (!rawData || typeof rawData === "string") {
-                throw new Error("No valid data found");
+                throw new Error('No valid data found');
             }
 
-            const datapost = rawData.map((item) => {
-                const data = item.data;
-                if (Array.isArray(data)) {
-                    return data.map(i => i.text).filter(Boolean);
-                } else if (typeof data === "object" && data !== null) {
-                    return data.text ? [data.text] : [];
-                }
-                return [];
-            }).flat();
-    
-            if (datapost.length === 0) {
-                throw new Error("No valid text content found in the data");
-            }
-    
-            // **Tự động phát hiện danh mục dựa trên từ khóa xuất hiện**
-            const detectCategory = (post) => {
-                if (post.match(/\b(crypto|bitcoin|eth|nft|blockchain|web3|airdrop|token|memecoin|wallet|sui|solana)\b/i)) return "CRYPTO";
-                if (post.match(/\b(ai|ml|machine learning|neural|chatbot|grok|xai)\b/i)) return "ML_AI";
-                if (post.match(/\b(dev|code|protocol|extension|api|sdk|framework|smartcontract|dapp)\b/i)) return "DEVELOPMENT";
-                if (post.match(/\b(defi|finance|payment|trading|swap|yield|lending|staking)\b/i)) return "DEFI";
-                if (post.match(/\b(twitter|social|community|follow|rt|like|engagement)\b/i)) return "SOCIAL";
-                if (post.match(/\b(market|price|pump|dump|bull|bear|trend|season|altcoin)\b/i)) return "MARKET";
-                if (post.match(/\b(news|update|announcement|launch|release|progress)\b/i)) return "NEWS";
-                if (post.match(/\b(game|gaming|play|reward|naruto)\b/i)) return "GAMING";
-                if (post.match(/\b(event|hackathon|competition|prize|register|join)\b/i)) return "EVENTS";
-                return "Other"; // Nếu không xác định được danh mục
-            };
-    
-            // **Nhóm bài viết theo danh mục tự động**
-            const categorizedPosts = {};
+            // 🔥 Chuyển dữ liệu về dạng chuẩn
+            const authorCounts = {};
+            const formattedData = rawData.flatMap(item => {
+                const dataArray = Array.isArray(item.data) ? item.data : [item.data];
+                return dataArray.map(tweet => {
+                    const author = tweet.authorFullname || "anonymous";
+                    authorCounts[author] = (authorCounts[author] || 0) + 1;
 
-            datapost.forEach((post, index) => {
-                const category = detectCategory(post);
-                if (!categorizedPosts[category]) {
-                    categorizedPosts[category] = [];
-                }
-                categorizedPosts[category].push({ id: index + 1, content: post });
+                    return {
+                        id: `${author}_${authorCounts[author]}`,
+                        authorFullname: author,
+                        text: tweet.text,
+                        url: tweet.url
+                    };
+                });
             });
 
-            const result = Object.keys(categorizedPosts).map(category => {
-                const posts = categorizedPosts[category];
-                return {
-                    parentId: category,
-                    content: posts[0].content, 
-                    posts: posts.slice(1) 
-                };
-            });
+            // 🔥 Xây dựng prompt AI
+            const aiPrompt = `
+🔹 🔹 **Mục tiêu**
+- Chuyển danh sách bài đăng thành một mạng lưới gồm **nodes** (bài đăng, từ khóa quan trọng) và **edges** (mối quan hệ giữa chúng).
+- **Hashtags (#)** và **mentions (@)** chỉ được thêm vào danh sách keywords **nếu có nhiều bài đăng liên quan**.
+
+🔹 **Bước 1: Xử lý văn bản bài đăng**
+- Loại bỏ **URL** (ví dụ: "https://example.com").
+- Tách các **hashtags (#hashtag)** và **mentions (@username)**.
+- Loại bỏ ký tự đặc biệt **(trừ @ và #)**.
+- Chuyển toàn bộ chữ thành **chữ thường**.
+- **Bỏ qua bài đăng** nếu có ít hơn 5 từ.
+
+🔹 **Bước 2: Trích xuất từ khóa & hashtags**
+- **Chỉ giữ lại hashtags & mentions nếu xuất hiện trong từ 2 bài đăng trở lên**.
+- Bỏ hashtags & mentions nếu chỉ xuất hiện 1 lần.
+- Giữ lại **các từ khóa quan trọng** như **"blockchain", "zk-proof", "KYC", "DeFi", "wallet"**.
+
+🔹 **Bước 3: Xây dựng đồ thị**
+- **Nodes (nút):**
+  - Mỗi bài đăng là một node:
+    \`{ "id": "Movement_1", "type": "post" }\`
+  - Mỗi từ khóa quan trọng **(bao gồm các hashtags/mentions phổ biến)** là một node:
+    \`{ "id": "#defi", "type": "keyword" }\`
+    \`{ "id": "@elonmusk", "type": "keyword" }\`
+
+- **Edges (cạnh):**
+  - Kết nối bài đăng với từ khóa.
+  - Kết nối bài đăng nếu có chung hashtag hoặc mention xuất hiện **trong ít nhất 2 bài**.
+  - Ví dụ:
+    \`{ "source": "Movement_1", "target": "#defi" }\`
+    \`{ "source": "Movement_1", "target": "rushi_2" }\` (nếu cả hai có cùng hashtag)
+
+🔹 **Dữ liệu đầu vào (JSON)**
+Dưới đây là danh sách bài đăng:
+${JSON.stringify(formattedData, null, 2)}
+
+🔹 **Dữ liệu đầu ra mong muốn**
+- Trả về JSON với **nodes** và **edges** theo format sau:
+\`\`\`json
+{
+   "nodes": [
+        { "id": "Movement_1", "type": "post" },
+        { "id": "#defi", "type": "keyword" },
+        { "id": "@elonmusk", "type": "keyword" }
+   ],
+   "edges": [
+        { "source": "Movement_1", "target": "#defi" },
+        { "source": "Movement_1", "target": "@elonmusk" },
+        { "source": "rushi_2", "target": "#defi" }
+   ]
+}
+\`\`\`
+- **Chỉ trả về JSON**, không có văn bản thừa.
+- Giữ định dạng JSON chuẩn để có thể lưu vào file và sử dụng trực tiếp.
+`;
+    
+            // 🔥 Gọi Gemini API
+            const aiResponse = await callGemini(aiPrompt);
 
-            res.json(result);
+            if (typeof aiResponse === 'string') {
+                const graphData = JSON.parse(aiResponse);
+                
+                // Map content and url to post nodes
+                const contentMap = formattedData.reduce((map, item) => {
+                    map[item.id] = {
+                        text: item.text || "",
+                        url: item.url || ""
+                    };
+                    return map;
+                }, {});
+
+                graphData.nodes = graphData.nodes.map(node => ({
+                    ...node,
+                    content: node.type === "post" ? contentMap[node.id]?.text : undefined,
+                    url: node.type === "post" ? contentMap[node.id]?.url : undefined
+                }));
+
+                res.json(graphData);
+            } else {
+                throw new Error(aiResponse.error);
+            }
 
         } catch (error) {
             res.status(500).json({
@@ -392,17 +466,15 @@ export function createApiRouter(
 
             for (const agentRuntime of agents.values()) {
                 const teeLogService = agentRuntime
-                    .getService<TeeLogService>(ServiceType.TEE_LOG)
-                    .getInstance();
+                    .getService(ServiceType.TEE_LOG) as InstanceType<typeof TeeLogService>;
 
                 const agents = await teeLogService.getAllAgents();
                 allAgents.push(...agents);
             }
 
             const runtime: AgentRuntime = agents.values().next().value;
             const teeLogService = runtime
-                .getService<TeeLogService>(ServiceType.TEE_LOG)
-                .getInstance();
+                .getService(ServiceType.TEE_LOG) as InstanceType<typeof TeeLogService>;
             const attestation = await teeLogService.generateAttestation(
                 JSON.stringify(allAgents)
             );
@@ -425,8 +497,7 @@ export function createApiRouter(
             }
 
             const teeLogService = agentRuntime
-                .getService<TeeLogService>(ServiceType.TEE_LOG)
-                .getInstance();
+                .getService(ServiceType.TEE_LOG) as InstanceType<typeof TeeLogService>;
 
             const teeAgent = await teeLogService.getAgent(agentId);
             const attestation = await teeLogService.generateAttestation(
@@ -460,8 +531,7 @@ export function createApiRouter(
                 };
                 const agentRuntime: AgentRuntime = agents.values().next().value;
                 const teeLogService = agentRuntime
-                    .getService<TeeLogService>(ServiceType.TEE_LOG)
-                    .getInstance();
+                    .getService(ServiceType.TEE_LOG) as InstanceType<typeof TeeLogService>;
                 const pageQuery = await teeLogService.getLogs(
                     teeLogQuery,
                     page,

packages/client-direct/src/api.ts

@@ -88,6 +88,36 @@ export async function getFolderByUserAddress(userAddress: string) {
     return data;
 }
 
+export async function getFilesByParentId(parentId: string) {
+  if (!TUS_API_URL || !TUS_API_KEY) {
+    throw new Error("TUS_API_URL or TUSKY_API_KEY is not set");
+  }
+
+  const response = await fetch(
+    `${TUS_API_URL}/files?vaultId=${DEFAULT_VAULT}&parentId=${parentId}`,
+    {
+      method: "GET",
+      headers: {
+        "Api-Key": TUS_API_KEY,
+      },
+    }
+  ).then((response) => response.json());
+
+  const data = await Promise.all(
+    response.items.map(async (item: any) => {
+      const file = await getDataByID(item.id);
+      return {
+        ...item,
+        data: file,
+      };
+    })
+  );
+
+  return data;
+}
+
+
+
 // create vault to store files, Vaults in Tusky are secure storage containers for files.
 export async function createVault(vaultName: string) {
     try {