langchain/templates/sql-research-assistant/sql_research_assistant/chain.py

from langchain.pydantic_v1 import BaseModel
from langchain_core.runnables import RunnablePassthrough

from sql_research_assistant.search.web import chain as search_chain
from sql_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)


if __name__ == "__main__":
    print(  # noqa: T201
        chain.invoke({"question": "who is typically older: point guards or centers?"})
    )
WIP: sql research assistant (#14240) 2023-12-18 22:00:18 +00:00			`from langchain.pydantic_v1 import BaseModel`
langchain[patch], experimental[patch]: replace langchain.schema imports (#15410) Import from core instead. Ran: ```bash git grep -l 'from langchain.schema\.output_parser' \| xargs -L 1 sed -i '' "s/from\ langchain\.schema\.output_parser/from\ langchain_core.output_parsers/g" git grep -l 'from langchain.schema\.messages' \| xargs -L 1 sed -i '' "s/from\ langchain\.schema\.messages/from\ langchain_core.messages/g" git grep -l 'from langchain.schema\.document' \| xargs -L 1 sed -i '' "s/from\ langchain\.schema\.document/from\ langchain_core.documents/g" git grep -l 'from langchain.schema\.runnable' \| xargs -L 1 sed -i '' "s/from\ langchain\.schema\.runnable/from\ langchain_core.runnables/g" git grep -l 'from langchain.schema\.vectorstore' \| xargs -L 1 sed -i '' "s/from\ langchain\.schema\.vectorstore/from\ langchain_core.vectorstores/g" git grep -l 'from langchain.schema\.language_model' \| xargs -L 1 sed -i '' "s/from\ langchain\.schema\.language_model/from\ langchain_core.language_models/g" git grep -l 'from langchain.schema\.embeddings' \| xargs -L 1 sed -i '' "s/from\ langchain\.schema\.embeddings/from\ langchain_core.embeddings/g" git grep -l 'from langchain.schema\.storage' \| xargs -L 1 sed -i '' "s/from\ langchain\.schema\.storage/from\ langchain_core.stores/g" git checkout master libs/langchain/tests/unit_tests/schema/ make format cd libs/experimental make format cd ../langchain make format ``` 2024-01-02 20:09:45 +00:00			`from langchain_core.runnables import RunnablePassthrough`
WIP: sql research assistant (#14240) 2023-12-18 22:00:18 +00:00
			`from sql_research_assistant.search.web import chain as search_chain`
			`from sql_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)`


			`if __name__ == "__main__":`
infra: add print rule to ruff (#16221) Added noqa for existing prints. Can slowly remove / will prevent more being intro'd 2024-02-10 00:13:30 +00:00			`print( # noqa: T201`
WIP: sql research assistant (#14240) 2023-12-18 22:00:18 +00:00			`chain.invoke({"question": "who is typically older: point guards or centers?"})`
			`)`