global: MASSIVE snapshot

crates/brk_bindgen/src/generators/python/api.rs

@@ -2,7 +2,7 @@
 
 use std::fmt::Write;
 
-use crate::{Endpoint, Parameter, escape_python_keyword, to_snake_case};
+use crate::{Endpoint, Parameter, escape_python_keyword, generators::MANUAL_GENERIC_TYPES, to_snake_case};
 
 use super::client::generate_class_constants;
 use super::types::js_type_to_python;
@@ -41,11 +41,13 @@ pub fn generate_api_methods(output: &mut String, endpoints: &[Endpoint]) {
         }
 
         let method_name = endpoint_to_method_name(endpoint);
-        let return_type = endpoint
-            .response_type
-            .as_deref()
-            .map(js_type_to_python)
-            .unwrap_or_else(|| "Any".to_string());
+        let return_type = normalize_return_type(
+            &endpoint
+                .response_type
+                .as_deref()
+                .map(js_type_to_python)
+                .unwrap_or_else(|| "Any".to_string()),
+        );
 
         // Build method signature
         let params = build_method_params(endpoint);
@@ -149,3 +151,12 @@ fn build_path_template(path: &str, path_params: &[Parameter]) -> String {
     }
     result
 }
+
+/// Replace generic types with their Any variants in return types.
+fn normalize_return_type(return_type: &str) -> String {
+    let mut result = return_type.to_string();
+    for type_name in MANUAL_GENERIC_TYPES {
+        result = result.replace(type_name, &format!("Any{}", type_name));
+    }
+    result
+}

crates/brk_bindgen/src/generators/python/client.rs

@@ -118,11 +118,23 @@ def _m(acc: str, s: str) -> str:
     .unwrap();
 }
 
-/// Generate the Endpoint class
+/// Generate the MetricData and MetricEndpoint classes
 pub fn generate_endpoint_class(output: &mut String) {
     writeln!(
         output,
-        r#"class Endpoint(Generic[T]):
+        r#"class MetricData(TypedDict, Generic[T]):
+    """Metric data with range information."""
+    total: int
+    from_: int  # 'from' is reserved in Python
+    to: int
+    data: List[T]
+
+
+# Type alias for non-generic usage
+AnyMetricData = MetricData[Any]
+
+
+class MetricEndpoint(Generic[T]):
     """An endpoint for a specific metric + index combination."""
 
     def __init__(self, client: BrkClientBase, name: str, index: str):
@@ -130,11 +142,11 @@ pub fn generate_endpoint_class(output: &mut String) {
         self._name = name
         self._index = index
 
-    def get(self) -> List[T]:
+    def get(self) -> MetricData[T]:
         """Fetch all data points for this metric/index."""
         return self._client.get(self.path())
 
-    def range(self, from_val: Optional[int] = None, to_val: Optional[int] = None) -> List[T]:
+    def range(self, from_val: Optional[int] = None, to_val: Optional[int] = None) -> MetricData[T]:
         """Fetch data points within a range."""
         params = []
         if from_val is not None:
@@ -150,6 +162,10 @@ pub fn generate_endpoint_class(output: &mut String) {
         return f"/api/metric/{{self._name}}/{{self._index}}"
 
 
+# Type alias for non-generic usage
+AnyMetricEndpoint = MetricEndpoint[Any]
+
+
 class MetricPattern(Protocol[T]):
     """Protocol for metric patterns with different index sets."""
 
@@ -162,7 +178,7 @@ class MetricPattern(Protocol[T]):
         """Get the list of available indexes for this metric."""
         ...
 
-    def get(self, index: str) -> Optional[Endpoint[T]]:
+    def get(self, index: str) -> Optional[MetricEndpoint[T]]:
         """Get an endpoint for a specific index, if supported."""
         ...
 
@@ -199,10 +215,10 @@ pub fn generate_index_accessors(output: &mut String, patterns: &[IndexSetPattern
         for index in &pattern.indexes {
             let method_name = index_to_field_name(index);
             let index_name = index.serialize_long();
-            writeln!(output, "    def {}(self) -> Endpoint[T]:", method_name).unwrap();
+            writeln!(output, "    def {}(self) -> MetricEndpoint[T]:", method_name).unwrap();
             writeln!(
                 output,
-                "        return Endpoint(self._client, self._name, '{}')",
+                "        return MetricEndpoint(self._client, self._name, '{}')",
                 index_name
             )
             .unwrap();
@@ -250,7 +266,7 @@ pub fn generate_index_accessors(output: &mut String, patterns: &[IndexSetPattern
         writeln!(output).unwrap();
 
         // Generate get(index) method
-        writeln!(output, "    def get(self, index: str) -> Optional[Endpoint[T]]:").unwrap();
+        writeln!(output, "    def get(self, index: str) -> Optional[MetricEndpoint[T]]:").unwrap();
         writeln!(output, "        \"\"\"Get an endpoint for a specific index, if supported.\"\"\"").unwrap();
         for (i, index) in pattern.indexes.iter().enumerate() {
             let method_name = index_to_field_name(index);

crates/brk_bindgen/src/generators/python/types.rs

@@ -5,7 +5,7 @@ use std::fmt::Write;
 
 use serde_json::Value;
 
-use crate::{TypeSchemas, escape_python_keyword, ref_to_type_name};
+use crate::{TypeSchemas, escape_python_keyword, generators::MANUAL_GENERIC_TYPES, ref_to_type_name};
 
 /// Generate type definitions from schemas.
 pub fn generate_type_definitions(output: &mut String, schemas: &TypeSchemas) {
@@ -18,6 +18,10 @@ pub fn generate_type_definitions(output: &mut String, schemas: &TypeSchemas) {
     let sorted_names = topological_sort_schemas(schemas);
 
     for name in sorted_names {
+        if MANUAL_GENERIC_TYPES.contains(&name.as_str()) {
+            continue;
+        }
+
         let Some(schema) = schemas.get(&name) else {
             continue;
         };