notebook fmt (#12498)

This commit is contained in:
Bagatur
2023-10-29 15:50:09 -07:00
committed by GitHub
parent 56cc5b847c
commit 2424fff3f1
342 changed files with 8261 additions and 6796 deletions

View File

@@ -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))

View File

@@ -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)