From 3b5ac44e039888fa894a54442cf6eae7430cbbd5 Mon Sep 17 00:00:00 2001 From: Bakar Tavadze Date: Fri, 17 May 2024 19:08:48 +0400 Subject: [PATCH] langchain-robocorp[minor]: Enable passing additional headers to the action server. (#21809) Actions can optionally receive secrets via request headers. This PR enables this functionality. --- libs/partners/robocorp/langchain_robocorp/toolkits.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/libs/partners/robocorp/langchain_robocorp/toolkits.py b/libs/partners/robocorp/langchain_robocorp/toolkits.py index fc21cf0959e..812049ac1e9 100644 --- a/libs/partners/robocorp/langchain_robocorp/toolkits.py +++ b/libs/partners/robocorp/langchain_robocorp/toolkits.py @@ -103,6 +103,8 @@ class ActionServerToolkit(BaseModel): """Action Server URL""" api_key: str = Field(exclude=True, default="") """Action Server request API key""" + additional_headers: dict = Field(exclude=True, default_factory=dict) + """Additional headers to be passed to the Action Server""" report_trace: bool = Field(exclude=True, default=False) """Enable reporting Langsmith trace to Action Server runs""" _run_details: dict = PrivateAttr({}) @@ -226,6 +228,7 @@ class ActionServerToolkit(BaseModel): headers = { "Authorization": f"Bearer {self.api_key}", "Content-Type": "application/json", + **self.additional_headers, } try: