@elizaos/plugin-ollama 1.0.0-beta.8

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Shaw Walters and elizaOS Contributors
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,103 @@
+# Ollama Plugin
+
+This plugin provides integration with Ollama's local models through the ElizaOS platform.
+
+## Usage
+
+Add the plugin to your character configuration:
+
+```json
+"plugins": ["@elizaos/plugin-ollama"]
+```
+
+## Configuration
+
+The plugin requires these environment variables (can be set in .env file or character settings):
+
+```json
+"settings": {
+  "OLLAMA_API_ENDPOINT": "http://localhost:11434/api",
+  "OLLAMA_SMALL_MODEL": "llama3",
+  "OLLAMA_MEDIUM_MODEL": "your_medium_model",
+  "OLLAMA_LARGE_MODEL": "gemma3:latest",
+  "OLLAMA_EMBEDDING_MODEL": "nomic-embed-text"
+}
+```
+
+Or in `.env` file:
+
+```
+OLLAMA_API_ENDPOINT=http://localhost:11434/api
+OLLAMA_SMALL_MODEL=llama3
+OLLAMA_MEDIUM_MODEL=your_medium_model
+OLLAMA_LARGE_MODEL=gemma3:latest
+OLLAMA_EMBEDDING_MODEL=nomic-embed-text
+```
+
+### Configuration Options
+
+- `OLLAMA_API_ENDPOINT`: Ollama API endpoint (default: http://localhost:11434/api)
+- `OLLAMA_SMALL_MODEL`: Model for simpler tasks (default: llama3)
+- `OLLAMA_MEDIUM_MODEL`: Medium-complexity model
+- `OLLAMA_LARGE_MODEL`: Model for complex tasks (default: gemma3:latest)
+- `OLLAMA_EMBEDDING_MODEL`: Model for text embeddings (default: nomic-embed-text)
+
+The plugin provides these model classes:
+
+- `TEXT_SMALL`: Optimized for fast responses with simpler prompts
+- `TEXT_LARGE`: For complex tasks requiring deeper reasoning
+- `TEXT_EMBEDDING`: Text embedding model
+- `OBJECT_SMALL`: JSON object generation with simpler models
+- `OBJECT_LARGE`: JSON object generation with more complex models
+
+## Additional Features
+
+### Text Generation (Small Model)
+
+```js
+const text = await runtime.useModel(ModelType.TEXT_SMALL, {
+  prompt: 'What is the nature of reality?',
+  stopSequences: [], // optional
+});
+```
+
+### Text Generation (Large Model)
+
+```js
+const text = await runtime.useModel(ModelType.TEXT_LARGE, {
+  prompt: 'Write a detailed explanation of quantum physics',
+  stopSequences: [], // optional
+  maxTokens: 8192, // optional (default: 8192)
+  temperature: 0.7, // optional (default: 0.7)
+  frequencyPenalty: 0.7, // optional (default: 0.7)
+  presencePenalty: 0.7, // optional (default: 0.7)
+});
+```
+
+### Text Embeddings
+
+```js
+const embedding = await runtime.useModel(ModelType.TEXT_EMBEDDING, {
+  text: 'Text to embed',
+});
+// or
+const embedding = await runtime.useModel(ModelType.TEXT_EMBEDDING, 'Text to embed');
+```
+
+### Object Generation (Small Model)
+
+```js
+const object = await runtime.useModel(ModelType.OBJECT_SMALL, {
+  prompt: 'Generate a JSON object representing a user profile',
+  temperature: 0.7, // optional
+});
+```
+
+### Object Generation (Large Model)
+
+```js
+const object = await runtime.useModel(ModelType.OBJECT_LARGE, {
+  prompt: 'Generate a detailed JSON object representing a restaurant',
+  temperature: 0.7, // optional
+});
+```

package/dist/index.js

@@ -0,0 +1,280 @@
+// src/index.ts
+import { ModelType, logger } from "@elizaos/core";
+import { generateObject, generateText } from "ai";
+import { createOllama } from "ollama-ai-provider";
+var OLLAMA_API_URL = "http://localhost:11434/api";
+async function generateOllamaText(ollama, model, params) {
+  try {
+    const { text: ollamaResponse } = await generateText({
+      model: ollama(model),
+      prompt: params.prompt,
+      system: params.system,
+      temperature: params.temperature,
+      maxTokens: params.maxTokens,
+      frequencyPenalty: params.frequencyPenalty,
+      presencePenalty: params.presencePenalty,
+      stopSequences: params.stopSequences
+    });
+    return ollamaResponse;
+  } catch (error) {
+    logger.error("Error in generateOllamaText:", error);
+    return "Error generating text. Please try again later.";
+  }
+}
+async function generateOllamaObject(ollama, model, params) {
+  try {
+    const { object } = await generateObject({
+      model: ollama(model),
+      output: "no-schema",
+      prompt: params.prompt,
+      temperature: params.temperature
+    });
+    return object;
+  } catch (error) {
+    logger.error("Error generating object:", error);
+    return {};
+  }
+}
+var ollamaPlugin = {
+  name: "ollama",
+  description: "Ollama plugin",
+  config: {
+    OLLAMA_API_ENDPOINT: process.env.OLLAMA_API_ENDPOINT,
+    OLLAMA_SMALL_MODEL: process.env.OLLAMA_SMALL_MODEL,
+    OLLAMA_MEDIUM_MODEL: process.env.OLLAMA_MEDIUM_MODEL,
+    OLLAMA_LARGE_MODEL: process.env.OLLAMA_LARGE_MODEL,
+    OLLAMA_EMBEDDING_MODEL: process.env.OLLAMA_EMBEDDING_MODEL
+  },
+  models: {
+    [ModelType.TEXT_EMBEDDING]: async (runtime, params) => {
+      try {
+        const ollama = createOllama({
+          fetch: runtime.fetch,
+          baseURL: runtime.getSetting("OLLAMA_API_ENDPOINT") || OLLAMA_API_URL
+        });
+        const modelName = runtime.getSetting("OLLAMA_EMBEDDING_MODEL") || "nomic-embed-text";
+        const text = typeof params === "string" ? params : params?.text || "";
+        if (!text) {
+          logger.error("No text provided for embedding");
+          return Array(1536).fill(0);
+        }
+        try {
+          const response = await fetch(
+            `${runtime.getSetting("OLLAMA_API_ENDPOINT") || OLLAMA_API_URL}/embeddings`,
+            {
+              method: "POST",
+              headers: { "Content-Type": "application/json" },
+              body: JSON.stringify({
+                model: modelName,
+                prompt: text
+              })
+            }
+          );
+          if (!response.ok) {
+            throw new Error(`Embedding request failed: ${response.statusText}`);
+          }
+          const result = await response.json();
+          return result.embedding || Array(1536).fill(0);
+        } catch (embeddingError) {
+          logger.error("Error generating embedding:", embeddingError);
+          return Array(1536).fill(0);
+        }
+      } catch (error) {
+        logger.error("Error in TEXT_EMBEDDING model:", error);
+        return Array(1536).fill(0);
+      }
+    },
+    [ModelType.TEXT_SMALL]: async (runtime, { prompt, stopSequences = [] }) => {
+      try {
+        const temperature = 0.7;
+        const frequency_penalty = 0.7;
+        const presence_penalty = 0.7;
+        const max_response_length = 8e3;
+        const ollama = createOllama({
+          fetch: runtime.fetch,
+          baseURL: runtime.getSetting("OLLAMA_API_ENDPOINT") || OLLAMA_API_URL
+        });
+        const model = runtime.getSetting("OLLAMA_SMALL_MODEL") ?? runtime.getSetting("SMALL_MODEL") ?? "llama3";
+        logger.log("generating text");
+        logger.log(prompt);
+        return await generateOllamaText(ollama, model, {
+          prompt,
+          system: runtime.character.system ?? void 0,
+          temperature,
+          maxTokens: max_response_length,
+          frequencyPenalty: frequency_penalty,
+          presencePenalty: presence_penalty,
+          stopSequences
+        });
+      } catch (error) {
+        logger.error("Error in TEXT_SMALL model:", error);
+        return "Error generating text. Please try again later.";
+      }
+    },
+    [ModelType.TEXT_LARGE]: async (runtime, {
+      prompt,
+      stopSequences = [],
+      maxTokens = 8192,
+      temperature = 0.7,
+      frequencyPenalty = 0.7,
+      presencePenalty = 0.7
+    }) => {
+      try {
+        const model = runtime.getSetting("OLLAMA_LARGE_MODEL") ?? runtime.getSetting("LARGE_MODEL") ?? "gemma3:latest";
+        const ollama = createOllama({
+          fetch: runtime.fetch,
+          baseURL: runtime.getSetting("OLLAMA_API_ENDPOINT") || OLLAMA_API_URL
+        });
+        return await generateOllamaText(ollama, model, {
+          prompt,
+          system: runtime.character.system ?? void 0,
+          temperature,
+          maxTokens,
+          frequencyPenalty,
+          presencePenalty,
+          stopSequences
+        });
+      } catch (error) {
+        logger.error("Error in TEXT_LARGE model:", error);
+        return "Error generating text. Please try again later.";
+      }
+    },
+    [ModelType.OBJECT_SMALL]: async (runtime, params) => {
+      try {
+        const ollama = createOllama({
+          fetch: runtime.fetch,
+          baseURL: runtime.getSetting("OLLAMA_API_ENDPOINT") || OLLAMA_API_URL
+        });
+        const model = runtime.getSetting("OLLAMA_SMALL_MODEL") ?? runtime.getSetting("SMALL_MODEL") ?? "gemma3:latest";
+        if (params.schema) {
+          logger.info("Using OBJECT_SMALL without schema validation");
+        }
+        return await generateOllamaObject(ollama, model, params);
+      } catch (error) {
+        logger.error("Error in OBJECT_SMALL model:", error);
+        return {};
+      }
+    },
+    [ModelType.OBJECT_LARGE]: async (runtime, params) => {
+      try {
+        const ollama = createOllama({
+          fetch: runtime.fetch,
+          baseURL: runtime.getSetting("OLLAMA_API_ENDPOINT") || OLLAMA_API_URL
+        });
+        const model = runtime.getSetting("OLLAMA_LARGE_MODEL") ?? runtime.getSetting("LARGE_MODEL") ?? "gemma3:latest";
+        if (params.schema) {
+          logger.info("Using OBJECT_LARGE without schema validation");
+        }
+        return await generateOllamaObject(ollama, model, params);
+      } catch (error) {
+        logger.error("Error in OBJECT_LARGE model:", error);
+        return {};
+      }
+    }
+  },
+  tests: [
+    {
+      name: "ollama_plugin_tests",
+      tests: [
+        {
+          name: "ollama_test_url_validation",
+          fn: async (runtime) => {
+            try {
+              const baseURL = runtime.getSetting("OLLAMA_API_ENDPOINT") || OLLAMA_API_URL;
+              const response = await fetch(`${baseURL}/tags`);
+              const data = await response.json();
+              logger.log("Models Available:", data?.models?.length);
+              if (!response.ok) {
+                logger.error(`Failed to validate Ollama API: ${response.statusText}`);
+                return;
+              }
+            } catch (error) {
+              logger.error("Error in ollama_test_url_validation:", error);
+            }
+          }
+        },
+        {
+          name: "ollama_test_text_embedding",
+          fn: async (runtime) => {
+            try {
+              const embedding = await runtime.useModel(ModelType.TEXT_EMBEDDING, {
+                text: "Hello, world!"
+              });
+              logger.log("embedding", embedding);
+            } catch (error) {
+              logger.error("Error in test_text_embedding:", error);
+            }
+          }
+        },
+        {
+          name: "ollama_test_text_large",
+          fn: async (runtime) => {
+            try {
+              const text = await runtime.useModel(ModelType.TEXT_LARGE, {
+                prompt: "What is the nature of reality in 10 words?"
+              });
+              if (text.length === 0) {
+                logger.error("Failed to generate text");
+                return;
+              }
+              logger.log("generated with test_text_large:", text);
+            } catch (error) {
+              logger.error("Error in test_text_large:", error);
+            }
+          }
+        },
+        {
+          name: "ollama_test_text_small",
+          fn: async (runtime) => {
+            try {
+              const text = await runtime.useModel(ModelType.TEXT_SMALL, {
+                prompt: "What is the nature of reality in 10 words?"
+              });
+              if (text.length === 0) {
+                logger.error("Failed to generate text");
+                return;
+              }
+              logger.log("generated with test_text_small:", text);
+            } catch (error) {
+              logger.error("Error in test_text_small:", error);
+            }
+          }
+        },
+        {
+          name: "ollama_test_object_small",
+          fn: async (runtime) => {
+            try {
+              const object = await runtime.useModel(ModelType.OBJECT_SMALL, {
+                prompt: "Generate a JSON object representing a user profile with name, age, and hobbies",
+                temperature: 0.7
+              });
+              logger.log("Generated object:", object);
+            } catch (error) {
+              logger.error("Error in test_object_small:", error);
+            }
+          }
+        },
+        {
+          name: "ollama_test_object_large",
+          fn: async (runtime) => {
+            try {
+              const object = await runtime.useModel(ModelType.OBJECT_LARGE, {
+                prompt: "Generate a detailed JSON object representing a restaurant with name, cuisine type, menu items with prices, and customer reviews",
+                temperature: 0.7
+              });
+              logger.log("Generated object:", object);
+            } catch (error) {
+              logger.error("Error in test_object_large:", error);
+            }
+          }
+        }
+      ]
+    }
+  ]
+};
+var index_default = ollamaPlugin;
+export {
+  index_default as default,
+  ollamaPlugin
+};
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/index.ts"],"sourcesContent":["import type { ObjectGenerationParams, Plugin, TextEmbeddingParams } from '@elizaos/core';\nimport { type GenerateTextParams, ModelType, logger } from '@elizaos/core';\nimport { generateObject, generateText } from 'ai';\nimport { createOllama } from 'ollama-ai-provider';\n\n// Default Ollama API URL\nconst OLLAMA_API_URL = 'http://localhost:11434/api';\n\n/**\n * Generate text using Ollama API\n */\nasync function generateOllamaText(\n  ollama: ReturnType<typeof createOllama>,\n  model: string,\n  params: {\n    prompt: string;\n    system?: string;\n    temperature: number;\n    maxTokens: number;\n    frequencyPenalty: number;\n    presencePenalty: number;\n    stopSequences: string[];\n  }\n) {\n  try {\n    const { text: ollamaResponse } = await generateText({\n      model: ollama(model),\n      prompt: params.prompt,\n      system: params.system,\n      temperature: params.temperature,\n      maxTokens: params.maxTokens,\n      frequencyPenalty: params.frequencyPenalty,\n      presencePenalty: params.presencePenalty,\n      stopSequences: params.stopSequences,\n    });\n    return ollamaResponse;\n  } catch (error: unknown) {\n    logger.error('Error in generateOllamaText:', error);\n    return 'Error generating text. Please try again later.';\n  }\n}\n\n/**\n * Generate object using Ollama API with consistent error handling\n */\nasync function generateOllamaObject(\n  ollama: ReturnType<typeof createOllama>,\n  model: string,\n  params: ObjectGenerationParams\n) {\n  try {\n    const { object } = await generateObject({\n      model: ollama(model),\n      output: 'no-schema',\n      prompt: params.prompt,\n      temperature: params.temperature,\n    });\n    return object;\n  } catch (error: unknown) {\n    logger.error('Error generating object:', error);\n    return {};\n  }\n}\n\nexport const ollamaPlugin: Plugin = {\n  name: 'ollama',\n  description: 'Ollama plugin',\n  config: {\n    OLLAMA_API_ENDPOINT: process.env.OLLAMA_API_ENDPOINT,\n    OLLAMA_SMALL_MODEL: process.env.OLLAMA_SMALL_MODEL,\n    OLLAMA_MEDIUM_MODEL: process.env.OLLAMA_MEDIUM_MODEL,\n    OLLAMA_LARGE_MODEL: process.env.OLLAMA_LARGE_MODEL,\n    OLLAMA_EMBEDDING_MODEL: process.env.OLLAMA_EMBEDDING_MODEL,\n  },\n  models: {\n    [ModelType.TEXT_EMBEDDING]: async (\n      runtime,\n      params: TextEmbeddingParams | string | null\n    ): Promise<number[]> => {\n      try {\n        const ollama = createOllama({\n          fetch: runtime.fetch,\n          baseURL: runtime.getSetting('OLLAMA_API_ENDPOINT') || OLLAMA_API_URL,\n        });\n\n        const modelName = runtime.getSetting('OLLAMA_EMBEDDING_MODEL') || 'nomic-embed-text';\n        const text =\n          typeof params === 'string' ? params : (params as TextEmbeddingParams)?.text || '';\n\n        if (!text) {\n          logger.error('No text provided for embedding');\n          return Array(1536).fill(0);\n        }\n\n        // Generate embeddings - note we're using a simpler approach since generateEmbedding\n        // may not be available in the current version of the AI SDK\n        try {\n          // This is simplified and may need to be adjusted based on the actual API\n          const response = await fetch(\n            `${runtime.getSetting('OLLAMA_API_ENDPOINT') || OLLAMA_API_URL}/embeddings`,\n            {\n              method: 'POST',\n              headers: { 'Content-Type': 'application/json' },\n              body: JSON.stringify({\n                model: modelName,\n                prompt: text,\n              }),\n            }\n          );\n\n          if (!response.ok) {\n            throw new Error(`Embedding request failed: ${response.statusText}`);\n          }\n\n          const result = (await response.json()) as { embedding?: number[] };\n          return result.embedding || Array(1536).fill(0);\n        } catch (embeddingError) {\n          logger.error('Error generating embedding:', embeddingError);\n          return Array(1536).fill(0);\n        }\n      } catch (error) {\n        logger.error('Error in TEXT_EMBEDDING model:', error);\n        // Return a fallback vector rather than crashing\n        return Array(1536).fill(0);\n      }\n    },\n    [ModelType.TEXT_SMALL]: async (runtime, { prompt, stopSequences = [] }: GenerateTextParams) => {\n      try {\n        const temperature = 0.7;\n        const frequency_penalty = 0.7;\n        const presence_penalty = 0.7;\n        const max_response_length = 8000;\n        const ollama = createOllama({\n          fetch: runtime.fetch,\n          baseURL: runtime.getSetting('OLLAMA_API_ENDPOINT') || OLLAMA_API_URL,\n        });\n\n        const model =\n          runtime.getSetting('OLLAMA_SMALL_MODEL') ?? runtime.getSetting('SMALL_MODEL') ?? 'llama3';\n\n        logger.log('generating text');\n        logger.log(prompt);\n\n        return await generateOllamaText(ollama, model, {\n          prompt,\n          system: runtime.character.system ?? undefined,\n          temperature,\n          maxTokens: max_response_length,\n          frequencyPenalty: frequency_penalty,\n          presencePenalty: presence_penalty,\n          stopSequences,\n        });\n      } catch (error) {\n        logger.error('Error in TEXT_SMALL model:', error);\n        return 'Error generating text. Please try again later.';\n      }\n    },\n    [ModelType.TEXT_LARGE]: async (\n      runtime,\n      {\n        prompt,\n        stopSequences = [],\n        maxTokens = 8192,\n        temperature = 0.7,\n        frequencyPenalty = 0.7,\n        presencePenalty = 0.7,\n      }: GenerateTextParams\n    ) => {\n      try {\n        const model =\n          runtime.getSetting('OLLAMA_LARGE_MODEL') ??\n          runtime.getSetting('LARGE_MODEL') ??\n          'gemma3:latest';\n        const ollama = createOllama({\n          fetch: runtime.fetch,\n          baseURL: runtime.getSetting('OLLAMA_API_ENDPOINT') || OLLAMA_API_URL,\n        });\n\n        return await generateOllamaText(ollama, model, {\n          prompt,\n          system: runtime.character.system ?? undefined,\n          temperature,\n          maxTokens,\n          frequencyPenalty,\n          presencePenalty,\n          stopSequences,\n        });\n      } catch (error) {\n        logger.error('Error in TEXT_LARGE model:', error);\n        return 'Error generating text. Please try again later.';\n      }\n    },\n    [ModelType.OBJECT_SMALL]: async (runtime, params: ObjectGenerationParams) => {\n      try {\n        const ollama = createOllama({\n          fetch: runtime.fetch,\n          baseURL: runtime.getSetting('OLLAMA_API_ENDPOINT') || OLLAMA_API_URL,\n        });\n        const model =\n          runtime.getSetting('OLLAMA_SMALL_MODEL') ??\n          runtime.getSetting('SMALL_MODEL') ??\n          'gemma3:latest';\n\n        if (params.schema) {\n          logger.info('Using OBJECT_SMALL without schema validation');\n        }\n\n        return await generateOllamaObject(ollama, model, params);\n      } catch (error) {\n        logger.error('Error in OBJECT_SMALL model:', error);\n        // Return empty object instead of crashing\n        return {};\n      }\n    },\n    [ModelType.OBJECT_LARGE]: async (runtime, params: ObjectGenerationParams) => {\n      try {\n        const ollama = createOllama({\n          fetch: runtime.fetch,\n          baseURL: runtime.getSetting('OLLAMA_API_ENDPOINT') || OLLAMA_API_URL,\n        });\n        const model =\n          runtime.getSetting('OLLAMA_LARGE_MODEL') ??\n          runtime.getSetting('LARGE_MODEL') ??\n          'gemma3:latest';\n\n        if (params.schema) {\n          logger.info('Using OBJECT_LARGE without schema validation');\n        }\n\n        return await generateOllamaObject(ollama, model, params);\n      } catch (error) {\n        logger.error('Error in OBJECT_LARGE model:', error);\n        // Return empty object instead of crashing\n        return {};\n      }\n    },\n  },\n  tests: [\n    {\n      name: 'ollama_plugin_tests',\n      tests: [\n        {\n          name: 'ollama_test_url_validation',\n          fn: async (runtime) => {\n            try {\n              const baseURL = runtime.getSetting('OLLAMA_API_ENDPOINT') || OLLAMA_API_URL;\n              const response = await fetch(`${baseURL}/tags`);\n              const data = await response.json();\n              logger.log('Models Available:', (data as { models: unknown[] })?.models?.length);\n              if (!response.ok) {\n                logger.error(`Failed to validate Ollama API: ${response.statusText}`);\n                return;\n              }\n            } catch (error) {\n              logger.error('Error in ollama_test_url_validation:', error);\n            }\n          },\n        },\n        {\n          name: 'ollama_test_text_embedding',\n          fn: async (runtime) => {\n            try {\n              const embedding = await runtime.useModel(ModelType.TEXT_EMBEDDING, {\n                text: 'Hello, world!',\n              });\n              logger.log('embedding', embedding);\n            } catch (error) {\n              logger.error('Error in test_text_embedding:', error);\n            }\n          },\n        },\n        {\n          name: 'ollama_test_text_large',\n          fn: async (runtime) => {\n            try {\n              const text = await runtime.useModel(ModelType.TEXT_LARGE, {\n                prompt: 'What is the nature of reality in 10 words?',\n              });\n              if (text.length === 0) {\n                logger.error('Failed to generate text');\n                return;\n              }\n              logger.log('generated with test_text_large:', text);\n            } catch (error) {\n              logger.error('Error in test_text_large:', error);\n            }\n          },\n        },\n        {\n          name: 'ollama_test_text_small',\n          fn: async (runtime) => {\n            try {\n              const text = await runtime.useModel(ModelType.TEXT_SMALL, {\n                prompt: 'What is the nature of reality in 10 words?',\n              });\n              if (text.length === 0) {\n                logger.error('Failed to generate text');\n                return;\n              }\n              logger.log('generated with test_text_small:', text);\n            } catch (error) {\n              logger.error('Error in test_text_small:', error);\n            }\n          },\n        },\n        {\n          name: 'ollama_test_object_small',\n          fn: async (runtime) => {\n            try {\n              const object = await runtime.useModel(ModelType.OBJECT_SMALL, {\n                prompt:\n                  'Generate a JSON object representing a user profile with name, age, and hobbies',\n                temperature: 0.7,\n              });\n              logger.log('Generated object:', object);\n            } catch (error) {\n              logger.error('Error in test_object_small:', error);\n            }\n          },\n        },\n        {\n          name: 'ollama_test_object_large',\n          fn: async (runtime) => {\n            try {\n              const object = await runtime.useModel(ModelType.OBJECT_LARGE, {\n                prompt:\n                  'Generate a detailed JSON object representing a restaurant with name, cuisine type, menu items with prices, and customer reviews',\n                temperature: 0.7,\n              });\n              logger.log('Generated object:', object);\n            } catch (error) {\n              logger.error('Error in test_object_large:', error);\n            }\n          },\n        },\n      ],\n    },\n  ],\n};\nexport default ollamaPlugin;\n"],"mappings":";AACA,SAAkC,WAAW,cAAc;AAC3D,SAAS,gBAAgB,oBAAoB;AAC7C,SAAS,oBAAoB;AAG7B,IAAM,iBAAiB;AAKvB,eAAe,mBACb,QACA,OACA,QASA;AACA,MAAI;AACF,UAAM,EAAE,MAAM,eAAe,IAAI,MAAM,aAAa;AAAA,MAClD,OAAO,OAAO,KAAK;AAAA,MACnB,QAAQ,OAAO;AAAA,MACf,QAAQ,OAAO;AAAA,MACf,aAAa,OAAO;AAAA,MACpB,WAAW,OAAO;AAAA,MAClB,kBAAkB,OAAO;AAAA,MACzB,iBAAiB,OAAO;AAAA,MACxB,eAAe,OAAO;AAAA,IACxB,CAAC;AACD,WAAO;AAAA,EACT,SAAS,OAAgB;AACvB,WAAO,MAAM,gCAAgC,KAAK;AAClD,WAAO;AAAA,EACT;AACF;AAKA,eAAe,qBACb,QACA,OACA,QACA;AACA,MAAI;AACF,UAAM,EAAE,OAAO,IAAI,MAAM,eAAe;AAAA,MACtC,OAAO,OAAO,KAAK;AAAA,MACnB,QAAQ;AAAA,MACR,QAAQ,OAAO;AAAA,MACf,aAAa,OAAO;AAAA,IACtB,CAAC;AACD,WAAO;AAAA,EACT,SAAS,OAAgB;AACvB,WAAO,MAAM,4BAA4B,KAAK;AAC9C,WAAO,CAAC;AAAA,EACV;AACF;AAEO,IAAM,eAAuB;AAAA,EAClC,MAAM;AAAA,EACN,aAAa;AAAA,EACb,QAAQ;AAAA,IACN,qBAAqB,QAAQ,IAAI;AAAA,IACjC,oBAAoB,QAAQ,IAAI;AAAA,IAChC,qBAAqB,QAAQ,IAAI;AAAA,IACjC,oBAAoB,QAAQ,IAAI;AAAA,IAChC,wBAAwB,QAAQ,IAAI;AAAA,EACtC;AAAA,EACA,QAAQ;AAAA,IACN,CAAC,UAAU,cAAc,GAAG,OAC1B,SACA,WACsB;AACtB,UAAI;AACF,cAAM,SAAS,aAAa;AAAA,UAC1B,OAAO,QAAQ;AAAA,UACf,SAAS,QAAQ,WAAW,qBAAqB,KAAK;AAAA,QACxD,CAAC;AAED,cAAM,YAAY,QAAQ,WAAW,wBAAwB,KAAK;AAClE,cAAM,OACJ,OAAO,WAAW,WAAW,SAAU,QAAgC,QAAQ;AAEjF,YAAI,CAAC,MAAM;AACT,iBAAO,MAAM,gCAAgC;AAC7C,iBAAO,MAAM,IAAI,EAAE,KAAK,CAAC;AAAA,QAC3B;AAIA,YAAI;AAEF,gBAAM,WAAW,MAAM;AAAA,YACrB,GAAG,QAAQ,WAAW,qBAAqB,KAAK,cAAc;AAAA,YAC9D;AAAA,cACE,QAAQ;AAAA,cACR,SAAS,EAAE,gBAAgB,mBAAmB;AAAA,cAC9C,MAAM,KAAK,UAAU;AAAA,gBACnB,OAAO;AAAA,gBACP,QAAQ;AAAA,cACV,CAAC;AAAA,YACH;AAAA,UACF;AAEA,cAAI,CAAC,SAAS,IAAI;AAChB,kBAAM,IAAI,MAAM,6BAA6B,SAAS,UAAU,EAAE;AAAA,UACpE;AAEA,gBAAM,SAAU,MAAM,SAAS,KAAK;AACpC,iBAAO,OAAO,aAAa,MAAM,IAAI,EAAE,KAAK,CAAC;AAAA,QAC/C,SAAS,gBAAgB;AACvB,iBAAO,MAAM,+BAA+B,cAAc;AAC1D,iBAAO,MAAM,IAAI,EAAE,KAAK,CAAC;AAAA,QAC3B;AAAA,MACF,SAAS,OAAO;AACd,eAAO,MAAM,kCAAkC,KAAK;AAEpD,eAAO,MAAM,IAAI,EAAE,KAAK,CAAC;AAAA,MAC3B;AAAA,IACF;AAAA,IACA,CAAC,UAAU,UAAU,GAAG,OAAO,SAAS,EAAE,QAAQ,gBAAgB,CAAC,EAAE,MAA0B;AAC7F,UAAI;AACF,cAAM,cAAc;AACpB,cAAM,oBAAoB;AAC1B,cAAM,mBAAmB;AACzB,cAAM,sBAAsB;AAC5B,cAAM,SAAS,aAAa;AAAA,UAC1B,OAAO,QAAQ;AAAA,UACf,SAAS,QAAQ,WAAW,qBAAqB,KAAK;AAAA,QACxD,CAAC;AAED,cAAM,QACJ,QAAQ,WAAW,oBAAoB,KAAK,QAAQ,WAAW,aAAa,KAAK;AAEnF,eAAO,IAAI,iBAAiB;AAC5B,eAAO,IAAI,MAAM;AAEjB,eAAO,MAAM,mBAAmB,QAAQ,OAAO;AAAA,UAC7C;AAAA,UACA,QAAQ,QAAQ,UAAU,UAAU;AAAA,UACpC;AAAA,UACA,WAAW;AAAA,UACX,kBAAkB;AAAA,UAClB,iBAAiB;AAAA,UACjB;AAAA,QACF,CAAC;AAAA,MACH,SAAS,OAAO;AACd,eAAO,MAAM,8BAA8B,KAAK;AAChD,eAAO;AAAA,MACT;AAAA,IACF;AAAA,IACA,CAAC,UAAU,UAAU,GAAG,OACtB,SACA;AAAA,MACE;AAAA,MACA,gBAAgB,CAAC;AAAA,MACjB,YAAY;AAAA,MACZ,cAAc;AAAA,MACd,mBAAmB;AAAA,MACnB,kBAAkB;AAAA,IACpB,MACG;AACH,UAAI;AACF,cAAM,QACJ,QAAQ,WAAW,oBAAoB,KACvC,QAAQ,WAAW,aAAa,KAChC;AACF,cAAM,SAAS,aAAa;AAAA,UAC1B,OAAO,QAAQ;AAAA,UACf,SAAS,QAAQ,WAAW,qBAAqB,KAAK;AAAA,QACxD,CAAC;AAED,eAAO,MAAM,mBAAmB,QAAQ,OAAO;AAAA,UAC7C;AAAA,UACA,QAAQ,QAAQ,UAAU,UAAU;AAAA,UACpC;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,QACF,CAAC;AAAA,MACH,SAAS,OAAO;AACd,eAAO,MAAM,8BAA8B,KAAK;AAChD,eAAO;AAAA,MACT;AAAA,IACF;AAAA,IACA,CAAC,UAAU,YAAY,GAAG,OAAO,SAAS,WAAmC;AAC3E,UAAI;AACF,cAAM,SAAS,aAAa;AAAA,UAC1B,OAAO,QAAQ;AAAA,UACf,SAAS,QAAQ,WAAW,qBAAqB,KAAK;AAAA,QACxD,CAAC;AACD,cAAM,QACJ,QAAQ,WAAW,oBAAoB,KACvC,QAAQ,WAAW,aAAa,KAChC;AAEF,YAAI,OAAO,QAAQ;AACjB,iBAAO,KAAK,8CAA8C;AAAA,QAC5D;AAEA,eAAO,MAAM,qBAAqB,QAAQ,OAAO,MAAM;AAAA,MACzD,SAAS,OAAO;AACd,eAAO,MAAM,gCAAgC,KAAK;AAElD,eAAO,CAAC;AAAA,MACV;AAAA,IACF;AAAA,IACA,CAAC,UAAU,YAAY,GAAG,OAAO,SAAS,WAAmC;AAC3E,UAAI;AACF,cAAM,SAAS,aAAa;AAAA,UAC1B,OAAO,QAAQ;AAAA,UACf,SAAS,QAAQ,WAAW,qBAAqB,KAAK;AAAA,QACxD,CAAC;AACD,cAAM,QACJ,QAAQ,WAAW,oBAAoB,KACvC,QAAQ,WAAW,aAAa,KAChC;AAEF,YAAI,OAAO,QAAQ;AACjB,iBAAO,KAAK,8CAA8C;AAAA,QAC5D;AAEA,eAAO,MAAM,qBAAqB,QAAQ,OAAO,MAAM;AAAA,MACzD,SAAS,OAAO;AACd,eAAO,MAAM,gCAAgC,KAAK;AAElD,eAAO,CAAC;AAAA,MACV;AAAA,IACF;AAAA,EACF;AAAA,EACA,OAAO;AAAA,IACL;AAAA,MACE,MAAM;AAAA,MACN,OAAO;AAAA,QACL;AAAA,UACE,MAAM;AAAA,UACN,IAAI,OAAO,YAAY;AACrB,gBAAI;AACF,oBAAM,UAAU,QAAQ,WAAW,qBAAqB,KAAK;AAC7D,oBAAM,WAAW,MAAM,MAAM,GAAG,OAAO,OAAO;AAC9C,oBAAM,OAAO,MAAM,SAAS,KAAK;AACjC,qBAAO,IAAI,qBAAsB,MAAgC,QAAQ,MAAM;AAC/E,kBAAI,CAAC,SAAS,IAAI;AAChB,uBAAO,MAAM,kCAAkC,SAAS,UAAU,EAAE;AACpE;AAAA,cACF;AAAA,YACF,SAAS,OAAO;AACd,qBAAO,MAAM,wCAAwC,KAAK;AAAA,YAC5D;AAAA,UACF;AAAA,QACF;AAAA,QACA;AAAA,UACE,MAAM;AAAA,UACN,IAAI,OAAO,YAAY;AACrB,gBAAI;AACF,oBAAM,YAAY,MAAM,QAAQ,SAAS,UAAU,gBAAgB;AAAA,gBACjE,MAAM;AAAA,cACR,CAAC;AACD,qBAAO,IAAI,aAAa,SAAS;AAAA,YACnC,SAAS,OAAO;AACd,qBAAO,MAAM,iCAAiC,KAAK;AAAA,YACrD;AAAA,UACF;AAAA,QACF;AAAA,QACA;AAAA,UACE,MAAM;AAAA,UACN,IAAI,OAAO,YAAY;AACrB,gBAAI;AACF,oBAAM,OAAO,MAAM,QAAQ,SAAS,UAAU,YAAY;AAAA,gBACxD,QAAQ;AAAA,cACV,CAAC;AACD,kBAAI,KAAK,WAAW,GAAG;AACrB,uBAAO,MAAM,yBAAyB;AACtC;AAAA,cACF;AACA,qBAAO,IAAI,mCAAmC,IAAI;AAAA,YACpD,SAAS,OAAO;AACd,qBAAO,MAAM,6BAA6B,KAAK;AAAA,YACjD;AAAA,UACF;AAAA,QACF;AAAA,QACA;AAAA,UACE,MAAM;AAAA,UACN,IAAI,OAAO,YAAY;AACrB,gBAAI;AACF,oBAAM,OAAO,MAAM,QAAQ,SAAS,UAAU,YAAY;AAAA,gBACxD,QAAQ;AAAA,cACV,CAAC;AACD,kBAAI,KAAK,WAAW,GAAG;AACrB,uBAAO,MAAM,yBAAyB;AACtC;AAAA,cACF;AACA,qBAAO,IAAI,mCAAmC,IAAI;AAAA,YACpD,SAAS,OAAO;AACd,qBAAO,MAAM,6BAA6B,KAAK;AAAA,YACjD;AAAA,UACF;AAAA,QACF;AAAA,QACA;AAAA,UACE,MAAM;AAAA,UACN,IAAI,OAAO,YAAY;AACrB,gBAAI;AACF,oBAAM,SAAS,MAAM,QAAQ,SAAS,UAAU,cAAc;AAAA,gBAC5D,QACE;AAAA,gBACF,aAAa;AAAA,cACf,CAAC;AACD,qBAAO,IAAI,qBAAqB,MAAM;AAAA,YACxC,SAAS,OAAO;AACd,qBAAO,MAAM,+BAA+B,KAAK;AAAA,YACnD;AAAA,UACF;AAAA,QACF;AAAA,QACA;AAAA,UACE,MAAM;AAAA,UACN,IAAI,OAAO,YAAY;AACrB,gBAAI;AACF,oBAAM,SAAS,MAAM,QAAQ,SAAS,UAAU,cAAc;AAAA,gBAC5D,QACE;AAAA,gBACF,aAAa;AAAA,cACf,CAAC;AACD,qBAAO,IAAI,qBAAqB,MAAM;AAAA,YACxC,SAAS,OAAO;AACd,qBAAO,MAAM,+BAA+B,KAAK;AAAA,YACnD;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;AACA,IAAO,gBAAQ;","names":[]}

package/package.json

@@ -0,0 +1,72 @@
+{
+  "name": "@elizaos/plugin-ollama",
+  "version": "1.0.0-beta.8",
+  "type": "module",
+  "main": "dist/index.js",
+  "module": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/elizaos-plugins/plugin-ollama"
+  },
+  "exports": {
+    "./package.json": "./package.json",
+    ".": {
+      "import": {
+        "types": "./dist/index.d.ts",
+        "default": "./dist/index.js"
+      }
+    }
+  },
+  "files": [
+    "dist"
+  ],
+  "dependencies": {
+    "@ai-sdk/ui-utils": "1.1.9",
+    "@elizaos/core": "1.0.0-beta.8",
+    "ai": "^4.1.25",
+    "js-tiktoken": "^1.0.18",
+    "ollama-ai-provider": "^1.2.0",
+    "tsup": "8.4.0"
+  },
+  "scripts": {
+    "build": "tsup",
+    "dev": "tsup --watch",
+    "lint": "prettier --write ./src",
+    "clean": "rm -rf dist .turbo node_modules .turbo-tsconfig.json tsconfig.tsbuildinfo",
+    "format": "prettier --write ./src",
+    "format:check": "prettier --check ./src"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "agentConfig": {
+    "pluginType": "elizaos:plugin:1.0.0",
+    "pluginParameters": {
+      "OLLAMA_SMALL_MODEL": {
+        "type": "string",
+        "description": "The name of the small Ollama model to use"
+      },
+      "OLLAMA_LARGE_MODEL": {
+        "type": "string",
+        "description": "The name of the large Ollama model to use"
+      },
+      "OLLAMA_MEDIUM_MODEL": {
+        "type": "string",
+        "description": "The name of the medium Ollama model to use"
+      },
+      "OLLAMA_EMBEDDING_MODEL": {
+        "type": "string",
+        "description": "The name of the embedding Ollama model to use"
+      },
+      "OLLAMA_API_ENDPOINT": {
+        "type": "string",
+        "description": "The URL of the Ollama API"
+      }
+    }
+  },
+  "gitHead": "a84e25ee21e33387a1303e62a2f95dc670ffddd4",
+  "devDependencies": {
+    "prettier": "3.5.3"
+  }
+}