mirror of
https://github.com/hwchase17/langchain.git
synced 2025-09-19 00:58:32 +00:00
notebook fmt (#12498)
This commit is contained in:
@@ -1,5 +1,5 @@
|
||||
from neo4j_parent.chain import chain
|
||||
|
||||
if __name__ == "__main__":
|
||||
original_query = "What is the plot of the Dune?"
|
||||
print(chain.invoke(original_query))
|
||||
original_query = "What is the plot of the Dune?"
|
||||
print(chain.invoke(original_query))
|
||||
|
@@ -16,8 +16,7 @@ vectorstore = Neo4jVector.from_existing_index(
|
||||
index_name="retrieval",
|
||||
node_label="Child",
|
||||
embedding_node_property="embedding",
|
||||
retrieval_query=retrieval_query
|
||||
|
||||
retrieval_query=retrieval_query,
|
||||
)
|
||||
retriever = vectorstore.as_retriever()
|
||||
|
||||
@@ -37,8 +36,10 @@ chain = (
|
||||
| StrOutputParser()
|
||||
)
|
||||
|
||||
|
||||
# Add typing for input
|
||||
class Question(BaseModel):
|
||||
__root__: str
|
||||
|
||||
|
||||
chain = chain.with_types(input_type=Question)
|
||||
|
Reference in New Issue
Block a user