Merge pull request #600 from bmgalego/add-knowledge

feat: add knowledge to state
elizaOS · Nov 26, 2024 · 6d49f57 · 6d49f57
2 parents db9a857 + baaccb6
commit 6d49f57
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 8 deletions.
diff --git a/packages/core/src/knowledge.ts b/packages/core/src/knowledge.ts
@@ -6,7 +6,7 @@ import { embeddingZeroVector } from "./memory.ts";
 import { splitChunks } from "./generation.ts";
 import elizaLogger from "./logger.ts";
 
-async function get(runtime: AgentRuntime, message: Memory): Promise<string[]> {
+async function get(runtime: AgentRuntime, message: Memory): Promise<KnowledgeItem[]> {
     const processed = preprocess(message.content.text);
     elizaLogger.log(`Querying knowledge for: ${processed}`);
     const embedding = await embed(runtime, processed);
@@ -36,10 +36,9 @@ async function get(runtime: AgentRuntime, message: Memory): Promise<string[]> {
         )
     );
 
-    const knowledge = knowledgeDocuments
+    return knowledgeDocuments
         .filter((memory) => memory !== null)
-        .map((memory) => memory.content.text);
-    return knowledge;
+        .map((memory) => ({ id: memory.id, content: memory.content }));
 }
 
 async function set(

diff --git a/packages/core/src/runtime.ts b/packages/core/src/runtime.ts
@@ -29,6 +29,7 @@ import {
     ICacheManager,
     IDatabaseAdapter,
     IMemoryManager,
+    KnowledgeItem,
     ModelClass,
     ModelProviderName,
     Plugin,
@@ -952,10 +953,14 @@ Text: ${attachment.text}
                 .join(" ");
         }
 
+
+        const knowledegeData = await knowledge.get(this, message);
+
         const formattedKnowledge = formatKnowledge(
-            await knowledge.get(this, message)
+            knowledegeData
         );
 
+
         const initialState = {
             agentId: this.agentId,
             agentName,
@@ -971,6 +976,7 @@ Text: ${attachment.text}
                       ]
                     : "",
             knowledge: formattedKnowledge,
+            knowledgeData: knowledegeData,
             // Recent interactions between the sender and receiver, formatted as messages
             recentMessageInteractions: formattedMessageInteractions,
             // Recent interactions between the sender and receiver, formatted as posts
@@ -1097,7 +1103,7 @@ Text: ${attachment.text}
                     ? addHeader("# Attachments", formattedAttachments)
                     : "",
             ...additionalKeys,
-        };
+        } as State;
 
         const actionPromises = this.actions.map(async (action: Action) => {
             const result = await action.validate(this, message, initialState);
@@ -1235,6 +1241,6 @@ Text: ${attachment.text}
     }
 }
 
-const formatKnowledge = (knowledge: string[]) => {
-    return knowledge.map((knowledge) => `- ${knowledge}`).join("\n");
+const formatKnowledge = (knowledge: KnowledgeItem[]) => {
+    return knowledge.map((knowledge) => `- ${knowledge.content.text}`).join("\n");
 };
diff --git a/packages/core/src/types.ts b/packages/core/src/types.ts
@@ -294,6 +294,11 @@ export interface State {
     /** Optional formatted conversation */
     formattedConversation?: string;
 
+    /** Optional formatted knowledge */
+    knowledge?: string,
+    /** Optional knowledge data */
+    knowledgeData?: KnowledgeItem[],
+
     /** Additional dynamic properties */
     [key: string]: unknown;
 }