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

node-onj-extraction

Browse Source
This commit is contained in:
Gal Shubeli 2025-03-23 15:25:30 +02:00
parent 3848a1371d
commit 74e5c242c3
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
libs/community/langchain_community/chains/graph_qa/falkordb.py
View File

@ -4,6 +4,7 @@ from __future__ import annotations
import re
from typing import Any, Dict, List, Optional
from falkordb.node import Node
from langchain.chains.base import Chain
from langchain.chains.llm import LLMChain
@ -164,7 +165,12 @@ class FalkorDBQAChain(Chain):
intermediate_steps.append({"query": generated_cypher})
# Retrieve and limit the number of results
context = self.graph.query(generated_cypher)[: self.top_k]
context_list = self.graph.query(generated_cypher)[: self.top_k]
# If the context is a list of nodes (object), extract the properties
context = [
[r.properties if isinstance(r, Node) else r for r in sublist]
for sublist in context_list
]
if self.return_direct:
final_result = context