From 511f8e660174e8397a98f1d9ef1104bf4e6bd75b Mon Sep 17 00:00:00 2001
From: Sammy Oina <sammyoina@gmail.com>
Date: Thu, 8 Aug 2024 14:51:03 +0300
Subject: [PATCH] maintain filename

Signed-off-by: Sammy Oina <sammyoina@gmail.com>
---
 agent/service.go                 |  2 +-
 hal/linux/package/agent/agent.mk |  4 ++--
 test/computations/main.go        | 11 +----------
 3 files changed, 4 insertions(+), 13 deletions(-)

diff --git a/agent/service.go b/agent/service.go
index c6d25935..de3d8d0d 100644
--- a/agent/service.go
+++ b/agent/service.go
@@ -194,7 +194,7 @@ func (as *agentService) Data(ctx context.Context, dataset Dataset) error {
 
 	as.computation.Datasets = slices.Delete(as.computation.Datasets, index, index+1)
 
-	f, err := os.Create(fmt.Sprintf("%s/dataset-%d", algorithm.DatasetsDir, index))
+	f, err := os.Create(fmt.Sprintf("%s/%s", algorithm.DatasetsDir, dataset.Filename))
 	if err != nil {
 		return fmt.Errorf("error creating dataset file: %v", err)
 	}
diff --git a/hal/linux/package/agent/agent.mk b/hal/linux/package/agent/agent.mk
index f5241b89..6257711d 100644
--- a/hal/linux/package/agent/agent.mk
+++ b/hal/linux/package/agent/agent.mk
@@ -4,8 +4,8 @@
 #
 ################################################################################
 
-AGENT_VERSION = filename
-AGENT_SITE = $(call github,sammyoina,cocos-ai,$(AGENT_VERSION))
+AGENT_VERSION = main
+AGENT_SITE = $(call github,ultravioletrs,cocos,$(AGENT_VERSION))
 
 define AGENT_BUILD_CMDS 	
 	$(MAKE) -C $(@D) agent
diff --git a/test/computations/main.go b/test/computations/main.go
index 0843a436..e3006451 100644
--- a/test/computations/main.go
+++ b/test/computations/main.go
@@ -69,20 +69,11 @@ func (s *svc) Run(ipAdress string, reqChan chan *manager.ServerStreamMessage, au
 			return
 		}
 		dataHash := sha3.Sum256(data)
-		datasets = append(datasets, &manager.Dataset{Hash: dataHash[:], UserKey: pubPem.Bytes, Filename: "iris1.csv"})
+		datasets = append(datasets, &manager.Dataset{Hash: dataHash[:], UserKey: pubPem.Bytes})
 	}
 
 	algoHash := sha3.Sum256(algo)
 
-	// Uncomment this to run tests on the manager service on a SEV enabled backend.
-	reqChan <- &manager.ServerStreamMessage{
-		Message: &manager.ServerStreamMessage_BackendInfoReq{
-			BackendInfoReq: &manager.BackendInfoReq{
-				Id: "1",
-			},
-		},
-	}
-
 	reqChan <- &manager.ServerStreamMessage{
 		Message: &manager.ServerStreamMessage_RunReq{
 			RunReq: &manager.ComputationRunReq{