mirror of
https://github.com/hwchase17/langchain
synced 2024-10-29 17:07:25 +00:00
5f17c57174
Currently `ChatOutputParser` extracts actions by splitting the text on
"```", and then load the second part as a json string.
But sometimes the LLM will wrap the action in markdown code block like:
````markdown
```json
{
"action": "foo",
"action_input": "bar"
}
```
````
Splitting text on "```" will cause `OutputParserException` in such case.
This PR changes the behaviour to extract the `$JSON_BLOB` by regex, so
that it can handle both ` ``` ``` ` and ` ```json ``` `
@hinthornw
---------
Co-authored-by: Junlin Zhou <jlzhou@zjuici.com>
48 lines
1.1 KiB
Python
48 lines
1.1 KiB
Python
"""Unittests for langchain.agents.chat package."""
|
|
from typing import Tuple
|
|
|
|
from langchain.agents.chat.output_parser import ChatOutputParser
|
|
from langchain.schema import AgentAction
|
|
|
|
output_parser = ChatOutputParser()
|
|
|
|
|
|
def get_action_and_input(text: str) -> Tuple[str, str]:
|
|
output = output_parser.parse(text)
|
|
if isinstance(output, AgentAction):
|
|
return output.tool, str(output.tool_input)
|
|
else:
|
|
return "Final Answer", output.return_values["output"]
|
|
|
|
|
|
def test_parse_with_language() -> None:
|
|
llm_output = """I can use the `foo` tool to achieve the goal.
|
|
|
|
Action:
|
|
```json
|
|
{
|
|
"action": "foo",
|
|
"action_input": "bar"
|
|
}
|
|
```
|
|
"""
|
|
action, action_input = get_action_and_input(llm_output)
|
|
assert action == "foo"
|
|
assert action_input == "bar"
|
|
|
|
|
|
def test_parse_without_language() -> None:
|
|
llm_output = """I can use the `foo` tool to achieve the goal.
|
|
|
|
Action:
|
|
```
|
|
{
|
|
"action": "foo",
|
|
"action_input": "bar"
|
|
}
|
|
```
|
|
"""
|
|
action, action_input = get_action_and_input(llm_output)
|
|
assert action == "foo"
|
|
assert action_input == "bar"
|