forked from Archives/langchain
48b093823e
Arbitrary transformation chains that can be used to add dictionary extractions from llms/other chains
42 lines
1018 B
Python
42 lines
1018 B
Python
"""Chain that runs an arbitrary python function."""
|
|
from typing import Callable, Dict, List
|
|
|
|
from pydantic import BaseModel
|
|
|
|
from langchain.chains.base import Chain
|
|
|
|
|
|
class TransformChain(Chain, BaseModel):
|
|
"""Chain transform chain output.
|
|
|
|
Example:
|
|
.. code-block:: python
|
|
|
|
from langchain import TransformChain
|
|
transform_chain = TransformChain(input_variables=["text"],
|
|
output_variables["entities"], transform=func())
|
|
"""
|
|
|
|
input_variables: List[str]
|
|
output_variables: List[str]
|
|
transform: Callable[[Dict[str, str]], Dict[str, str]]
|
|
|
|
@property
|
|
def input_keys(self) -> List[str]:
|
|
"""Expect input keys.
|
|
|
|
:meta private:
|
|
"""
|
|
return self.input_variables
|
|
|
|
@property
|
|
def output_keys(self) -> List[str]:
|
|
"""Return output keys.
|
|
|
|
:meta private:
|
|
"""
|
|
return self.output_variables
|
|
|
|
def _call(self, inputs: Dict[str, str]) -> Dict[str, str]:
|
|
return self.transform(inputs)
|