import type { DelegateParams, DelegateResult } from './types';

const BASE = process.env.ZPM_OLLAMA_URL || 'http://REDACTED-IP:11434';

const DEFAULT_MODEL = process.env.ZPM_OLLAMA_DEFAULT || 'mistral-nemo:12b';

export async function ollamaAdapter(p: DelegateParams): Promise<DelegateResult> {

  const start = Date.now();

  const model = p.ollamaModel || DEFAULT_MODEL;

  const messages: Array<{ role: string; content: string }> = [];

  const sys = p.context ? `You are a helpful worker. Relevant context:\n${p.context}` : 'You are a helpful worker.';

  messages.push({ role: 'system', content: sys });

  messages.push({ role: 'user', content: p.task });

  const ctrl = new AbortController();

  const timer = setTimeout(() => ctrl.abort(), p.timeoutMs ?? 120_000);

  try {

    const res = await fetch(`${BASE}/api/chat`, {

      method: 'POST',

      headers: { 'content-type': 'application/json' },

      body: JSON.stringify({ model, messages, stream: false, options: { think: false } }),

      signal: ctrl.signal,

    });

    if (!res.ok) {

      const t = await res.text();

      return { ok: false, output: '', artifacts: [], error: `ollama ${res.status}: ${t.slice(0, 300)}`, durationMs: Date.now() - start };

    }

    const j = (await res.json()) as { message?: { content?: string } };

    const out = j.message?.content?.trim() ?? '';

    return { ok: out.length > 0, output: out, artifacts: [], error: out ? undefined : 'empty response', durationMs: Date.now() - start };

  } catch (e) {

    return { ok: false, output: '', artifacts: [], error: (e as Error).message, durationMs: Date.now() - start };

  } finally {

    clearTimeout(timer);

  }

}