style: format updated files with black

Author: Sakeeb91

app/cache.py

@@ -302,9 +302,7 @@ async def get(
             self._stats.misses += 1
             elapsed = (time.perf_counter() - start_time) * 1000
             self._update_miss_time(elapsed)
-            self._record_cache_operation(
-                namespace.value, "get", "miss", elapsed / 1000
-            )
+            self._record_cache_operation(namespace.value, "get", "miss", elapsed / 1000)
             logger.debug("cache_miss", key=key, namespace=namespace.value)
             return None
 
@@ -362,9 +360,7 @@ async def set(
                     namespace=namespace.value,
                     ttl=effective_ttl,
                 )
-                self._record_cache_operation(
-                    namespace.value, "set", "success", 0.0
-                )
+                self._record_cache_operation(namespace.value, "set", "success", 0.0)
                 return True
 
             # Fall back to in-memory cache

app/text2sql_engine.py

@@ -358,14 +358,18 @@ def validate(
         syntax_valid, syntax_errors = self._check_syntax(sql)
         if not syntax_valid:
             errors.extend(syntax_errors)
-            self._record_validation_metrics(database_id, ValidationStatus.INVALID_SYNTAX)
+            self._record_validation_metrics(
+                database_id, ValidationStatus.INVALID_SYNTAX
+            )
             return ValidationStatus.INVALID_SYNTAX, errors
 
         # Security check
         is_safe, security_errors = self._check_security(sql)
         if not is_safe:
             errors.extend(security_errors)
-            self._record_validation_metrics(database_id, ValidationStatus.DANGEROUS_QUERY)
+            self._record_validation_metrics(
+                database_id, ValidationStatus.DANGEROUS_QUERY
+            )
             return ValidationStatus.DANGEROUS_QUERY, errors
 
         # Schema alignment check
@@ -392,7 +396,9 @@ def _record_validation_metrics(
             valid=status == ValidationStatus.VALID,
         )
         if status == ValidationStatus.INVALID_SYNTAX:
-            metrics.record_sql_syntax_error(database_id=database_id, error_type="syntax")
+            metrics.record_sql_syntax_error(
+                database_id=database_id, error_type="syntax"
+            )
 
     def _check_syntax(self, sql: str) -> tuple[bool, list[str]]:
         """Check basic SQL syntax."""
@@ -801,11 +807,7 @@ async def generate_sql(
                 if show_reasoning:
                     reasoning_trace[-1].observation = "SQL validation passed"
 
-        if (
-            self._enable_validation
-            and valid_syntax
-            and inference_result.sql
-        ):
+        if self._enable_validation and valid_syntax and inference_result.sql:
             semantic_feedback = self._validator.check_semantics(
                 natural_query=natural_query,
                 sql=inference_result.sql,
@@ -1066,10 +1068,9 @@ def _build_prompt_cache_key(
             "use_default_examples": use_default_examples,
         }
 
-        return (
-            hashlib.sha256(json.dumps(key_data, sort_keys=True).encode())
-            .hexdigest()[:32]
-        )
+        return hashlib.sha256(
+            json.dumps(key_data, sort_keys=True).encode()
+        ).hexdigest()[:32]
 
     async def _generate_with_retry(
         self,

db/executor.py

@@ -269,9 +269,7 @@ async def execute(
                     self._execute_query(sql, params),
                     timeout=timeout,
                 )
-                self._tracing.set_span_attribute(
-                    "db.rows_returned", result.row_count
-                )
+                self._tracing.set_span_attribute("db.rows_returned", result.row_count)
 
             execution_time_ms = (time.perf_counter() - start_time) * 1000
             self._metrics.record_sql_query(

db/schema.py

@@ -112,12 +112,9 @@ def from_dict(cls, data: dict[str, Any]) -> "TableInfo":
         """Create from dictionary."""
         return cls(
             name=data.get("name", ""),
-            columns=[
-                ColumnInfo.from_dict(item) for item in data.get("columns", [])
-            ],
+            columns=[ColumnInfo.from_dict(item) for item in data.get("columns", [])],
             foreign_keys=[
-                ForeignKeyInfo.from_dict(item)
-                for item in data.get("foreign_keys", [])
+                ForeignKeyInfo.from_dict(item) for item in data.get("foreign_keys", [])
             ],
             primary_keys=list(data.get("primary_keys", [])),
             row_count=data.get("row_count"),

tests/unit/test_text2sql_engine.py

@@ -392,7 +392,9 @@ def test_semantic_top_limit_warning(self) -> None:
         )
 
         assert any("limit" in warning.lower() for warning in feedback.warnings)
-        assert any("limit 5" in suggestion.lower() for suggestion in feedback.suggestions)
+        assert any(
+            "limit 5" in suggestion.lower() for suggestion in feedback.suggestions
+        )
 
 
 # =============================================================================