Pass through Tidewave team configuration

Author: josevalim

src/tidewave/django/__init__.py

@@ -29,6 +29,7 @@ class Middleware(MiddlewareMixin):
     Configuration:
         - ALLOWED_HOSTS: Used as allowed origins
         - TIDEWAVE['allow_remote_access']: Whether to allow remote connections (default False)
+        - TIDEWAVE['team']: Enable Tidewave for teams
     """
 
     def __init__(self, get_response: Callable):
@@ -66,6 +67,7 @@ def _build_config(self) -> dict[str, Any]:
         tidewave_settings = getattr(settings, "TIDEWAVE", {})
         allow_remote_access = tidewave_settings.get("allow_remote_access", False)
         client_url = tidewave_settings.get("client_url", "https://tidewave.ai")
+        team = tidewave_settings.get("team", {})
 
         project_name = "django_app"
         try:
@@ -79,6 +81,7 @@ def _build_config(self) -> dict[str, Any]:
             "client_url": client_url,
             "allow_remote_access": allow_remote_access,
             "allowed_origins": allowed_hosts,
+            "team": team,
         }
 
         return config

src/tidewave/middleware.py

@@ -92,6 +92,7 @@ def _handle_home_route(self, start_response: Callable) -> Iterator[bytes]:
         client_config = {
             "project_name": self.config.get("project_name", "unknown"),
             "framework_type": self.config.get("framework_type", "unknown"),
+            "team": self.config.get("team", {}),
             "tidewave_version": __version__,
         }
         config_json = html.escape(json.dumps(client_config))

tests/test_middleware.py

@@ -62,13 +62,13 @@ def test_non_mcp_route_passes_through(self):
 
         result = self.middleware(environ, self.start_response)
 
-        # Should call the wrapped app
         self.demo_app.assert_called_once_with(environ, self.start_response)
         self.assertEqual(result, [b"demo response"])
 
     def test_home_route_returns_html(self):
         """Test that /tidewave/ returns a valid HTML response"""
-        result = self.middleware._handle_home_route(self.start_response)
+        environ = self._create_environ("/tidewave")
+        result = self.middleware(environ, self.start_response)
 
         self.start_response.assert_called_once()
         call_args = self.start_response.call_args[0]
@@ -79,6 +79,7 @@ def test_home_route_returns_html(self):
         self.assertTrue(result)
         self.assertIsInstance(result[0], bytes)
         self.assertIn(b"tidewave:config", result[0].lower())
+        self.assertIn(b""team": {}", result[0].lower())
 
     def test_mcp_get_returns_405(self):
         """Test that GET requests to /tidewave/mcp return 405"""