github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-07 22:11:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

beta: logger (#307)

Browse Source
This commit is contained in:
Harrison Chase
2022-12-10 23:17:19 -08:00
committed by GitHub
parent 36b4c58acf
commit e02d6b2288
10 changed files with 186 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
tests/unit_tests/test_input.py
View File

@@ -3,7 +3,8 @@
import sys
from io import StringIO
from langchain.input import ChainedInput, get_color_mapping
from langchain.agents.input import ChainedInput
from langchain.input import get_color_mapping
def test_chained_input_not_verbose() -> None:
@@ -18,11 +19,11 @@ def test_chained_input_not_verbose() -> None:
old_stdout = sys.stdout
sys.stdout = mystdout = StringIO()
chained_input.add("bar")
chained_input.add_observation("bar", "1", "2", None)
sys.stdout = old_stdout
output = mystdout.getvalue()
assert output == ""
assert chained_input.input == "foobar"
assert chained_input.input == "foo\n1bar\n2"
def test_chained_input_verbose() -> None:
@@ -37,19 +38,19 @@ def test_chained_input_verbose() -> None:
old_stdout = sys.stdout
sys.stdout = mystdout = StringIO()
chained_input.add("bar")
chained_input.add_observation("bar", "1", "2", None)
sys.stdout = old_stdout
output = mystdout.getvalue()
assert output == "bar"
assert chained_input.input == "foobar"
assert output == "\n1bar\n2"
assert chained_input.input == "foo\n1bar\n2"
old_stdout = sys.stdout
sys.stdout = mystdout = StringIO()
chained_input.add("baz", color="blue")
chained_input.add_observation("baz", "3", "4", "blue")
sys.stdout = old_stdout
output = mystdout.getvalue()
assert output == "\x1b[36;1m\x1b[1;3mbaz\x1b[0m"
assert chained_input.input == "foobarbaz"
assert output == "\n3\x1b[36;1m\x1b[1;3mbaz\x1b[0m\n4"
assert chained_input.input == "foo\n1bar\n2\n3baz\n4"
def test_get_color_mapping() -> None: