codebot-ai 1.0.0 → 1.0.1

package/dist/providers/openai.js

@@ -52,6 +52,9 @@ class OpenAIProvider {
         const decoder = new TextDecoder();
         let buffer = '';
         const toolCalls = new Map();
+        // Track <think>...</think> blocks (used by qwen3, deepseek, etc.)
+        let insideThink = false;
+        let contentBuffer = '';
         try {
             while (true) {
                 const { done, value } = await reader.read();
@@ -65,6 +68,11 @@ class OpenAIProvider {
                     if (!trimmed || !trimmed.startsWith('data: '))
                         continue;
                     if (trimmed === 'data: [DONE]') {
+                        // Flush any remaining content buffer
+                        if (contentBuffer && !insideThink) {
+                            yield { type: 'text', text: contentBuffer };
+                            contentBuffer = '';
+                        }
                         for (const [, tc] of toolCalls) {
                             yield {
                                 type: 'tool_call_end',
@@ -84,7 +92,48 @@ class OpenAIProvider {
                         if (!delta)
                             continue;
                         if (delta.content) {
-                            yield { type: 'text', text: delta.content };
+                            contentBuffer += delta.content;
+                            // Process buffer for <think>...</think> tags
+                            let changed = true;
+                            while (changed) {
+                                changed = false;
+                                if (insideThink) {
+                                    const end = contentBuffer.indexOf('</think>');
+                                    if (end !== -1) {
+                                        // End of think block — discard thinking content, continue
+                                        contentBuffer = contentBuffer.slice(end + 8);
+                                        insideThink = false;
+                                        changed = true;
+                                    }
+                                    // else: still inside think, wait for more data
+                                }
+                                else {
+                                    const start = contentBuffer.indexOf('<think>');
+                                    if (start !== -1) {
+                                        // Found <think> — output everything before it, enter think mode
+                                        const before = contentBuffer.slice(0, start);
+                                        if (before)
+                                            yield { type: 'text', text: before };
+                                        contentBuffer = contentBuffer.slice(start + 7);
+                                        insideThink = true;
+                                        changed = true;
+                                    }
+                                    else {
+                                        // No think tag — check if buffer ends with a partial "<think" prefix
+                                        let holdBack = 0;
+                                        for (let len = Math.min(6, contentBuffer.length); len >= 1; len--) {
+                                            if ('<think>'.startsWith(contentBuffer.slice(-len))) {
+                                                holdBack = len;
+                                                break;
+                                            }
+                                        }
+                                        const safe = contentBuffer.slice(0, contentBuffer.length - holdBack);
+                                        contentBuffer = contentBuffer.slice(contentBuffer.length - holdBack);
+                                        if (safe)
+                                            yield { type: 'text', text: safe };
+                                    }
+                                }
+                            }
                         }
                         if (delta.tool_calls) {
                             for (const tc of delta.tool_calls) {

package/dist/setup.js

@@ -194,16 +194,16 @@ async function runSetup() {
     const choice = await ask(rl, fmt(`\nSelect [1-${options.length}]: `, 'cyan'));
     const selected = options[parseInt(choice, 10) - 1] || options[0];
     // Step 4: Show available models for chosen provider
-    const providerModels = Object.entries(registry_1.MODEL_REGISTRY)
-        .filter(([, info]) => {
-        if (selected.provider === 'openai' && !info.provider)
-            return true; // local models
-        return info.provider === selected.provider;
-    })
-        .map(([name]) => name);
+    // For local servers, use the actual installed models instead of the hardcoded registry
+    const matchedServer = localServers.find(s => s.url === selected.baseUrl);
+    const providerModels = matchedServer && matchedServer.models.length > 0
+        ? matchedServer.models
+        : Object.entries(registry_1.MODEL_REGISTRY)
+            .filter(([, info]) => info.provider === selected.provider)
+            .map(([name]) => name);
     if (providerModels.length > 1) {
-        console.log(fmt(`\nAvailable models for ${selected.label}:`, 'bold'));
-        providerModels.slice(0, 10).forEach((m, i) => {
+        console.log(fmt(`\nAvailable models${matchedServer ? ` on ${matchedServer.name}` : ''}:`, 'bold'));
+        providerModels.slice(0, 15).forEach((m, i) => {
             const marker = m === selected.model ? fmt(' (default)', 'green') : '';
             console.log(`  ${fmt(`${i + 1}`, 'cyan')} ${m}${marker}`);
         });
@@ -214,7 +214,7 @@ async function runSetup() {
                 selected.model = providerModels[modelIdx];
             }
             else if (modelChoice.length > 2) {
-                // Treat as model name
+                // Treat as model name typed directly
                 selected.model = modelChoice;
             }
         }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "codebot-ai",
-  "version": "1.0.0",
+  "version": "1.0.1",
   "description": "Local-first AI coding assistant. Zero dependencies. Works with Ollama, LM Studio, vLLM, Claude, GPT, Gemini, and more.",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",