From c9cc67f6635dacec172b42cf08e6956d4785f41c Mon Sep 17 00:00:00 2001
From: Martino Ferrari <martinogiordano.ferrari@iter.org>
Date: Sat, 24 Jan 2026 15:33:23 +0100
Subject: [PATCH] Minimal changes

---
 internal/lsp/server.go    | 17 ++++++-----------
 internal/schema/marte.cue |  2 +-
 2 files changed, 7 insertions(+), 12 deletions(-)

diff --git a/internal/lsp/server.go b/internal/lsp/server.go
index 323cb55..5429ac8 100644
--- a/internal/lsp/server.go
+++ b/internal/lsp/server.go
@@ -20,8 +20,6 @@ import (
 	"cuelang.org/go/cue"
 )
 
-
-
 type CompletionParams struct {
 	TextDocument TextDocumentIdentifier `json:"textDocument"`
 	Position     Position               `json:"position"`
@@ -171,7 +169,6 @@ type TextEdit struct {
 	NewText string `json:"newText"`
 }
 
-
 func RunServer() {
 	reader := bufio.NewReader(os.Stdin)
 	for {
@@ -783,11 +780,11 @@ func suggestCUEEnums(container *index.ProjectNode, field string) *CompletionList
 		if err != nil {
 			continue
 		}
-        
-        // Ensure strings are quoted
-        if v.Kind() == cue.StringKind && !strings.HasPrefix(str, "\"") {
-            str = fmt.Sprintf("\"%s\"", str)
-        }
+
+		// Ensure strings are quoted
+		if v.Kind() == cue.StringKind && !strings.HasPrefix(str, "\"") {
+			str = fmt.Sprintf("\"%s\"", str)
+		}
 
 		items = append(items, CompletionItem{
 			Label:  str,
@@ -986,9 +983,7 @@ func formatNodeInfo(node *index.ProjectNode) string {
 		if typ != "" {
 			sigInfo += fmt.Sprintf("**Type**: `%s` ", typ)
 		}
-		if ds != "" {
-			sigInfo += fmt.Sprintf("**DataSource**: `%s` ", ds)
-		}
+		sigInfo += fmt.Sprintf("**DataSource**: `%s` ", ds)
 
 		// Size
 		dims := node.Metadata["NumberOfDimensions"]
diff --git a/internal/schema/marte.cue b/internal/schema/marte.cue
index 8220268..0504056 100644
--- a/internal/schema/marte.cue
+++ b/internal/schema/marte.cue
@@ -13,7 +13,7 @@ package schema
 	StateMachineEvent: {
 		NextState!:                                                  string
 		NextStateError!:                                             string
-		Timeout:                                                     uint32
+		Timeout?:                                                    uint32
 		[_= !~"^(Class|NextState|Timeout|NextStateError|[#_$].+)$"]: Message
 		...
 	}