feat(agents): Add ollama:pull script

Author: grypez

packages/agents/package.json

@@ -38,6 +38,7 @@
     "lint:eslint": "eslint . --cache",
     "lint:fix": "yarn lint:eslint --fix && yarn lint:misc --write && yarn constraints --fix && yarn lint:dependencies",
     "lint:misc": "prettier --no-error-on-unmatched-pattern '**/*.json' '**/*.md' '**/*.html' '!**/CHANGELOG.old.md' '**/*.yml' '!.yarnrc.yml' '!merged-packages/**' --ignore-path ../../.gitignore",
+    "ollama:pull": "node scripts/ollama-pull.js",
     "publish:preview": "yarn npm publish --tag preview",
     "test": "vitest run --config vitest.config.ts",
     "test:clean": "yarn test --no-cache --coverage.clean",

packages/agents/scripts/ollama-pull.js

@@ -0,0 +1,38 @@
+import { Ollama } from 'ollama';
+
+main().catch(console.error);
+
+/**
+ * The main function for the script.
+ */
+async function main() {
+  // Make a new ollama client.
+  const ollama = new Ollama({ host: 'http://localhost:11434' });
+
+  let nDots = 0;
+  const interval = setInterval(() => (nDots = (nDots + 1) % 4), 250);
+  try {
+    const model = process.argv[2];
+    const labelDuring = `Pulling ${model}`;
+    const labelAfter = `Pulled ${model}`;
+    const makeProgress = (fraction) =>
+      [
+        labelDuring,
+        `${'.'.repeat(nDots)}${' '.repeat(4 - nDots)}`,
+        '[',
+        `${'='.repeat(Math.floor(fraction * 10))}`,
+        `${' '.repeat(10 - Math.floor(fraction * 10))}`,
+        ']',
+        ` ${Math.floor(fraction * 100)}%`,
+      ].join('');
+    const result = await ollama.pull({ model, stream: true });
+    for await (const chunk of result) {
+      process.stdout.write(`${makeProgress(chunk.completed / chunk.total)}\r`);
+    }
+    console.log(
+      `${labelAfter} ${' '.repeat(15 + labelDuring.length - labelAfter.length)}`,
+    );
+  } finally {
+    clearInterval(interval);
+  }
+}