github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-18 16:16:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

Harrison/base agent without docs (#2166)

Browse Source
This commit is contained in:
Harrison Chase
2023-03-29 22:11:25 -07:00
committed by GitHub
parent 1b7cfd7222
commit 5c907d9998
5 changed files with 212 additions and 81 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/unit_tests/agents/test_agent.py
View File

@@ -149,7 +149,7 @@ def test_agent_with_callbacks_local() -> None:
callback_manager=manager,
)
agent.agent.llm_chain.verbose = True
agent.agent.llm_chain.verbose = True # type: ignore
output = agent.run("when was langchain made")
assert output == "curses foiled again"
@@ -285,8 +285,8 @@ def test_agent_with_new_prefix_suffix() -> None:
)
# avoids "BasePromptTemplate" has no attribute "template" error
assert hasattr(agent.agent.llm_chain.prompt, "template")
prompt_str = agent.agent.llm_chain.prompt.template
assert hasattr(agent.agent.llm_chain.prompt, "template") # type: ignore
prompt_str = agent.agent.llm_chain.prompt.template # type: ignore
assert prompt_str.startswith(prefix), "Prompt does not start with prefix"
assert prompt_str.endswith(suffix), "Prompt does not end with suffix"