From 0e42c9002b73264eed559b3ff51fdbf0df16e0bc Mon Sep 17 00:00:00 2001 From: John Costa Date: Sun, 5 Oct 2025 16:25:00 +0100 Subject: [PATCH] fix: minor bugs --- backend/agents/create_list_agent.go | 9 ++++++--- backend/schema.sql | 4 +++- 2 files changed, 9 insertions(+), 4 deletions(-) diff --git a/backend/agents/create_list_agent.go b/backend/agents/create_list_agent.go index 6abac40..1e87aad 100644 --- a/backend/agents/create_list_agent.go +++ b/backend/agents/create_list_agent.go @@ -7,6 +7,7 @@ import ( "screenmark/screenmark/.gen/haystack/haystack/model" "screenmark/screenmark/agents/client" "screenmark/screenmark/models" + "strings" "github.com/charmbracelet/log" "github.com/google/uuid" @@ -109,12 +110,14 @@ func (agent *CreateListAgent) CreateList(log *log.Logger, userID uuid.UUID, stac content := resp.Choices[0].Message.Content - structuredOutput := content[len("```json") : len(content)-3] + if strings.HasPrefix(content, "```json") { + content = content[len("```json") : len(content)-3] + } - log.Info("", "res", structuredOutput) + log.Info("", "res", content) var createListArgs createNewListArguments - err = json.Unmarshal([]byte(structuredOutput), &createListArgs) + err = json.Unmarshal([]byte(content), &createListArgs) if err != nil { return err } diff --git a/backend/schema.sql b/backend/schema.sql index 587f2ba..96b28ab 100644 --- a/backend/schema.sql +++ b/backend/schema.sql @@ -45,7 +45,9 @@ CREATE TABLE haystack.image_stacks ( id UUID PRIMARY KEY DEFAULT gen_random_uuid(), image_id UUID NOT NULL REFERENCES haystack.image (id) ON DELETE CASCADE, - stack_id UUID NOT NULL REFERENCES haystack.stacks (id) ON DELETE CASCADE + stack_id UUID NOT NULL REFERENCES haystack.stacks (id) ON DELETE CASCADE, + + UNIQUE(image_id, stack_id) ); CREATE TABLE haystack.schema_items (