diff --git a/bin/add_help_to_readme.py b/bin/add_help_to_readme.py index ead0a38..c9c8c44 100755 --- a/bin/add_help_to_readme.py +++ b/bin/add_help_to_readme.py @@ -10,7 +10,7 @@ quote = '```' f = io.StringIO() -with (redirect_stdout(f), suppress(SystemExit)): +with redirect_stdout(f), suppress(SystemExit): cli.main(['-h']) help_text = f.getvalue().replace('add_help_to_readme.py', 'charla') diff --git a/charla/client/ollama.py b/charla/client/ollama.py index aba7381..1963ab2 100644 --- a/charla/client/ollama.py +++ b/charla/client/ollama.py @@ -37,7 +37,8 @@ def generate(self, prompt: str): self.add_message(role='user', text=prompt) response = self.client.generate( - model=self.model, prompt=prompt, context=self.context, stream=True, system=self.system) + model=self.model, prompt=prompt, context=self.context, stream=True, system=self.system + ) # Make sure system message is set only once. self.system = '' diff --git a/tests/test_cli.py b/tests/test_cli.py index 64ffa66..4789147 100644 --- a/tests/test_cli.py +++ b/tests/test_cli.py @@ -8,4 +8,3 @@ def test_cli_help(capsys): main(['-h']) output = capsys.readouterr().out assert 'language model' in output -