github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-04 06:03:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

python repl improvement for csv agent (#9618)

Browse Source
This commit is contained in:
Harrison Chase 2023-08-22 17:06:18 -07:00 committed by GitHub
parent 632a83c48e
commit 02545a54b3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
libs/langchain/langchain/tools/python/tool.py
View File

@ -6,13 +6,13 @@ import re
import sys import sys
from contextlib import redirect_stdout from contextlib import redirect_stdout
from io import StringIO from io import StringIO
from typing import Any, Dict, Optional from typing import Any, Dict, Optional, Type
from langchain.callbacks.manager import ( from langchain.callbacks.manager import (
AsyncCallbackManagerForToolRun, AsyncCallbackManagerForToolRun,
CallbackManagerForToolRun, CallbackManagerForToolRun,
) )
from langchain.pydantic_v1 import Field, root_validator from langchain.pydantic_v1 import BaseModel, Field, root_validator
from langchain.tools.base import BaseTool from langchain.tools.base import BaseTool
from langchain.utilities import PythonREPL from langchain.utilities import PythonREPL
@ -77,6 +77,10 @@ class PythonREPLTool(BaseTool):
return result return result
class PythonInputs(BaseModel):
query: str = Field(description="code snippet to run")
class PythonAstREPLTool(BaseTool): class PythonAstREPLTool(BaseTool):
"""A tool for running python code in a REPL.""" """A tool for running python code in a REPL."""
@ -90,6 +94,7 @@ class PythonAstREPLTool(BaseTool):
globals: Optional[Dict] = Field(default_factory=dict) globals: Optional[Dict] = Field(default_factory=dict)
locals: Optional[Dict] = Field(default_factory=dict) locals: Optional[Dict] = Field(default_factory=dict)
sanitize_input: bool = True sanitize_input: bool = True
args_schema: Type[BaseModel] = PythonInputs
@root_validator(pre=True) @root_validator(pre=True)
def validate_python_version(cls, values: Dict) -> Dict: def validate_python_version(cls, values: Dict) -> Dict: