mirror of
https://github.com/hwchase17/langchain
synced 2024-11-18 09:25:54 +00:00
17 lines
432 B
Python
17 lines
432 B
Python
from langchain.pydantic_v1 import BaseModel
|
|
from langchain.schema.runnable import RunnablePassthrough
|
|
|
|
from research_assistant.search.web import chain as search_chain
|
|
from research_assistant.writer import chain as writer_chain
|
|
|
|
chain_notypes = (
|
|
RunnablePassthrough().assign(research_summary=search_chain) | writer_chain
|
|
)
|
|
|
|
|
|
class InputType(BaseModel):
|
|
question: str
|
|
|
|
|
|
chain = chain_notypes.with_types(input_type=InputType)
|