openaxies 0.5.2 → 0.5.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "openaxies",
-  "version": "0.5.2",
+  "version": "0.5.4",
   "private": false,
   "type": "module",
   "bin": {

package/src/App.js

@@ -180,23 +180,21 @@ function AppRoot() {
         }
         if (event.type === 'thinking' || event.type === 'token') {
           const content = typeof event.content === 'string' ? event.content : '';
-          for (let i = 0; i < content.length; i++) {
-            accumulated = accumulated + content[i];
-            const inside = checkThink(accumulated);
-            setStreamText(accumulated);
-            setIsThinking(inside);
-            if (inside === true && thinkStarted === false) {
-              thinkStarted = true;
-              thinkStartRef.current = Date.now();
-              setThinkingElapsed(0);
-            }
-            if (inside === false && thinkStarted === true) {
-              thinkStarted = false;
-              thinkStartRef.current = null;
-              setThinkingElapsed(0);
-            }
-            await new Promise(function (r) { setTimeout(r, 0); });
+          accumulated = accumulated + content;
+          const inside = checkThink(accumulated);
+          setStreamText(accumulated);
+          setIsThinking(inside);
+          if (inside === true && thinkStarted === false) {
+            thinkStarted = true;
+            thinkStartRef.current = Date.now();
+            setThinkingElapsed(0);
+          }
+          if (inside === false && thinkStarted === true) {
+            thinkStarted = false;
+            thinkStartRef.current = null;
+            setThinkingElapsed(0);
           }
+          await new Promise(function (r) { setTimeout(r, 0); });
         }
         if (event.type === 'done') break;
         if (event.type === 'timeout') {
@@ -210,10 +208,9 @@ function AppRoot() {
       thinkStartRef.current = null;
       setThinkingElapsed(0);
       setStreamingActive(false);
-      const finalText = streamText;
       setStreamText('');
-      if (finalText.length > 0) {
-        setMessages(function (p) { return p.concat([{ role: 'assistant', content: finalText, id: 'a-' + Date.now() }]); });
+      if (accumulated.length > 0) {
+        setMessages(function (p) { return p.concat([{ role: 'assistant', content: accumulated, id: 'a-' + Date.now() }]); });
       }
     } catch (err) {
       if (connRef.current !== null) clearInterval(connRef.current), connRef.current = null;
@@ -279,7 +276,7 @@ function AppRoot() {
       h(Text, { key: 'l' + r, color: LOGO_COLORS[r % LOGO_COLORS.length], bold: true }, LOGO[r])
     );
   }
-  const logo = h(Box, { flexDirection: 'column', width: '100%', flexShrink: 0, paddingLeft: 1 }, ...logoEls);
+  const logo = h(Box, { flexDirection: 'column', width: '100%', flexShrink: 0, paddingLeft: 0 }, ...logoEls);
 
   const viewLines = [];
   for (let i = 0; i < messages.length; i++) {
@@ -310,7 +307,10 @@ function AppRoot() {
   }
 
   if (viewLines.length === 0) {
-    viewLines.push({ t: 'idle', text: '  \u25B8 type a message to begin...     [/help for commands]' });
+    viewLines.push({ t: 'idle', text: '  \u25B8 type a message to begin   |   /model to switch models   |   /help for commands' });
+    viewLines.push({ t: 'idle', text: '' });
+    viewLines.push({ t: 'idle-sm', text: '  Models: OpenAxies Llama  |  OpenAxies GPT  |  OpenAxies DeepSeek' });
+    viewLines.push({ t: 'idle-sm', text: '  Current: ' + activeLabel });
   }
 
   const visible = Math.max(1, availLines - 1);
@@ -352,7 +352,13 @@ function AppRoot() {
     } else if (l.t === 'idle') {
       viewEls.push(
         h(Box, { key: 'i-' + i, height: 1, paddingLeft: 1 },
-          h(Text, { color: '#444466' }, l.text)
+          h(Text, { color: '#555577' }, l.text)
+        )
+      );
+    } else if (l.t === 'idle-sm') {
+      viewEls.push(
+        h(Box, { key: 'is-' + i, height: 1, paddingLeft: 1 },
+          h(Text, { color: '#333355' }, l.text)
         )
       );
     }

package/src/providers/index.js

@@ -82,6 +82,7 @@ export async function* callModel(modelConfig, messages, signal) {
     messages: modelMessages,
     max_tokens: 4096,
     temperature: 0.7,
+    stream: true,
   };
 
   let lastError = null;

package/src/providers/streaming.js

@@ -78,7 +78,7 @@ export async function* streamResponse(endpoint, body, signal) {
     }, READ_TIMEOUT);
   }
 
-  resetReadTimeout();
+  let firstChunk = true;
 
   try {
     while (true) {
@@ -86,12 +86,48 @@ export async function* streamResponse(endpoint, body, signal) {
       resetReadTimeout();
 
       if (result.done === true) {
+        if (buffer.trim().length > 0 && firstChunk === false) {
+          try {
+            const parsed = JSON.parse(buffer);
+            const choice = parsed.choices && parsed.choices[0];
+            if (choice !== null && choice !== undefined) {
+              const msg = choice.message || choice.delta;
+              if (msg !== null && msg !== undefined) {
+                if (typeof msg.content === 'string' && msg.content.length > 0) {
+                  yield { type: 'token', content: msg.content };
+                }
+              }
+            }
+          } catch (_) {
+          }
+        }
         yield { type: 'done' };
         return;
       }
 
       const chunk = decoder.decode(result.value, { stream: true });
       buffer = buffer + chunk;
+
+      if (firstChunk === true && buffer.includes('data: ') === false && buffer.trim().startsWith('{') === true) {
+        try {
+          const parsed = JSON.parse(buffer);
+          const choice = parsed.choices && parsed.choices[0];
+          if (choice !== null && choice !== undefined) {
+            const msg = choice.message || choice.delta;
+            if (msg !== null && msg !== undefined) {
+              if (typeof msg.content === 'string' && msg.content.length > 0) {
+                yield { type: 'token', content: msg.content };
+              }
+            }
+          }
+          yield { type: 'done' };
+          return;
+        } catch (_) {
+        }
+      }
+
+      firstChunk = false;
+
       const lines = buffer.split('\n');
       buffer = lines.pop() || '';