@orchagent/cli 0.2.14 → 0.2.15

package/dist/lib/llm-errors.js

@@ -28,6 +28,12 @@ const FALLBACKS = {
         502: 'Gemini is temporarily unavailable.',
         503: 'Gemini is overloaded. Try again later.',
     },
+    ollama: {
+        401: 'Authentication error (Ollama typically does not require auth)',
+        404: 'Model not found. Run: ollama pull <model>',
+        500: 'Ollama server error',
+        502: 'Cannot connect to Ollama. Is it running?',
+    },
 };
 const DEFAULT = 'LLM provider error. Check your API key and try again.';
 function isHtml(text) {
@@ -67,13 +73,25 @@ function parseGemini(text, status) {
     catch { }
     return FALLBACKS.gemini[status] || DEFAULT;
 }
+function parseOllama(text, status) {
+    // Ollama uses OpenAI-compatible error format
+    try {
+        const p = JSON.parse(text);
+        const msg = p.error?.message || (typeof p.error === 'string' ? p.error : null);
+        if (msg)
+            return sanitize(msg);
+    }
+    catch { }
+    return FALLBACKS.ollama[status] || DEFAULT;
+}
 function parseLlmError(provider, text, status) {
     if (isHtml(text)) {
         return new errors_1.CliError(`${provider} error: ${FALLBACKS[provider][status] || DEFAULT}`);
     }
     const msg = provider === 'openai' ? parseOpenAI(text, status)
         : provider === 'anthropic' ? parseAnthropic(text, status)
-            : parseGemini(text, status);
+            : provider === 'ollama' ? parseOllama(text, status)
+                : parseGemini(text, status);
     const display = provider.charAt(0).toUpperCase() + provider.slice(1);
     return new errors_1.CliError(`${display} API error: ${msg}`);
 }

package/dist/lib/llm.js

@@ -53,12 +53,14 @@ exports.PROVIDER_ENV_VARS = {
     openai: 'OPENAI_API_KEY',
     anthropic: 'ANTHROPIC_API_KEY',
     gemini: 'GEMINI_API_KEY',
+    ollama: 'OLLAMA_HOST',
 };
 // Default models for each provider
 exports.DEFAULT_MODELS = {
     openai: 'gpt-4o',
     anthropic: 'claude-sonnet-4-20250514',
     gemini: 'gemini-1.5-pro',
+    ollama: 'llama3.2',
 };
 /**
  * Detect LLM API key from environment variables based on supported providers.
@@ -162,6 +164,9 @@ async function callLlm(provider, apiKey, model, prompt, outputSchema) {
     else if (provider === 'gemini') {
         return callGemini(apiKey, model, prompt, outputSchema);
     }
+    else if (provider === 'ollama') {
+        return callOllama(apiKey, model, prompt, outputSchema);
+    }
     throw new errors_1.CliError(`Unsupported provider: ${provider}`);
 }
 async function callOpenAI(apiKey, model, prompt, outputSchema) {
@@ -242,11 +247,42 @@ async function callGemini(apiKey, model, prompt, _outputSchema) {
         return { result: content };
     }
 }
+async function callOllama(endpoint, model, prompt, _outputSchema) {
+    // endpoint is passed via apiKey param for Ollama (it's the OLLAMA_HOST)
+    const baseUrl = endpoint || 'http://localhost:11434';
+    // Ensure /v1 path
+    const normalizedBase = baseUrl.endsWith('/v1') ? baseUrl : `${baseUrl.replace(/\/$/, '')}/v1`;
+    const url = `${normalizedBase}/chat/completions`;
+    const response = await fetch(url, {
+        method: 'POST',
+        headers: { 'Content-Type': 'application/json' },
+        body: JSON.stringify({
+            model,
+            messages: [{ role: 'user', content: prompt }],
+            options: { num_ctx: 8192 },
+        }),
+    });
+    if (!response.ok) {
+        if (response.status === 404) {
+            throw new errors_1.CliError(`Model '${model}' not found. Run: ollama pull ${model}`);
+        }
+        const text = await response.text();
+        throw (0, llm_errors_1.parseLlmError)('ollama', text, response.status);
+    }
+    const data = await response.json();
+    const content = data.choices?.[0]?.message?.content || '';
+    try {
+        return JSON.parse(content);
+    }
+    catch {
+        return { result: content };
+    }
+}
 /**
  * Validate a provider string against known providers.
  */
 function validateProvider(provider) {
-    const validProviders = ['openai', 'anthropic', 'gemini'];
+    const validProviders = ['openai', 'anthropic', 'gemini', 'ollama'];
     if (!validProviders.includes(provider)) {
         throw new errors_1.CliError(`Invalid provider: ${provider}. Valid: ${validProviders.join(', ')}`);
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@orchagent/cli",
-  "version": "0.2.14",
+  "version": "0.2.15",
   "description": "Command-line interface for the OrchAgent AI agent marketplace",
   "license": "MIT",
   "author": "OrchAgent <hello@orchagent.io>",