wip: agent model

Author: aktky

backend/devtools/create_migrate.sh

@@ -0,0 +1,29 @@
+package core
+
+import (
+	"time"
+
+	"github.com/gofrs/uuid/v5"
+)
+
+type Agent struct {
+	ID             uuid.UUID `db:"id"`
+	OrganizationID uuid.UUID `db:"organization_id"`
+	EnvironmentID  uuid.UUID `db:"environment_id"`
+	APIKeyID       uuid.UUID `db:"api_key_id"`
+	Name           string    `db:"name"`
+	Description    string    `db:"description"`
+	Instructions   string    `db:"instructions"`
+	Model          string    `db:"model"`
+	CreatedAt      time.Time `db:"created_at"`
+	UpdatedAt      time.Time `db:"updated_at"`
+}
+
+type AgentTool struct {
+	ID          uuid.UUID `db:"id"`
+	AgentID     uuid.UUID `db:"agent_id"`
+	Name        string    `db:"name"`
+	Description string    `db:"description"`
+	CreatedAt   time.Time `db:"created_at"`
+	UpdatedAt   time.Time `db:"updated_at"`
+}

backend/internal/core/agent.go

@@ -0,0 +1,17 @@
+BEGIN;
+
+-- drop agent_tool
+DROP TRIGGER IF EXISTS update_agent_tool_updated_at ON "agent_tool";
+DROP INDEX IF EXISTS idx_agent_tool_agent_name;
+DROP TABLE IF EXISTS "agent_tool";
+
+-- drop agent
+DROP TRIGGER IF EXISTS validate_agent ON "agent";
+DROP TRIGGER IF EXISTS update_agent_updated_at ON "agent";
+DROP INDEX IF EXISTS idx_agent_organization_api_key_name;
+DROP TABLE IF EXISTS "agent";
+
+-- drop function
+DROP FUNCTION IF EXISTS validate_agent();
+
+END;

backend/migrations/000002_create_table_agent.down.sql

@@ -0,0 +1,80 @@
+BEGIN;
+
+-- agent table
+
+CREATE OR REPLACE FUNCTION validate_agent()
+RETURNS TRIGGER AS $$
+DECLARE
+    environment_org_id UUID;
+    api_key_org_id UUID;
+BEGIN
+    SELECT organization_id INTO environment_org_id
+    FROM "environment"
+    WHERE id = NEW.environment_id;
+
+    IF environment_org_id != NEW.organization_id THEN
+        RAISE EXCEPTION 'Environment % must belong to organization % to create an agent', NEW.environment_id, NEW.organization_id;
+    END IF;
+
+    SELECT organization_id INTO api_key_org_id
+    FROM "api_key"
+    WHERE id = NEW.api_key_id;
+
+    IF api_key_org_id != NEW.organization_id THEN
+        RAISE EXCEPTION 'API key % must belong to organization % to create an agent', NEW.api_key_id, NEW.organization_id;
+    END IF;
+
+    RETURN NEW;
+END;
+$$ LANGUAGE plpgsql;
+
+CREATE TABLE "agent" (
+  "id"              UUID          NOT NULL,
+  "organization_id" UUID          NOT NULL,
+  "environment_id"  UUID          NOT NULL,
+  "api_key_id"      UUID          NOT NULL,
+  "name"            VARCHAR(255)  NOT NULL,
+  "description"     TEXT          NOT NULL DEFAULT '',
+  "instructions"    TEXT          NOT NULL DEFAULT '',
+  "model"           VARCHAR(255)  NOT NULL,
+  "created_at"      TIMESTAMPTZ   NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  "updated_at"      TIMESTAMPTZ   NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  FOREIGN KEY ("organization_id") REFERENCES "organization"("id") ON DELETE CASCADE,
+  FOREIGN KEY ("environment_id")  REFERENCES "environment"("id") ON DELETE CASCADE,
+  FOREIGN KEY ("api_key_id")      REFERENCES "api_key"("id") ON DELETE CASCADE,
+  PRIMARY KEY ("id")
+);
+
+CREATE UNIQUE INDEX idx_agent_organization_api_key_name ON "agent" ("organization_id", "api_key_id", "name");
+
+CREATE TRIGGER update_agent_updated_at
+    BEFORE UPDATE ON "agent"
+    FOR EACH ROW
+    EXECUTE FUNCTION update_updated_at_column();
+
+CREATE TRIGGER validate_agent
+    BEFORE INSERT OR UPDATE ON "agent"
+    FOR EACH ROW
+    EXECUTE FUNCTION validate_agent();
+
+-- agent_tool table
+
+CREATE TABLE "agent_tool" (
+  "id"          UUID          NOT NULL,
+  "agent_id"    UUID          NOT NULL,
+  "name"        VARCHAR(255)  NOT NULL,
+  "description" TEXT          NOT NULL DEFAULT '',
+  "created_at"  TIMESTAMPTZ   NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  "updated_at"  TIMESTAMPTZ   NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  FOREIGN KEY ("agent_id") REFERENCES "agent"("id") ON DELETE CASCADE,
+  PRIMARY KEY ("id")
+);
+
+CREATE UNIQUE INDEX idx_agent_tool_agent_name ON "agent_tool" ("agent_id", "name");
+
+CREATE TRIGGER update_agent_tool_updated_at
+    BEFORE UPDATE ON "agent_tool"
+    FOR EACH ROW
+    EXECUTE FUNCTION update_updated_at_column();
+
+END;