mirror/OpenLLM
1
0
Fork 0
mirror of https://github.com/bentoml/OpenLLM.git synced 2026-03-11 19:47:03 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix: agent log (#37)

Browse Source
This commit is contained in:
Aaron Pham
2023-06-19 14:11:39 -04:00
committed by GitHub
parent 78a537079e
commit ca802d9d1a
2 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
changelog.d/37.change.md Normal file
View File

@@ -0,0 +1 @@
Fixes setting logs for agents to info instead of logger object.

2
src/openllm/cli.py
View File

@@ -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")