Bugfixes for Wikipedia extract and search commands (#242)

monarch-initiative · Oct 18, 2023 · 3f1202a · 3f1202a
2 parents 40f2182 + 7c4d424
commit 3f1202a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 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]

diff --git a/src/ontogpt/clients/wikipedia_client.py b/src/ontogpt/clients/wikipedia_client.py
@@ -13,7 +13,7 @@ class WikipediaClient:
 
     def text(self, title: str) -> str:
         """Get the text of a an article."""
-        wiki = wikipediaapi.Wikipedia(self.language)
+        wiki = wikipediaapi.Wikipedia("ontogpt", self.language)
         page = wiki.page(title)
 
         if page.exists():