Fix database schema mismatch causing 500 errors - correct INSERT field mappings

backend/database/__init__.py

@@ -144,21 +144,19 @@ def add_sample_data_for_hf():
                      '{"scenario": "customer_service", "agents": ["RouterAgent", "OrderAgent", "CompensationAgent", "SupervisorAgent"]}')
                 )
                 
-                # Insert knowledge graph
+                # Insert knowledge graph with correct field names
                 conn.execute(
-                    """INSERT INTO knowledge_graphs (filename, creator, entity_count, relation_count, namespace,
-                       system_name, system_summary, status, trace_id, window_index, window_total, processing_run_id)
-                       VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)""",
-                    ("demo_kg.json", "AgentGraph Demo", 5, 4, "demo", "Customer Service System",
-                     "Multi-agent customer service system with error handling and coordination", 
-                     "completed", trace_id, 0, 1, "demo_run")
+                    """INSERT INTO knowledge_graphs (filename, entity_count, relation_count,
+                       status, trace_id, window_index, window_total, processing_run_id)
+                       VALUES (?, ?, ?, ?, ?, ?, ?, ?)""",
+                    ("demo_kg.json", 5, 4, "completed", trace_id, 0, 1, "demo_run")
                 )
                 
                 # Get KG ID
                 kg_result = conn.execute("SELECT id FROM knowledge_graphs WHERE trace_id = ?", (trace_id,))
                 kg_id = kg_result.fetchone()[0]
                 
-                # Insert sample entities
+                # Insert sample entities with correct field names
                 entities = [
                     ("agent_1", "agent", "RouterAgent", '{"role": "routing", "priority_handling": true}'),
                     ("agent_2", "agent", "OrderAgent", '{"role": "order_tracking", "data_sources": ["db", "api"]}'),
@@ -167,14 +165,19 @@ def add_sample_data_for_hf():
                     ("issue_1", "issue", "PaymentSystemFailure", '{"severity": "high", "impact": "service_disruption"}')
                 ]
                 
+                # Insert entities and store their database IDs
+                entity_db_ids = {}
                 for entity_id, entity_type, name, properties in entities:
                     conn.execute(
-                        """INSERT INTO entities (graph_id, entity_id, type, name, properties, knowledge_graph_namespace)
-                           VALUES (?, ?, ?, ?, ?, ?)""",
-                        (kg_id, entity_id, entity_type, name, properties, "demo")
+                        """INSERT INTO entities (graph_id, entity_id, type, name, properties)
+                           VALUES (?, ?, ?, ?, ?)""",
+                        (kg_id, entity_id, entity_type, name, properties)
                     )
+                    # Get the database ID for this entity
+                    result = conn.execute("SELECT id FROM entities WHERE graph_id = ? AND entity_id = ?", (kg_id, entity_id))
+                    entity_db_ids[entity_id] = result.fetchone()[0]
                 
-                # Insert sample relations
+                # Insert sample relations using database IDs as foreign keys
                 relations = [
                     ("rel_1", "agent_1", "routes_to", "agent_2", '{"priority": "high", "success": true}'),
                     ("rel_2", "agent_2", "escalates_to", "agent_3", '{"reason": "compensation_needed"}'),
@@ -183,10 +186,12 @@ def add_sample_data_for_hf():
                 ]
                 
                 for relation_id, from_entity, relation_type, to_entity, properties in relations:
+                    source_db_id = entity_db_ids[from_entity]
+                    target_db_id = entity_db_ids[to_entity]
                     conn.execute(
-                        """INSERT INTO relations (graph_id, relation_id, from_entity_id, relation_type, to_entity_id, 
-                           properties, knowledge_graph_namespace) VALUES (?, ?, ?, ?, ?, ?, ?)""",
-                        (kg_id, relation_id, from_entity, relation_type, to_entity, properties, "demo")
+                        """INSERT INTO relations (graph_id, relation_id, type, source_id, target_id, properties) 
+                           VALUES (?, ?, ?, ?, ?, ?)""",
+                        (kg_id, relation_id, relation_type, source_db_id, target_db_id, properties)
                     )
                 
                 # Commit transaction