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

Harrison/multiline commands (#2280)

Browse Source 
Co-authored-by: Marc Päpper <mpaepper@users.noreply.github.com>
This commit is contained in:
Harrison Chase 2023-04-01 12:54:06 -07:00 committed by GitHub
parent a9dddd8a32
commit acfda4d1d8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 20 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
langchain/tools/python/tool.py
View File

@ -25,9 +25,12 @@ class PythonREPLTool(BaseTool):
"with `print(...)`."
)
python_repl: PythonREPL = Field(default_factory=_get_default_python_repl)
sanitize_input: bool = True
def _run(self, query: str) -> str:
"""Use the tool."""
if self.sanitize_input:
query = query.strip().strip("```")
return self.python_repl.run(query)
async def _arun(self, query: str) -> str:

17
tests/unit_tests/test_python.py
View File

@ -1,6 +1,15 @@
"""Test functionality of Python REPL."""
from langchain.python import PythonREPL
from langchain.tools.python.tool import PythonREPLTool
_SAMPLE_CODE = """
```
def multiply():
print(5*6)
multiply()
```
"""
def test_python_repl() -> None:
@ -43,6 +52,14 @@ def test_functionality() -> None:
assert output == "2\n"
def test_functionality_multiline() -> None:
"""Test correct functionality for ChatGPT multiline commands."""
chain = PythonREPL()
tool = PythonREPLTool(python_repl=chain)
output = tool.run(_SAMPLE_CODE)
assert output == "30\n"
def test_function() -> None:
"""Test correct functionality."""
chain = PythonREPL()