github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-27 08:58:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

community[patch]: fix llama index imports and fields access (#17870)

Browse Source 
- **Description:** Fixing outdated imports after v0.10 llama index
update and updating metadata and source text access
  - **Issue:** #17860
  - **Twitter handle:** @maximeperrin_

---------

Co-authored-by: Maxime Perrin <mperrin@doing.fr>
This commit is contained in:
Maxime Perrin 2024-02-26 03:14:23 +01:00 committed by GitHub
parent 5d2d80a9a8
commit c06a8732aa
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
libs/community/langchain_community/retrievers/llama_index.py
View File

@ -22,8 +22,8 @@ class LlamaIndexRetriever(BaseRetriever):
) -> List[Document]:
"""Get documents relevant for a query."""
try:
from llama_index.indices.base import BaseGPTIndex
from llama_index.response.schema import Response
from llama_index.core.base.response.schema import Response
from llama_index.core.indices.base import BaseGPTIndex
except ImportError:
raise ImportError(
"You need to install `pip install llama-index` to use this retriever."
@ -35,9 +35,9 @@ class LlamaIndexRetriever(BaseRetriever):
# parse source nodes
docs = []
for source_node in response.source_nodes:
metadata = source_node.extra_info or {}
metadata = source_node.metadata or {}
docs.append(
Document(page_content=source_node.source_text, metadata=metadata)
Document(page_content=source_node.get_content(), metadata=metadata)
)
return docs
@ -58,11 +58,11 @@ class LlamaIndexGraphRetriever(BaseRetriever):
) -> List[Document]:
"""Get documents relevant for a query."""
try:
from llama_index.composability.graph import (
from llama_index.core.base.response.schema import Response
from llama_index.core.composability.base import (
QUERY_CONFIG_TYPE,
ComposableGraph,
)
from llama_index.response.schema import Response
except ImportError:
raise ImportError(
"You need to install `pip install llama-index` to use this retriever."
@ -79,8 +79,8 @@ class LlamaIndexGraphRetriever(BaseRetriever):
# parse source nodes
docs = []
for source_node in response.source_nodes:
metadata = source_node.extra_info or {}
metadata = source_node.metadata or {}
docs.append(
Document(page_content=source_node.source_text, metadata=metadata)
Document(page_content=source_node.get_content(), metadata=metadata)
)
return docs