diff --git a/libs/community/langchain_community/retrievers/embedchain.py b/libs/community/langchain_community/retrievers/embedchain.py index 0fae1b0f65a..9c64f628e1e 100644 --- a/libs/community/langchain_community/retrievers/embedchain.py +++ b/libs/community/langchain_community/retrievers/embedchain.py @@ -65,7 +65,10 @@ class EmbedchainRetriever(BaseRetriever): docs.append( Document( page_content=r["context"], - metadata={"source": r["source"], "document_id": r["document_id"]}, + metadata={ + "source": r["metadata"]["url"], + "document_id": r["metadata"]["doc_id"], + }, ) ) return docs diff --git a/libs/community/tests/integration_tests/retrievers/test_embedchain.py b/libs/community/tests/integration_tests/retrievers/test_embedchain.py index 1049616ef21..dc58938db17 100644 --- a/libs/community/tests/integration_tests/retrievers/test_embedchain.py +++ b/libs/community/tests/integration_tests/retrievers/test_embedchain.py @@ -19,8 +19,10 @@ os.environ["OPENAI_API_KEY"] = "sk-xxxx" context_value = [ { "context": "this document is about John", - "source": "source#1", - "document_id": 123, + "metadata": { + "source": "source#1", + "doc_id": 123, + }, }, ]