mirror of
https://github.com/hwchase17/langchain
synced 2024-11-04 06:00:26 +00:00
8e0d5813c2
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 ```
23 lines
572 B
Python
23 lines
572 B
Python
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(
|
|
chain.invoke({"question": "who is typically older: point guards or centers?"})
|
|
)
|