docs: robocorpai dosctrings (#17968)

Added missing docstrings

---------

Co-authored-by: Bagatur <22008038+baskaryan@users.noreply.github.com>
pull/17986/head
Leonid Ganeline 7 months ago committed by GitHub
parent b0cfb86c48
commit 971d29e718
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -33,7 +33,14 @@ LLM_TRACE_HEADER = "X-action-trace"
class RunDetailsCallbackHandler(BaseCallbackHandler):
"""Callback handler to add run details to the run."""
def __init__(self, run_details: dict) -> None:
"""Initialize the callback handler.
Args:
run_details (dict): Run details.
"""
self.run_details = run_details
def on_tool_start(
@ -50,10 +57,14 @@ class RunDetailsCallbackHandler(BaseCallbackHandler):
class ToolInputSchema(BaseModel):
"""Tool input schema."""
question: str = Field(...)
class ToolArgs(TypedDict):
"""Tool arguments."""
name: str
description: str
callback_manager: CallbackManager

Loading…
Cancel
Save