Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-18 09:25:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

docs: robocorpai dosctrings (#17968)

Browse Source 
Added missing docstrings

---------

Co-authored-by: Bagatur <22008038+baskaryan@users.noreply.github.com>
This commit is contained in:
Leonid Ganeline 2024-02-22 15:55:01 -08:00 committed by GitHub
parent b0cfb86c48
commit 971d29e718
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
libs/partners/robocorp/langchain_robocorp/toolkits.py
View File

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