mirror of
https://github.com/Mintplex-Labs/langchain-python.git
synced 2026-07-21 16:55:25 -04:00
Harrison Chase
f23b3ceb49
consolidating logic for when a chain is able to run with single input text, single output text open to feedback on naming, logic, usefulness
52 lines
1.2 KiB
Python
52 lines
1.2 KiB
Python
"""Chain that runs python code.
|
|
|
|
Heavily borrowed from https://replit.com/@amasad/gptpy?v=1#main.py
|
|
"""
|
|
import sys
|
|
from io import StringIO
|
|
from typing import Dict, List
|
|
|
|
from pydantic import BaseModel
|
|
|
|
from langchain.chains.base import Chain
|
|
from langchain.python import PythonREPL
|
|
|
|
|
|
class PythonChain(Chain, BaseModel):
|
|
"""Chain to run python code.
|
|
|
|
Example:
|
|
.. code-block:: python
|
|
|
|
from langchain import PythonChain
|
|
python_chain = PythonChain()
|
|
"""
|
|
|
|
input_key: str = "code" #: :meta private:
|
|
output_key: str = "output" #: :meta private:
|
|
|
|
@property
|
|
def input_keys(self) -> List[str]:
|
|
"""Expect input key.
|
|
|
|
:meta private:
|
|
"""
|
|
return [self.input_key]
|
|
|
|
@property
|
|
def output_keys(self) -> List[str]:
|
|
"""Return output key.
|
|
|
|
:meta private:
|
|
"""
|
|
return [self.output_key]
|
|
|
|
def _call(self, inputs: Dict[str, str]) -> Dict[str, str]:
|
|
python_repl = PythonREPL()
|
|
old_stdout = sys.stdout
|
|
sys.stdout = mystdout = StringIO()
|
|
python_repl.run(inputs[self.input_key])
|
|
sys.stdout = old_stdout
|
|
output = mystdout.getvalue()
|
|
return {self.output_key: output}
|