core: docstrings `BaseCallbackHandler update (#24948)

Added missed docstrings
This commit is contained in:
Leonid Ganeline 2024-08-01 17:46:53 -07:00 committed by GitHub
parent 6e45dba471
commit 4092876863
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -25,7 +25,14 @@ class RetrieverManagerMixin:
parent_run_id: Optional[UUID] = None,
**kwargs: Any,
) -> Any:
"""Run when Retriever errors."""
"""Run when Retriever errors.
Args:
error (BaseException): The error that occurred.
run_id (UUID): The run ID. This is the ID of the current run.
parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
kwargs (Any): Additional keyword arguments.
"""
def on_retriever_end(
self,
@ -35,7 +42,14 @@ class RetrieverManagerMixin:
parent_run_id: Optional[UUID] = None,
**kwargs: Any,
) -> Any:
"""Run when Retriever ends running."""
"""Run when Retriever ends running.
Args:
documents (Sequence[Document]): The documents retrieved.
run_id (UUID): The run ID. This is the ID of the current run.
parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
kwargs (Any): Additional keyword arguments.
"""
class LLMManagerMixin: