Dh-5567/reformat

Dataherald · Mar 22, 2024 · a7d1f2f · a7d1f2f
1 parent 6196e2d
commit a7d1f2f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/dataherald/sql_generator/__init__.py b/dataherald/sql_generator/__init__.py
@@ -137,7 +137,11 @@ def construct_intermediate_steps(
 
     def truncate_observations(self, obervarion: str, max_length: int = 2000) -> str:
         """Truncate the tool input."""
-        return obervarion[:max_length] + "... (truncated)" if len(obervarion) > max_length else obervarion
+        return (
+            obervarion[:max_length] + "... (truncated)"
+            if len(obervarion) > max_length
+            else obervarion
+        )
 
     @abstractmethod
     def generate_response(