mirror of
https://github.com/hwchase17/langchain
synced 2024-11-08 07:10:35 +00:00
11 lines
187 B
Python
11 lines
187 B
Python
|
from langchain.pydantic_v1 import BaseModel
|
||
|
|
|
||
|
|
from .router import branched_chain
|
||
|
|
|
||
|
|
|
||
|
|
class ChainInput(BaseModel):
|
||
|
|
message: str
|
||
|
|
|
||
|
|
|
||
|
|
chain = branched_chain.with_types(input_type=ChainInput)
|