Convert context into query string for FGA queries

Author: stanleyphu

tests/test_fga.py

@@ -534,3 +534,14 @@ def test_query(self, mock_query_response, mock_http_client_with_response):
             warrant_token="warrant_token",
         )
         assert response.dict(exclude_none=True) == mock_query_response
+
+    def test_query_with_context(self, mock_query_response, mock_http_client_with_response):
+        mock_http_client_with_response(self.http_client, mock_query_response, 200)
+
+        response = self.fga.query(
+            q="select member of type user for permission:view-docs",
+            order="asc",
+            warrant_token="warrant_token",
+            context={"region": "us", "subscription": "pro"},
+        )
+        assert response.dict(exclude_none=True) == mock_query_response

workos/fga.py

@@ -1,3 +1,4 @@
+import json
 from typing import Any, Mapping, Optional, Protocol, Sequence
 from workos.types.fga import (
     CheckOperation,
@@ -621,11 +622,16 @@ def query(
             "after": after,
             "context": context,
         }
+        parsed_list_params = {
+            key: json.dumps(value) if key == "context" and value is not None else value
+            for key, value in list_params.items()
+            if value is not None
+        }
 
         response = self._http_client.request(
             "fga/v1/query",
             method=REQUEST_METHOD_GET,
-            params=list_params,
+            params=parsed_list_params,
             headers={"Warrant-Token": warrant_token} if warrant_token else None,
         )