ultravioletrs · drasko · Sep 12, 2024 · Jul 8, 2024 · Aug 21, 2024 · Aug 28, 2024
diff --git a/agent/algorithm/logging.go b/agent/algorithm/logging.go
@@ -4,6 +4,7 @@ package algorithm
 
 import (
 	"bytes"
+	"encoding/json"
 	"io"
 	"log/slog"
 
@@ -65,7 +66,7 @@ func (s *Stderr) Write(p []byte) (n int, err error) {
 		s.Logger.Error(string(buf[:n]))
 	}
 
-	if err := s.EventSvc.SendEvent("algorithm-run", "error", nil); err != nil {
+	if err := s.EventSvc.SendEvent("algorithm-run", "error", json.RawMessage{}); err != nil {
 		return len(p), err
 	}
 

diff --git a/agent/algorithm/python/python.go b/agent/algorithm/python/python.go
@@ -18,15 +18,15 @@ import (
 
 const (
 	PyRuntime    = "python3"
-	pyRuntimeKey = "python_runtime"
+	PyRuntimeKey = "python_runtime"
 )
 
 func PythonRunTimeToContext(ctx context.Context, runtime string) context.Context {
-	return metadata.AppendToOutgoingContext(ctx, pyRuntimeKey, runtime)
+	return metadata.AppendToOutgoingContext(ctx, PyRuntimeKey, runtime)
 }
 
 func PythonRunTimeFromContext(ctx context.Context) string {
-	return metadata.ValueFromIncomingContext(ctx, pyRuntimeKey)[0]
+	return metadata.ValueFromIncomingContext(ctx, PyRuntimeKey)[0]
 }
 
 var _ algorithm.Algorithm = (*python)(nil)

diff --git a/agent/quoteprovider/mocks/QuoteProvider.go b/agent/quoteprovider/mocks/QuoteProvider.go
diff --git a/agent/service.go b/agent/service.go
@@ -55,6 +55,8 @@ var (
 	ErrFileNameMismatch = errors.New("malformed data, filename does not match manifest")
 	// ErrAllResultsConsumed indicates all results have been consumed.
 	ErrAllResultsConsumed = errors.New("all results have been consumed by declared consumers")
+	// ErrAttestationFailed attestation failed.
+	ErrAttestationFailed = errors.New("failed to get raw quote")
 )
 
 // Service specifies an API that must be fullfiled by the domain service
@@ -124,7 +126,7 @@ func (as *agentService) Algo(ctx context.Context, algo Algorithm) error {
 		return fmt.Errorf("error getting current directory: %v", err)
 	}
 
-	f, err := os.Create(filepath.Join(currentDir, "algorithm"))
+	f, err := os.Create(filepath.Join(currentDir, "algo"))
 	if err != nil {
 		return fmt.Errorf("error creating algorithm file: %v", err)
 	}
@@ -317,8 +319,9 @@ func (as *agentService) runComputation() {
 }
 
 func (as *agentService) publishEvent(status string, details json.RawMessage) func() {
+	st := as.sm.GetState().String()
 	return func() {
-		if err := as.eventSvc.SendEvent(as.sm.State.String(), status, details); err != nil {
+		if err := as.eventSvc.SendEvent(st, status, details); err != nil {
 			as.sm.logger.Warn(err.Error())
 		}
 	}