docs: robocorpai dosctrings (#17968)

Added missing docstrings --------- Co-authored-by: Bagatur <[email protected]>
langchain-ai · Feb 22, 2024 · 971d29e · 971d29e
1 parent b0cfb86
commit 971d29e
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/libs/partners/robocorp/langchain_robocorp/toolkits.py b/libs/partners/robocorp/langchain_robocorp/toolkits.py
@@ -33,7 +33,14 @@
 
 
 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 @@ 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