core,community: add beta decorator to missed GraphVectorStore extensions (#25562)

This commit is contained in:
Erick Friis
2024-08-19 17:29:09 -07:00
committed by GitHub
parent dd2d094adc
commit e01c6789c4
10 changed files with 23 additions and 0 deletions

View File

@@ -9,6 +9,7 @@ from typing import (
Type,
)
from langchain_core._api import beta
from langchain_core.documents import Document
from langchain_core.embeddings import Embeddings
from langchain_core.graph_vectorstores.base import (
@@ -23,6 +24,7 @@ if TYPE_CHECKING:
from cassandra.cluster import Session
@beta()
class CassandraGraphVectorStore(GraphVectorStore):
def __init__(
self,

View File

@@ -1,5 +1,6 @@
from typing import Any, Dict, Iterable, List, Optional, Set, Union
from langchain_core._api import beta
from langchain_core.documents import Document
from langchain_core.graph_vectorstores.links import Link
@@ -11,6 +12,7 @@ from langchain_community.graph_vectorstores.extractors.link_extractor import (
GLiNERInput = Union[str, Document]
@beta()
class GLiNERLinkExtractor(LinkExtractor[GLiNERInput]):
"""Link documents with common named entities using GLiNER <https://github.com/urchade/GLiNER>."""

View File

@@ -1,5 +1,6 @@
from typing import Callable, List, Set
from langchain_core._api import beta
from langchain_core.documents import Document
from langchain_core.graph_vectorstores.links import Link
@@ -18,6 +19,7 @@ _CHILD: str = "c:"
_SIBLING: str = "s:"
@beta()
class HierarchyLinkExtractor(LinkExtractor[HierarchyInput]):
def __init__(
self,

View File

@@ -4,6 +4,7 @@ from dataclasses import dataclass
from typing import TYPE_CHECKING, List, Optional, Set, Union
from urllib.parse import urldefrag, urljoin, urlparse
from langchain_core._api import beta
from langchain_core.documents import Document
from langchain_core.graph_vectorstores import Link
@@ -61,6 +62,7 @@ class HtmlInput:
base_url: str
@beta()
class HtmlLinkExtractor(LinkExtractor[HtmlInput]):
def __init__(self, *, kind: str = "hyperlink", drop_fragments: bool = True):
"""Extract hyperlinks from HTML content.

View File

@@ -1,5 +1,6 @@
from typing import Any, Dict, Iterable, Optional, Set, Union
from langchain_core._api import beta
from langchain_core.documents import Document
from langchain_core.graph_vectorstores.links import Link
@@ -10,6 +11,7 @@ from langchain_community.graph_vectorstores.extractors.link_extractor import (
KeybertInput = Union[str, Document]
@beta()
class KeybertLinkExtractor(LinkExtractor[KeybertInput]):
def __init__(
self,

View File

@@ -3,6 +3,7 @@ from __future__ import annotations
from abc import ABC, abstractmethod
from typing import Generic, Iterable, Set, TypeVar
from langchain_core._api import beta
from langchain_core.graph_vectorstores import Link
InputT = TypeVar("InputT")
@@ -10,6 +11,7 @@ InputT = TypeVar("InputT")
METADATA_LINKS_KEY = "links"
@beta()
class LinkExtractor(ABC, Generic[InputT]):
"""Interface for extracting links (incoming, outgoing, bidirectional)."""

View File

@@ -1,5 +1,6 @@
from typing import Callable, Iterable, Set, TypeVar
from langchain_core._api import beta
from langchain_core.graph_vectorstores import Link
from langchain_community.graph_vectorstores.extractors.link_extractor import (
@@ -10,6 +11,7 @@ InputT = TypeVar("InputT")
UnderlyingInputT = TypeVar("UnderlyingInputT")
@beta()
class LinkExtractorAdapter(LinkExtractor[InputT]):
def __init__(
self,

View File

@@ -1,5 +1,6 @@
from typing import Any, Sequence
from langchain_core._api import beta
from langchain_core.documents import Document
from langchain_core.documents.transformers import BaseDocumentTransformer
from langchain_core.graph_vectorstores.links import copy_with_links
@@ -9,6 +10,7 @@ from langchain_community.graph_vectorstores.extractors.link_extractor import (
)
@beta()
class LinkExtractorTransformer(BaseDocumentTransformer):
"""DocumentTransformer for applying one or more LinkExtractors.