File size: 187 Bytes
ed4d993
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
from langchain.pydantic_v1 import BaseModel

from .router import branched_chain


class ChainInput(BaseModel):
    message: str


chain = branched_chain.with_types(input_type=ChainInput)