Update output

BenConstable9 · BenConstable9 · commit 5bdab7c9c107 · 2025-02-03T19:47:39.000Z
diff --git a/text_2_sql/text_2_sql_core/src/text_2_sql_core/structured_outputs/user_message_rewrite_agent.py b/text_2_sql/text_2_sql_core/src/text_2_sql_core/structured_outputs/user_message_rewrite_agent.py
@@ -4,7 +4,7 @@
 
 
 class UserMessageRewriteAgentOutput(BaseModel):
-    decomposed_user_messages: list[list[str]]
-    combination_logic: str
-    query_type: str
-    all_non_database_query: bool
+    """The output of the user message rewrite agent."""
+
+    steps: list[list[str]]
+    requires_sql_queries: bool
