From 80dcfea4207c6d36f58565a37d7aec745ebf4f93 Mon Sep 17 00:00:00 2001
From: Leonardo Christino <leomilho@gmail.com>
Date: Tue, 29 Aug 2023 17:52:38 +0200
Subject: [PATCH] Re-enable machine learning backend

---
 entityv2/queryStruct.go | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/entityv2/queryStruct.go b/entityv2/queryStruct.go
index 6bbaeab..a0a691a 100644
--- a/entityv2/queryStruct.go
+++ b/entityv2/queryStruct.go
@@ -17,11 +17,19 @@ type IncomingQueryJSON struct {
 	// Entities        []QueryEntityStruct   `json:"entities"`
 	// Relations       []QueryRelationStruct `json:"relations"`
 	// GroupBys        []QueryGroupByStruct  `json:"groupBys"`
-	// MachineLearning []QueryMLStruct       `json:"machineLearning"`
+	MachineLearning []QueryMLStruct       `json:"machineLearning"`
 	// Modifiers []QueryModifierStruct
 	// Prefix    string
 }
 
+/*
+QueryMLStruct holds info for machinelearning
+*/
+type QueryMLStruct struct {
+	Type       string
+	Parameters []string
+}
+
 type QueryStruct struct {
 	ID        string     `json:"id"`
 	Node      NodeStruct `json:"node"`
-- 
GitLab