initial pipeline test passed

src/providers/ollama_provider.mjs

@@ -1,37 +1,58 @@
 // src/providers/ollama_provider.mjs
-import { BaseProvider } from "./base.mjs";
+import { BaseProvider } from './base.mjs';
+
+function normalizeBase(url) {
+  // strip trailing slashes so we can safely append /api/generate
+  return url.replace(/\/+$/, '');
+}
 
 export class OllamaProvider extends BaseProvider {
-  constructor(stage = "generator") {
+  /**
+   * @param {object} opts
+   * @param {string} [opts.model]   - model name/tag in Ollama
+   * @param {string} [opts.baseUrl] - base Ollama URL (without /api/generate)
+   */
+  constructor(opts = {}) {
     super();
-    this.url = process.env.OLLAMA_URL || "http://localhost:11434/api/generate";
 
-    // STAGE-SELECTED MODEL
+    // Base URL: env or default, WITHOUT endpoint path
+    const envBase = process.env.OLLAMA_URL || 'http://localhost:11434';
+    this.baseUrl = normalizeBase(opts.baseUrl || envBase);
+
+    // Model: allow stage-specific env, then generic, then default
     this.model =
-      process.env[`${stage.toUpperCase()}_MODEL`] ||
-      process.env.MODEL ||
-      "qwen3-8b-instruct";
+      opts.model ||
+      process.env.GENERATOR_MODEL ||
+      process.env.OLLAMA_MODEL ||
+      'qwen3-vl:8b-thinking';
   }
 
+  /**
+   * Call Ollama /api/generate
+   * @param {string} prompt
+   * @returns {Promise<string>} the model's response text
+   */
   async generate(prompt) {
-    const body = {
-      model: this.model,
-      prompt,
-      stream: false,
-      options: {
-        temperature: Number(process.env.TEMP || "0.3")
-      }
-    };
-
-    const res = await fetch(this.url, {
-      method: "POST",
-      headers: { "Content-Type": "application/json" },
-      body: JSON.stringify(body)
+    const url = `${this.baseUrl}/api/generate`;
+
+    const res = await fetch(url, {
+      method: 'POST',
+      headers: { 'Content-Type': 'application/json' },
+      body: JSON.stringify({
+        model: this.model,
+        prompt,
+        stream: false,
+      }),
     });
 
-    if (!res.ok) throw new Error(`OllamaProvider error: ${res.status}`);
+    if (!res.ok) {
+      throw new Error(`OllamaProvider error: ${res.status}`);
+    }
 
-    const json = await res.json();
-    return json.response;
+    const data = await res.json();
+    // Standard Ollama /api/generate response uses `response`
+    return data.response ?? '';
   }
 }
+
+export default OllamaProvider;