update readme & chatllm: support dynamic loading

Author: Judd

README.md

@@ -11,12 +11,20 @@ goal to [make](https://github.com/google-deepmind/alphageometry/issues/130)
 
 * Indent with **four** spaces (🖕 two spaces).
 
-Plan:
+Plan/Roadmap:
 
 * [x] LM Beam search.
 
+* [ ] Rewrite in Nim.
+
 * [ ] A new description language like [this](https://reference.wolfram.com/legacy/language/v14/guide/PlaneGeometry.html).
 
+* [ ] Catch up with AlphaGeometry2.
+
+## Documentation
+
+
+
 --------
 
 # Solving Olympiad Geometry without Human Demonstrations
@@ -75,7 +83,7 @@ A list of command line flags.
 
 * `--problem_name`: name of a problem in the given problem file
 
-    Example: `--problems_file orthocenter`. `orthocenter` is a problem in `examples\examples.txt`.
+    Example: `--problem_name orthocenter`. `orthocenter` is a problem in `examples\examples.txt`.
 
 * `--defs_file` & `--rules_file`: definitions & deduction rules.
 

src/chatllm/bindings/chatllm.py

@@ -36,13 +36,14 @@ class LibChatLLM:
     _obj2id = {}
     _id2obj = {}
 
-    def __init__(self, lib: str = '', model_storage: str = '') -> None:
+    def __init__(self, lib: str = '', model_storage: str = '', init_params: list[str] = []) -> None:
 
         if lib == '':
             lib = os.path.dirname(os.path.abspath(sys.argv[0]))
         self._lib_path = lib
         self.model_storage = os.path.abspath(model_storage if model_storage != '' else os.path.join(lib, '..', 'quantized'))
 
+        init_params = ['--ggml_dir', lib] + init_params
         lib = os.path.join(lib, 'libchatllm.')
         if sys.platform == 'win32':
             lib = lib + 'dll'
@@ -60,6 +61,17 @@ def __init__(self, lib: str = '', model_storage: str = '') -> None:
             self._PRINTFUNC = CFUNCTYPE(None, c_void_p, c_int, c_char_p)
             self._ENDFUNC = CFUNCTYPE(None, c_void_p)
 
+        _chatllm_append_init_param = self._lib.chatllm_append_init_param
+        _chatllm_append_init_param.restype = None
+        _chatllm_append_init_param.argtypes = [c_char_p]
+        _chatllm_init = self._lib.chatllm_init
+        _chatllm_init.restype = c_int
+        _chatllm_init.argtypes = []
+
+        for s in init_params:
+            _chatllm_append_init_param(c_char_p(s.encode()))
+        assert _chatllm_init() == 0
+
         self._chatllm_create            = self._lib.chatllm_create
         self._chatllm_append_param      = self._lib.chatllm_append_param
         self._chatllm_start             = self._lib.chatllm_start
@@ -170,8 +182,12 @@ def callback_print(user_data: int, print_type: c_int, s: bytes) -> None:
             obj.callback_text_tokenize(txt)
         elif print_type == PrintType.PRINTLN_ERROR.value:
             raise Exception(txt)
+        elif print_type == PrintType.PRINTLN_LOGGING.value:
+            obj.callback_print_log(txt)
         elif print_type == PrintType.PRINTLN_BEAM_SEARCH.value:
             obj.callback_print_beam_search(txt)
+        elif print_type == PrintType.PRINT_EVT_ASYNC_COMPLETED.value:
+            obj.callback_async_done()
         else:
             raise Exception(f"unhandled print_type({print_type}): {txt}")
 
@@ -372,6 +388,8 @@ def load_session(self, file_name: str) -> str:
 
     def callback_print_reference(self, s: str) -> None:
         self.references.append(s)
+    def callback_print_log(self, s: str) -> None:
+        print(s)
 
     def callback_print_beam_search(self, s: str) -> None:
         l = s.split(',', maxsplit=1)