added right params for caching, length

elizaOS · Feb 14, 2025 · 8657273 · 8657273
1 parent 18dee81
commit 8657273
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 7 deletions.
diff --git a/.gitignore b/.gitignore
@@ -91,4 +91,6 @@ packages/*/dist/*
 **/.docusaurus/**
 **/docs/build/**
 
-**lock**
+**lock**
+
+packages/agent/local_**
diff --git a/packages/plugin-anthropic/src/index.ts b/packages/plugin-anthropic/src/index.ts
@@ -85,12 +85,13 @@ export const anthropicPlugin: Plugin = {
       return text;
     },
     [ModelClass.TEXT_EMBEDDING]: async (runtime, text: string | null) => {
-
-      // TODO: Make this fallback only!!!
-      // TODO: pass on cacheDir to FlagEmbedding.init
-      if (!text) return new Array(1536).fill(0);
-
-      const model = await FlagEmbedding.init({ model: EmbeddingModel.BGESmallENV15 });
+      // TODO: check if other plugin provides TEXT_EMBEDDING model 
+      // Runtime will break if openai was used for TEXT_EMBEDDING before
+      const model = await FlagEmbedding.init({ 
+        model: EmbeddingModel.BGESmallENV15,
+        cacheDir: runtime.cacheDir,
+        maxLength: 512
+      });
       const embedding = await model.queryEmbed(text);
 
       const finalEmbedding = Array.isArray(embedding)