From ca802d9d1acaa53da681a86c3459a90fe10a4a7f Mon Sep 17 00:00:00 2001
From: Aaron Pham <29749331+aarnphm@users.noreply.github.com>
Date: Mon, 19 Jun 2023 14:11:39 -0400
Subject: [PATCH] fix: agent log (#37)

---
 changelog.d/37.change.md | 1 +
 src/openllm/cli.py       | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)
 create mode 100644 changelog.d/37.change.md

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