Merge pull request #340 from Sema4AI/action-server-additional-headers

Enable passing additional headers to the action server.
langchain-ai · May 28, 2024 · 9432c0b · 9432c0b
2 parents b85eb8d + ee19331
commit 9432c0b
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 10 deletions.
diff --git a/backend/app/tools.py b/backend/app/tools.py
@@ -287,7 +287,11 @@ def _get_tavily_answer():
 
 
 def _get_action_server(**kwargs: ActionServerConfig):
-    toolkit = ActionServerToolkit(url=kwargs["url"], api_key=kwargs["api_key"])
+    toolkit = ActionServerToolkit(
+        url=kwargs["url"],
+        api_key=kwargs["api_key"],
+        additional_headers=kwargs.get("additional_headers", {}),
+    )
     tools = toolkit.get_tools()
     return tools
 

diff --git a/backend/poetry.lock b/backend/poetry.lock
diff --git a/backend/pyproject.toml b/backend/pyproject.toml
@@ -32,7 +32,7 @@ wikipedia = "^1.4.0"
 langchain-google-vertexai = "^1.0.1"
 setuptools = "^69.0.3"
 pdfminer-six = "^20231228"
-langchain-robocorp = "^0.0.5"
+langchain-robocorp = "^0.0.8"
 fireworks-ai = "^0.11.2"
 httpx = { version = "0.25.2", extras = ["socks"] }
 unstructured = {extras = ["doc", "docx"], version = "^0.12.5"}