Allow specification of model for wiki extract + search

monarch-initiative · Oct 18, 2023 · 7c4d424 · 7c4d424
1 parent 1912818
commit 7c4d424
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/ontogpt/cli.py b/src/ontogpt/cli.py
@@ -562,7 +562,7 @@ def wikipedia_extract(model, article, template, output, output_format, show_prom
     model_source = selectmodel["provider"]
 
     if model_source == "OpenAI":
-        ke = SPIRESEngine(template, **kwargs)
+        ke = SPIRESEngine(template=template, model=model, **kwargs)
         if settings.cache_db:
             ke.client.cache_db_path = settings.cache_db
         if settings.skip_annotators:
@@ -603,7 +603,7 @@ def wikipedia_search(model, topic, keyword, template, output, output_format, sho
     model_source = selectmodel["provider"]
 
     if model_source == "OpenAI":
-        ke = SPIRESEngine(template, **kwargs)
+        ke = SPIRESEngine(template=template, model=model, **kwargs)
 
     elif model_source == "GPT4All":
         model_name = selectmodel["alternative_names"][0]