github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-05 13:06:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

templates: Update neo4j templates (#19789)

Browse Source
This commit is contained in:
Tomaz Bratanic
2024-03-30 15:40:05 +01:00
committed by GitHub
parent 765d6762bc
commit ed49cca191
13 changed files with 89 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
templates/neo4j-advanced-rag/ingest.py
View File

@@ -1,13 +1,11 @@
from pathlib import Path
from typing import List
from langchain.chains.openai_functions import create_structured_output_chain
from langchain_community.chat_models import ChatOpenAI
from langchain_community.document_loaders import TextLoader
from langchain_community.embeddings.openai import OpenAIEmbeddings
from langchain_community.graphs import Neo4jGraph
from langchain_core.prompts import ChatPromptTemplate
from langchain_core.pydantic_v1 import BaseModel, Field
from langchain_openai import ChatOpenAI, OpenAIEmbeddings
from langchain_text_splitters import TokenTextSplitter
from neo4j.exceptions import ClientError
@@ -116,10 +114,10 @@ questions_prompt = ChatPromptTemplate.from_messages(
]
)
question_chain = create_structured_output_chain(Questions, llm, questions_prompt)
question_chain = questions_prompt | llm.with_structured_output(Questions)
for i, parent in enumerate(parent_documents):
questions = question_chain.run(parent.page_content).questions
questions = question_chain.invoke(parent.page_content).questions
params = {
"parent_id": i,
"questions": [