Update: Sync Tool Definitions with Backend

src/core/agent_middleware.py

@@ -11,36 +11,57 @@ class AgentMiddleware:
             self.engine = None
 
     def get_db_schema(self):
-        if not self.engine: return "Database not connected."
-        schema_text = []
-        try:
-            with self.engine.connect() as conn:
-                if 'postgres' in self.config.DB_URL:
-                    sql = text("SELECT table_name FROM information_schema.tables WHERE table_schema = 'public'")
-                else:
-                    sql = text("SELECT name FROM sqlite_master WHERE type='table'")
-                
-                tables = conn.execute(sql).fetchall()
-                for t in tables:
-                    table_name = t[0]
-                    if table_name in ['sqlite_sequence', 'alembic_version']: continue
-                    try:
-                        cols = conn.execute(text(f"SELECT * FROM {table_name} LIMIT 0")).keys()
-                        schema_text.append(f"- TABLE {table_name.upper()}: {list(cols)}")
-                    except: pass
-            return "\n".join(schema_text)
-        except Exception as e:
-            return f"Error fetching schema: {e}"
+        # The AI on HF doesn't need to know its OWN local DB schema anymore, 
+        # because the Flask App will inject the REAL schema. 
+        # We return a placeholder here or keep it for internal memory.
+        return "Schema provided in user context."
 
     def get_workflow_tools(self):
         return """
         [AVAILABLE WORKFLOW NODES]
-        1. 'google_sheet_read' { "sheetId": "...", "range": "A1:Z" }
-        2. 'google_sheet_write' { "sheetId": "...", "data": "{{parent.output}}", "mode": "append" }
-        3. 'gmail_send' { "to": "...", "subject": "...", "body": "..." }
-        4. 'filter' { "condition": "contains", "field": "status", "value": "active" }
-        5. 'database_query' { "query": "SELECT * FROM sales WHERE amount > 1000" }
+        Use ONLY these node types. Do not invent new ones.
+        
+        1. 'google_sheet_read'
+           - config: { "sheetId": "spreadsheet_id", "range": "A1:Z" }
+        2. 'google_sheet_write'
+           - config: { "sheetId": "spreadsheet_id", "range": "A1", "data": "{{parent_id.output}}", "writeMode": "append" }
+        3. 'google_doc_read'
+           - config: { "docId": "document_id" }
+        4. 'gmail_send'
+           - config: { "to": "email@address.com", "subject": "Subject", "body": "Message" }
+        5. 'slack_notify'
+           - config: { "url": "webhook_url", "message": "Message" }
+        6. 'discord_notify'
+           - config: { "url": "webhook_url", "message": "Message" }
+        7. 'filter'
+           - config: { "keyword": "status" }
+        8. 'invoice_ocr'
+           - config: { "fileUrl": "path/to/file" }
+        
+        [DATABASE ACTIONS]
+        To query or modify data, use the 'database_query' action.
+        
+        [OUTPUT FORMAT - WORKFLOW]
+        To build a flow:
+        ```json
+        {
+          "action": "create_workflow",
+          "name": "Workflow Name",
+          "payload": {
+            "nodes": [
+              {"id": "1", "type": "google_sheet_read", "config": {...}, "position": {"x": 0, "y": 0}}
+            ],
+            "edges": []
+          }
+        }
+        ```
         
-        [OUTPUT FORMAT]
-        JSON with 'nodes' and 'edges'.
+        [OUTPUT FORMAT - SQL]
+        To read/write data directly:
+        ```json
+        {
+          "action": "query_db",
+          "query": "INSERT INTO customers (name) VALUES ('New User')"
+        }
+        ```
         """