diff --git a/changelog.d/37.change.md b/changelog.d/37.change.md new file mode 100644 index 00000000..ad216a31 --- /dev/null +++ b/changelog.d/37.change.md @@ -0,0 +1 @@ +Fixes setting logs for agents to info instead of logger object. diff --git a/src/openllm/cli.py b/src/openllm/cli.py index 3acf388b..f855b596 100644 --- a/src/openllm/cli.py +++ b/src/openllm/cli.py @@ -1153,7 +1153,7 @@ def instruct( _memoized = {k: v[0] for k, v in _memoized.items() if v} - client._hf_agent.set_stream(logger) + client._hf_agent.set_stream(logger.info) if output != "porcelain": _echo(f"Sending the following prompt ('{task}') with the following vars: {_memoized}", fg="magenta")