Fixing headers for http_request

databricks · Jan 3, 2025 · 9485351 · 9485351
1 parent 966bb41
commit 9485351
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 8 deletions.
diff --git a/src/databricks_ai_bridge/external_tools.py b/src/databricks_ai_bridge/external_tools.py
@@ -50,8 +50,8 @@ def http_request(
         "method": method,
         "path": path,
         "json": js.dumps(json),
-        "header": headers,
-        "params": params,
+        "headers": js.dumps(headers),
+        "params": js.dumps(params),
     }
 
     return requests.post(url, headers=request_headers, json=payload)
diff --git a/tests/databricks_ai_bridge/test_external_tools.py b/tests/databricks_ai_bridge/test_external_tools.py
@@ -41,10 +41,8 @@ def test_http_request_success(mock_post, mock_workspace_client):
             "method": "POST",
             "path": "/mock-path",
             "json": '{"key": "value"}',
-            "header": {
-                "Custom-Header": "HeaderValue",
-            },
-            "params": {"query": "test"},
+            "headers": '{"Custom-Header": "HeaderValue"}',
+            "params": '{"query": "test"}',
         },
     )
 
@@ -83,7 +81,7 @@ def test_http_request_error_response(mock_post, mock_workspace_client):
             "method": "POST",
             "path": "/mock-path",
             "json": '{"key": "value"}',
-            "header": None,
-            "params": None,
+            "headers": "null",
+            "params": "null",
         },
     )