agent serialization (#4642)

This commit is contained in:
Harrison Chase
2023-05-13 21:47:10 -07:00
committed by GitHub
parent ef49c659f6
commit fbfa49f2c1
5 changed files with 51 additions and 2 deletions

View File

@@ -0,0 +1,19 @@
from pathlib import Path
from tempfile import TemporaryDirectory
from langchain.agents.agent_types import AgentType
from langchain.agents.initialize import initialize_agent, load_agent
from langchain.llms.fake import FakeListLLM
def test_mrkl_serialization() -> None:
agent = initialize_agent(
[],
FakeListLLM(responses=[]),
agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
verbose=True,
)
with TemporaryDirectory() as tempdir:
file = Path(tempdir) / "agent.json"
agent.save_agent(file)
load_agent(file)