From 010a234f1ea20edf8d4c75c09ab9b8fd560717fb Mon Sep 17 00:00:00 2001 From: Tomaz Bratanic Date: Fri, 8 Mar 2024 04:25:41 +0100 Subject: [PATCH] docs: Fix diffbot graph transformer description (#18736) The previous docstring was invalid --- .../graph_transformers/diffbot.py | 21 ++++++------------- 1 file changed, 6 insertions(+), 15 deletions(-) diff --git a/libs/experimental/langchain_experimental/graph_transformers/diffbot.py b/libs/experimental/langchain_experimental/graph_transformers/diffbot.py index 0e8613b3f2..b174b1353d 100644 --- a/libs/experimental/langchain_experimental/graph_transformers/diffbot.py +++ b/libs/experimental/langchain_experimental/graph_transformers/diffbot.py @@ -123,24 +123,15 @@ class DiffbotGraphTransformer: Example: .. code-block:: python + from langchain_experimental.graph_transformers import DiffbotGraphTransformer + from langchain_core.documents import Document - class DiffbotGraphTransformer(BaseGraphDocumentTransformer): + diffbot_api_key = "DIFFBOT_API_KEY" + diffbot_nlp = DiffbotGraphTransformer(diffbot_api_key=diffbot_api_key) - def transform_documents( - self, documents: Sequence[Document], **kwargs: Any - ) -> Sequence[GraphDocument]: - results = [] + document = Document(page_content="Mike Tunge is the CEO of Diffbot.") + graph_documents = diffbot_nlp.convert_to_graph_documents([document]) - for document in documents: - raw_results = self.nlp_request(document.page_content) - graph_document = self.process_response(raw_results, document) - results.append(graph_document) - return results - - async def atransform_documents( - self, documents: Sequence[Document], **kwargs: Any - ) -> Sequence[Document]: - raise NotImplementedError """ def __init__(