forked from Archives/langchain
Move PythonRepl -> langchain.utilities (#2917)
parent
7c73e9df5d
commit
36aa7f30e4
@ -1,26 +1,4 @@
|
|||||||
"""Mock Python REPL."""
|
"""For backwards compatibility."""
|
||||||
import sys
|
from langchain.utilities.python import PythonREPL
|
||||||
from io import StringIO
|
|
||||||
from typing import Dict, Optional
|
|
||||||
|
|
||||||
from pydantic import BaseModel, Field
|
__all__ = ["PythonREPL"]
|
||||||
|
|
||||||
|
|
||||||
class PythonREPL(BaseModel):
|
|
||||||
"""Simulates a standalone Python REPL."""
|
|
||||||
|
|
||||||
globals: Optional[Dict] = Field(default_factory=dict, alias="_globals")
|
|
||||||
locals: Optional[Dict] = Field(default_factory=dict, alias="_locals")
|
|
||||||
|
|
||||||
def run(self, command: str) -> str:
|
|
||||||
"""Run command with own globals/locals and returns anything printed."""
|
|
||||||
old_stdout = sys.stdout
|
|
||||||
sys.stdout = mystdout = StringIO()
|
|
||||||
try:
|
|
||||||
exec(command, self.globals, self.locals)
|
|
||||||
sys.stdout = old_stdout
|
|
||||||
output = mystdout.getvalue()
|
|
||||||
except Exception as e:
|
|
||||||
sys.stdout = old_stdout
|
|
||||||
output = str(e)
|
|
||||||
return output
|
|
||||||
|
@ -0,0 +1,25 @@
|
|||||||
|
import sys
|
||||||
|
from io import StringIO
|
||||||
|
from typing import Dict, Optional
|
||||||
|
|
||||||
|
from pydantic import BaseModel, Field
|
||||||
|
|
||||||
|
|
||||||
|
class PythonREPL(BaseModel):
|
||||||
|
"""Simulates a standalone Python REPL."""
|
||||||
|
|
||||||
|
globals: Optional[Dict] = Field(default_factory=dict, alias="_globals")
|
||||||
|
locals: Optional[Dict] = Field(default_factory=dict, alias="_locals")
|
||||||
|
|
||||||
|
def run(self, command: str) -> str:
|
||||||
|
"""Run command with own globals/locals and returns anything printed."""
|
||||||
|
old_stdout = sys.stdout
|
||||||
|
sys.stdout = mystdout = StringIO()
|
||||||
|
try:
|
||||||
|
exec(command, self.globals, self.locals)
|
||||||
|
sys.stdout = old_stdout
|
||||||
|
output = mystdout.getvalue()
|
||||||
|
except Exception as e:
|
||||||
|
sys.stdout = old_stdout
|
||||||
|
output = str(e)
|
||||||
|
return output
|
Loading…
Reference in New Issue