core: docstrings documents (#23506)

Added missed docstrings. Formatted docstrings to the consistent form.
This commit is contained in:
Leonid Ganeline
2024-07-16 07:43:54 -07:00
committed by GitHub
parent 77dd327282
commit 5fcf2ef7ca
3 changed files with 57 additions and 11 deletions

View File

@@ -32,7 +32,16 @@ class BaseDocumentCompressor(BaseModel, ABC):
query: str,
callbacks: Optional[Callbacks] = None,
) -> Sequence[Document]:
"""Compress retrieved documents given the query context."""
"""Compress retrieved documents given the query context.
Args:
documents: The retrieved documents.
query: The query context.
callbacks: Optional callbacks to run during compression.
Returns:
The compressed documents.
"""
async def acompress_documents(
self,
@@ -40,7 +49,16 @@ class BaseDocumentCompressor(BaseModel, ABC):
query: str,
callbacks: Optional[Callbacks] = None,
) -> Sequence[Document]:
"""Compress retrieved documents given the query context."""
"""Async compress retrieved documents given the query context.
Args:
documents: The retrieved documents.
query: The query context.
callbacks: Optional callbacks to run during compression.
Returns:
The compressed documents.
"""
return await run_in_executor(
None, self.compress_documents, documents, query, callbacks
)

View File

@@ -10,9 +10,9 @@ if TYPE_CHECKING:
class BaseDocumentTransformer(ABC):
"""Abstract base class for document transformation systems.
"""Abstract base class for document transformation.
A document transformation system takes a sequence of Documents and returns a
A document transformation takes a sequence of Documents and returns a
sequence of transformed Documents.
Example:
@@ -55,7 +55,7 @@ class BaseDocumentTransformer(ABC):
documents: A sequence of Documents to be transformed.
Returns:
A list of transformed Documents.
A sequence of transformed Documents.
"""
async def atransform_documents(
@@ -67,7 +67,7 @@ class BaseDocumentTransformer(ABC):
documents: A sequence of Documents to be transformed.
Returns:
A list of transformed Documents.
A sequence of transformed Documents.
"""
return await run_in_executor(
None, self.transform_documents, documents, **kwargs