diff --git a/example/lib/main.dart b/example/lib/main.dart index 87e7983..170cb5f 100644 --- a/example/lib/main.dart +++ b/example/lib/main.dart @@ -4,7 +4,6 @@ import 'package:flutter/material.dart'; import 'dart:async'; import 'package:file_picker/file_picker.dart'; -import 'package:llama/bindings.dart'; import 'package:llama/llama.dart'; void main() { @@ -59,8 +58,6 @@ class _LlamaAppState extends State { ) ); - print('LlamaCPP created'); - Stream stream = llamaCpp.prompt(_messages); setState(() { diff --git a/lib/src/llama_cpp_native.dart b/lib/src/llama_cpp_native.dart index 4e08595..b3a8831 100644 --- a/lib/src/llama_cpp_native.dart +++ b/lib/src/llama_cpp_native.dart @@ -63,6 +63,10 @@ class LlamaCppNative { } Stream prompt(List messages) async* { + assert(_model != ffi.nullptr, 'Model is not loaded'); + assert(_context != ffi.nullptr, 'Context is not initialized'); + assert(_sampler != ffi.nullptr, 'Sampler is not initialized'); + _completer = Completer(); final nCtx = lib.llama_n_ctx(_context);