From b9b6cd2f219fab663ae7d1b6a48eb64902c4fec1 Mon Sep 17 00:00:00 2001
From: Jhen <developer@jhen.me>
Date: Wed, 2 Aug 2023 15:53:10 +0800
Subject: [PATCH] server : fix completion_probabilities undefined if not set
 n_probs

---
 examples/server/public/index.html | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/examples/server/public/index.html b/examples/server/public/index.html
index c6d674e16..ff17c8a18 100644
--- a/examples/server/public/index.html
+++ b/examples/server/public/index.html
@@ -137,7 +137,7 @@
 
   <script type="module">
     import {
-      html, h, signal, effect, computed, render, useSignal, useEffect, useRef
+      html, h, signal, effect, computed, render, useSignal, useEffect, useRef, Component
     } from '/index.js';
 
     import { llama } from '/completion.js';
@@ -288,7 +288,13 @@
       const chatLine = ([user, data], index) => {
         let message
         const isArrayMessage = Array.isArray(data)
-        if (isArrayMessage && data.every(msg => msg.completion_probabilities.length > 0)) {
+        if (
+          isArrayMessage &&
+          data.every(msg =>
+            msg.completion_probabilities &&
+            msg.completion_probabilities.length > 0
+          )
+        ) {
           message = html`<${Probabilites} data=${data} />`
         } else {
           const text = isArrayMessage ?