diff --git a/src/ontogpt/cli.py b/src/ontogpt/cli.py index d1003f3c7..f723c917e 100644 --- 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 index 1e39f70d0..565e2702e 100644 --- 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():