Spaces:
Sleeping
Sleeping
File size: 929 Bytes
da21448 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | from collections.abc import Sequence
from typing import Protocol
from techdocs_graphrag.models import Chunk, GraphExtraction, GraphFact, RetrievedChunk
class LanguageModel(Protocol):
def complete(self, system_prompt: str, user_prompt: str, *, json_mode: bool = False) -> str: ...
def embed(self, texts: Sequence[str]) -> list[list[float]]: ...
class KnowledgeGraph(Protocol):
def ensure_schema(self) -> None: ...
def await_indexes(self) -> None: ...
def upsert_chunks(self, chunks: Sequence[Chunk], embeddings: Sequence[list[float]]) -> None: ...
def upsert_extraction(self, chunk: Chunk, extraction: GraphExtraction) -> None: ...
def vector_search(self, embedding: list[float], top_k: int) -> list[RetrievedChunk]: ...
def search_entities(self, text: str, limit: int) -> list[str]: ...
def expand(self, seed_keys: Sequence[str], max_hops: int, limit: int) -> list[GraphFact]: ...
|