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

Core: fix Anthropic json issue in streaming (#16670)

Browse Source 
**Description:** fix ChatAnthropic json issue in streaming 
**Issue:** https://github.com/langchain-ai/langchain/issues/16423
**Dependencies:** n/a

---------

Co-authored-by: Harrison Chase <hw.chase.17@gmail.com>
This commit is contained in:
Tze Min 2024-01-29 08:41:17 +08:00 committed by GitHub
parent e451c8adc1
commit 6ef718c5f4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 42 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
libs/core/langchain_core/output_parsers/json.py
View File

@ -138,7 +138,7 @@ def parse_json_markdown(
The parsed JSON object as a Python dictionary. The parsed JSON object as a Python dictionary.
""" """
# Try to find JSON string within triple backticks # Try to find JSON string within triple backticks
match = re.search(r"```(json)?(.*)```", json_string, re.DOTALL) match = re.search(r"```(json)?(.*)(```)?", json_string, re.DOTALL)
# If no match found, assume the entire string is a JSON string # If no match found, assume the entire string is a JSON string
if match is None: if match is None:

41
libs/core/tests/unit_tests/output_parsers/test_json.py
View File

@ -137,6 +137,43 @@ TEXT_BEFORE_AND_AFTER = """Action: Testing
``` ```
This should do the trick""" This should do the trick"""
WITHOUT_END_BRACKET = """Here is a response formatted as schema:
```json
{
"foo": "bar"
"""
WITH_END_BRACKET = """Here is a response formatted as schema:
```json
{
"foo": "bar"
}
"""
WITH_END_TICK = """Here is a response formatted as schema:
```json
{
"foo": "bar"
}
```
"""
WITH_END_TEXT = """Here is a response formatted as schema:
```
{
"foo": "bar"
```
This should do the trick
"""
TEST_CASES = [ TEST_CASES = [
GOOD_JSON, GOOD_JSON,
JSON_WITH_NEW_LINES, JSON_WITH_NEW_LINES,
@ -148,6 +185,10 @@ TEST_CASES = [
TEXT_BEFORE, TEXT_BEFORE,
TEXT_AFTER, TEXT_AFTER,
TEXT_BEFORE_AND_AFTER, TEXT_BEFORE_AND_AFTER,
WITHOUT_END_BRACKET,
WITH_END_BRACKET,
WITH_END_TICK,
WITH_END_TEXT,
] ]