forked from elizaOS/eliza
-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat: add new plugins and enhance user data extraction capabilities
- Introduced `@ai16z/plugin-cortex` and `@ai16z/plugin-maitrix` for advanced agent functionalities. - Implemented user data extraction logic to gather name, location, and occupation from conversations. - Added evaluators and providers to support user data collection and emotional responses. - Updated character configurations to include new plugins and modified existing character settings. - Enhanced logging for better debugging and context tracking during data extraction processes.
- Loading branch information
1 parent
a61dc75
commit 263baad
Showing
24 changed files
with
23,686 additions
and
18,970 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
|
|
@@ -52,3 +52,4 @@ tsup.config.bundled_*.mjs | |
|
|
||
| coverage | ||
| .eslintcache | ||
| packages/client-twitter/src/interactions.ts.bak | ||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,249 @@ | ||
| import path from 'path'; | ||
| import { | ||
| IAgentRuntime, | ||
| Memory, | ||
| generateText, | ||
| ModelClass | ||
| } from "@ai16z/eliza"; | ||
|
|
||
| // Core interfaces | ||
| export interface UserData { | ||
| name: string | undefined; | ||
| description: string | undefined; | ||
| walletAddress: string | undefined; | ||
| lastUpdated: number; | ||
| isComplete: boolean; | ||
| confirmed: boolean; | ||
| userId: string; | ||
| } | ||
|
|
||
| export interface FieldGuidance { | ||
| field: keyof Omit<UserData, 'lastUpdated' | 'isComplete' | 'confirmed'>; | ||
| description: string; | ||
| validExamples: string[]; | ||
| invalidExamples: string[]; | ||
| extractionHints: string[]; | ||
| } | ||
|
|
||
| // Constants | ||
| export const CACHE_TTL = 24 * 60 * 60 * 1000; // 24 hours | ||
| export const BASE_CACHE_KEY = 'userdata'; | ||
|
|
||
| // Field-specific guidance | ||
| export const FIELD_GUIDANCE: FieldGuidance[] = [ | ||
| { | ||
| field: 'name', | ||
| description: "The desired name for the agent", | ||
| validExamples: [ | ||
| "Let's call the agent Alex", | ||
| "Name the agent Ava", | ||
| "The agent's name should be John" | ||
| ], | ||
| invalidExamples: [ | ||
| "My friend Mike suggested the name", | ||
| "I had an agent named Jay before", | ||
| "Agents often have names like Alexa or Siri" | ||
| ], | ||
| extractionHints: [ | ||
| "Must be a direct statement about the agent's name", | ||
| "Should be in present or future tense", | ||
| "Must not be a reference to other agents or people", | ||
| "Should be a definite, decided name" | ||
| ] | ||
| }, | ||
| { | ||
| field: 'description', | ||
| description: "A brief description of the agent's purpose or capabilities", | ||
| validExamples: [ | ||
| "The agent should help with scheduling and productivity", | ||
| "It's an AI assistant for managing my tasks", | ||
| "The agent will be focused on providing helpful information and answers" | ||
| ], | ||
| invalidExamples: [ | ||
| "I had an agent before that was good at math", | ||
| "Most AI assistants can answer questions", | ||
| "It would be cool if it could also play games" | ||
| ], | ||
| extractionHints: [ | ||
| "Must be a direct statement about this agent's description", | ||
| "Should describe intended purpose or capabilities", | ||
| "Must not be a reference to other agents or hypotheticals", | ||
| "Should be specific to this agent" | ||
| ] | ||
| }, | ||
| { | ||
| field: 'walletAddress', | ||
| description: "The user's cryptocurrency wallet address", | ||
| validExamples: [ | ||
| "My wallet address is 0x742d35Cc6634C0532925a3b844Bc454e4438f44e", | ||
| "0x1234567890123456789012345678901234567890 is my wallet", | ||
| "You can use this address for my wallet: 0x0987654321098765432109876543210987654321" | ||
| ], | ||
| invalidExamples: [ | ||
| "I had a different wallet address before", | ||
| "Most wallets have around 40 characters in the address", | ||
| "Let me check what my wallet address is" | ||
| ], | ||
| extractionHints: [ | ||
| "Must be a direct statement of the user's wallet address", | ||
| "Should be in the correct format for a wallet address", | ||
| "Must not be a reference to past or other wallets", | ||
| "Should be provided definitively by the user" | ||
| ] | ||
| } | ||
| ]; | ||
|
|
||
| export function formatFieldGuidance(missingFields: string[]): string { | ||
| return FIELD_GUIDANCE | ||
| .filter(g => missingFields.includes(g.field)) | ||
| .map(guidance => `${guidance.field.toUpperCase()}: | ||
| - Description: ${guidance.description} | ||
| - Looking for statements like: | ||
| ${guidance.validExamples.map(ex => `✓ "${ex}"`).join('\n ')} | ||
| - Ignore statements like: | ||
| ${guidance.invalidExamples.map(ex => `✗ "${ex}"`).join('\n ')} | ||
| - Requirements: | ||
| ${guidance.extractionHints.map(hint => `- ${hint}`).join('\n ')} | ||
| `).join('\n\n'); | ||
| } | ||
|
|
||
| // Template for information extraction | ||
| export const userDataExtractionTemplate = ` | ||
| TASK: Extract user information from conversation messages. | ||
| # START OF EXAMPLES | ||
| These are examples of the expected output of this task: | ||
| {{evaluatorExamples}} | ||
| # END OF EXAMPLES | ||
| # INSTRUCTIONS | ||
| Extract the following information about the user if directly stated: | ||
| - Name: The desired name for the agent | ||
| - Description: A brief description of the agent's purpose or capabilities | ||
| - Wallet Address: The user's cryptocurrency wallet address | ||
| Extract ONLY information that is: | ||
| - Directly stated by the user | ||
| - Clear and unambiguous | ||
| - Matches the example formats | ||
| Recent Messages: | ||
| {{recentMessages}} | ||
| Response must be a JSON array like this: | ||
| \`\`\`json | ||
| [ | ||
| {"field": "name", "value": "exact stated agent name"}, | ||
| {"field": "description", "value": "stated agent description"}, | ||
| {"field": "walletAddress", "value": "user's provided wallet address"} | ||
| ] | ||
| \`\`\``; | ||
|
|
||
| // Helper function to get cache key | ||
| export function getCacheKey(agentName: string, userId: string): string { | ||
| return path.join(BASE_CACHE_KEY, agentName, userId); | ||
| } | ||
|
|
||
|
|
||
| export async function extractUserData(text: string, runtime: IAgentRuntime, message: Memory): Promise<void> { | ||
|
|
||
| const contextExtract = ` | ||
| Extract agent information like name, description / purpose / intent, and wallet address of the agents as a JSON array from ${text}. | ||
| # START OF EXAMPLES | ||
| These are examples of the expected output of this task: | ||
| {{evaluationExamples}} | ||
| # END OF EXAMPLES | ||
| # INSTRUCTIONS | ||
| Extract specific information about the user from the conversation: | ||
| - Extract only directly stated information about the user's name for the agent, description for the agent, and the user's wallet address | ||
| - Only include information stated by the user themselves | ||
| - Skip any ambiguous or uncertain information | ||
| - Ignore information about other people | ||
| - Each piece of information should be a separate object in the array | ||
| - The 'field' property should be one of 'name', 'description', or 'walletAddress' | ||
| - The 'value' property should contain the corresponding value as a string | ||
| - If no information is found for a field, do not include an object for that field | ||
| - The resulting array should be a valid JSON array, with objects separated by commas | ||
| - Do not include any other text, formatting, or markup - ONLY the JSON array | ||
| IMPORTANT: Return ONLY the raw JSON array. Do not include any markdown formatting, backticks, or language identifiers. | ||
| Recent Messages: | ||
| {{recentMessages}} | ||
| Response should be ONLY a valid JSON array, like this but not exactly this: | ||
| [ | ||
| {"field": "name", "value": "stated name for the agent"}, | ||
| {"field": "description", "value": "stated description or purpose for the agent"}, | ||
| {"field": "walletAddress", "value": "user's crypto wallet address"} | ||
| ] | ||
| ` | ||
|
|
||
| const responseExtract = await generateText({ | ||
| runtime, | ||
| context:contextExtract, | ||
| modelClass: ModelClass.SMALL, | ||
| stop: ["\n"], | ||
| }); | ||
|
|
||
| console.log("responseExtract from Twitter interaction.js: "); | ||
| console.log(responseExtract); | ||
|
|
||
| // Parse the extracted text as JSON array | ||
| let extractedData: Array<{field: string, value: string}> = []; | ||
| try { | ||
| extractedData = JSON.parse(responseExtract); | ||
| } catch (error) { | ||
| console.error("Error parsing JSON array:", error); | ||
| console.error("Extracted text:", responseExtract); | ||
| return; | ||
| } | ||
|
|
||
| if (!extractedData || !Array.isArray(extractedData)) { | ||
| console.warn("Extracted data is not an array:", responseExtract); | ||
| return; | ||
| } | ||
|
|
||
| // Filter out objects with missing or invalid field/value properties | ||
| extractedData = extractedData.filter(data => | ||
| data.field && ['name', 'description', 'walletAddress'].includes(data.field) && | ||
| data.value && typeof data.value === 'string' | ||
| ); | ||
|
|
||
| console.log("Extracted data"); | ||
| console.log(extractedData); | ||
|
|
||
| // Get or initialize user data | ||
| const cacheKey = getCacheKey(runtime.character.name, message.userId); | ||
| let userData = await runtime.cacheManager.get<UserData>(cacheKey) || { | ||
| name: undefined, | ||
| description: undefined, | ||
| walletAddress: undefined, | ||
| lastUpdated: Date.now(), | ||
| isComplete: false, | ||
| confirmed: false, | ||
| userId: message.userId | ||
| }; | ||
|
|
||
| // Update with new data | ||
| let updated = false; | ||
| for (const data of extractedData) { | ||
| const field = data.field as keyof Omit<UserData, 'lastUpdated' | 'isComplete' | 'confirmed'>; | ||
| if (!userData[field] && data.value?.trim()) { | ||
| userData[field] = data.value.trim(); | ||
| updated = true; | ||
| } | ||
| } | ||
|
|
||
| if (updated) { | ||
| userData.isComplete = Boolean(userData.name && userData.description && userData.walletAddress); | ||
| userData.lastUpdated = Date.now(); | ||
|
|
||
| await runtime.cacheManager.set(cacheKey, userData); | ||
| } | ||
|
|
||
| console.log("User data updated"); | ||
| console.log(userData); | ||
|
|
||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
|
|
@@ -5,6 +5,7 @@ | |
| "rootDir": "src" | ||
| }, | ||
| "include": [ | ||
| "src/**/*.ts" | ||
| "src/**/*.ts", | ||
| "src/interactions.ts.bak" | ||
| ] | ||
| } | ||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.