Skip to content

Commit

Permalink
added dosctrings
Browse files Browse the repository at this point in the history
  • Loading branch information
leo-gan committed Feb 22, 2024
1 parent f6e3aa9 commit 88ce110
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions libs/partners/robocorp/langchain_robocorp/toolkits.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,13 @@


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(
Expand All @@ -50,10 +56,12 @@ def on_tool_start(


class ToolInputSchema(BaseModel):
"""Tool input schema."""
question: str = Field(...)


class ToolArgs(TypedDict):
"""Tool arguments."""
name: str
description: str
callback_manager: CallbackManager
Expand Down

0 comments on commit 88ce110

Please sign in to comment.