brick - add require on list and details

Author: giulio93

internal/orchestrator/bricks/bricks.go

@@ -58,12 +58,16 @@ func (s *Service) List() (BrickListResult, error) {
 	res := BrickListResult{Bricks: make([]BrickListItem, len(s.bricksIndex.Bricks))}
 	for i, brick := range s.bricksIndex.Bricks {
 		res.Bricks[i] = BrickListItem{
-			ID:          brick.ID,
-			Name:        brick.Name,
-			Author:      "Arduino", // TODO: for now we only support our bricks
-			Description: brick.Description,
-			Category:    brick.Category,
-			Status:      "installed",
+			ID:            brick.ID,
+			Name:          brick.Name,
+			Author:        "Arduino", // TODO: for now we only support our bricks
+			Description:   brick.Description,
+			Category:      brick.Category,
+			Status:        "installed",
+			ModelRequired: brick.ModelRequired,
+			Models: f.Map(s.modelsIndex.GetModelsByBrick(brick.ID), func(m modelsindex.AIModel) string {
+				return m.ID
+			}),
 		}
 	}
 	return res, nil
@@ -198,17 +202,18 @@ func (s *Service) BricksDetails(id string, idProvider *app.IDProvider,
 		return BrickDetailsResult{}, fmt.Errorf("unable to get used by apps: %w", err)
 	}
 	return BrickDetailsResult{
-		ID:           id,
-		Name:         brick.Name,
-		Author:       "Arduino", // TODO: for now we only support our bricks
-		Description:  brick.Description,
-		Category:     brick.Category,
-		Status:       "installed", // For now every Arduino brick are installed
-		Variables:    variables,
-		Readme:       readme,
-		ApiDocsPath:  apiDocsPath,
-		CodeExamples: codeExamples,
-		UsedByApps:   usedByApps,
+		ID:            id,
+		Name:          brick.Name,
+		Author:        "Arduino", // TODO: for now we only support our bricks
+		Description:   brick.Description,
+		Category:      brick.Category,
+		ModelRequired: brick.ModelRequired,
+		Status:        "installed", // For now every Arduino brick are installed
+		Variables:     variables,
+		Readme:        readme,
+		ApiDocsPath:   apiDocsPath,
+		CodeExamples:  codeExamples,
+		UsedByApps:    usedByApps,
 		CompatibleModels: f.Map(s.modelsIndex.GetModelsByBrick(brick.ID), func(m modelsindex.AIModel) AIModel {
 			return AIModel{
 				ID:          m.ID,

internal/orchestrator/bricks/types.go

@@ -20,12 +20,14 @@ type BrickListResult struct {
 }
 
 type BrickListItem struct {
-	ID          string `json:"id"`
-	Name        string `json:"name"`
-	Author      string `json:"author"`
-	Description string `json:"description"`
-	Category    string `json:"category"`
-	Status      string `json:"status"`
+	ID            string   `json:"id"`
+	Name          string   `json:"name"`
+	Author        string   `json:"author"`
+	Description   string   `json:"description"`
+	Category      string   `json:"category"`
+	Status        string   `json:"status"`
+	ModelRequired bool     `json:"model_required"`
+	Models        []string `json:"models"`
 }
 
 type AppBrickInstancesResult struct {
@@ -78,6 +80,7 @@ type BrickDetailsResult struct {
 	Description      string                   `json:"description"`
 	Category         string                   `json:"category"`
 	Status           string                   `json:"status"`
+	ModelRequired    bool                     `json:"model_required"`
 	Variables        map[string]BrickVariable `json:"variables,omitempty"`
 	Readme           string                   `json:"readme"`
 	ApiDocsPath      string                   `json:"api_docs_path"`

internal/orchestrator/bricksindex/bricks_index.go

@@ -58,6 +58,7 @@ type Brick struct {
 	RequireModel              bool            `yaml:"require_model"`
 	Variables                 []BrickVariable `yaml:"variables,omitempty"`
 	Ports                     []string        `yaml:"ports,omitempty"`
+	ModelRequired             bool            `yaml:"model_required,omitempty"`
 	ModelName                 string          `yaml:"model_name,omitempty"`
 	MountDevicesIntoContainer bool            `yaml:"mount_devices_into_container,omitempty"`
 	RequiredDevices           []string        `yaml:"required_devices,omitempty"`