fix entry

example/lib/main.dart

@@ -1,5 +1,4 @@
 import 'dart:io';
-import 'dart:math';
 
 import 'package:flutter/material.dart';
 
@@ -42,7 +41,7 @@ class _LlamaAppState extends State<LlamaApp> {
       throw Exception('File does not exist');
     }
 
-    final llamaCpp = LlamaNative(
+    final llamaCpp = LlamaIsolated(
       modelParams: ModelParams(
         path: result.files.single.path!
       ),

lib/src/context_params.dart

@@ -104,9 +104,9 @@ class ContextParams {
     nSeqMax: map['nSeqMax'],
     nThreads: map['nThreads'],
     nThreadsBatch: map['nThreadsBatch'],
-    ropeScalingType: RopeScalingType.fromString(map['ropeScalingType']),
-    poolingType: PoolingType.fromString(map['poolingType']),
-    attentionType: AttentionType.fromString(map['attentionType']),
+    ropeScalingType: map['ropeScalingType'] != null ? RopeScalingType.fromString(map['ropeScalingType']) : null,
+    poolingType: map['poolingType'] != null ? PoolingType.fromString(map['poolingType']) : null,
+    attentionType: map['attentionType'] != null ? AttentionType.fromString(map['attentionType']) : null,
     ropeFrequencyBase: map['ropeFrequencyBase'],
     ropeFrequencyScale: map['ropeFrequencyScale'],
     yarnExtrapolationFactor: map['yarnExtrapolationFactor'],
@@ -115,8 +115,8 @@ class ContextParams {
     yarnBetaSlow: map['yarnBetaSlow'],
     yarnOriginalContext: map['yarnOriginalContext'],
     defragmentationThreshold: map['defragmentationThreshold'],
-    typeK: GgmlType.fromString(map['typeK']),
-    typeV: GgmlType.fromString(map['typeV']),
+    typeK: map['typeK'] != null ? GgmlType.fromString(map['typeK']) : null,
+    typeV: map['typeV'] != null ? GgmlType.fromString(map['typeV']) : null,
     embeddings: map['embeddings'],
     offloadKqv: map['offloadKqv'],
     flashAttention: map['flashAttention'],

lib/src/llama_isolate_entry.dart

@@ -66,7 +66,7 @@ class LlamaIsolateEntry {
       }
     }
     catch (e) {
-      sendPort.send((message: e.toString()));
+      log('LlamaIsolateEntry: $e');
     }
   }
 

lib/src/llama_isolated.dart

@@ -44,6 +44,7 @@ class LlamaIsolated implements Llama {
         _initialized.complete();
       }
       else if (data == null) {
+        print('LlamaIsolated: stopping');
         _responseController.close();
         break;
       }
@@ -62,13 +63,10 @@ class LlamaIsolated implements Llama {
 
     messages.add(ChatMessage(
       role: 'assistant',
-      content: ''
+      content: await _responseController.stream.first
     ));
 
-    await for (final message in _responseController.stream) {
-      messages.last.content += message;
-      _responseController.close();
-    }
+    _responseController.close();
 
     return messages.last.content;
   }