fix automated release workflow

Author: gurgah

internal/docker/client.go

@@ -4,6 +4,7 @@ package docker
 import (
 	"context"
 	"fmt"
+	"github.com/docker/docker/api/types/container"
 	"io"
 	"os"
 
@@ -258,3 +259,11 @@ func fileExists(path string) bool {
 	_, err := os.Stat(path)
 	return err == nil
 }
+func (c *Client) GetDockerClient() *client.Client {
+	return c.docker
+}
+
+// ContainerWait waits for a container to reach a certain condition
+func (c *Client) ContainerWait(ctx context.Context, containerID string, condition container.WaitCondition) (<-chan container.WaitResponse, <-chan error) {
+	return c.docker.ContainerWait(ctx, containerID, condition)
+}

internal/orchestrator/orchestrator.go

@@ -5,15 +5,13 @@ import (
 	"context"
 	"encoding/json"
 	"fmt"
-	"os"
 	"sync"
 	"syscall"
 	"time"
 
 	"github.com/docker/docker/api/types"
 	"github.com/docker/docker/client"
 	"github.com/localcloud-sh/localcloud/internal/config"
-	"github.com/localcloud-sh/localcloud/internal/docker"
 )
 
 // HealthMonitor monitors service health and triggers recovery

internal/recovery/health_monitor.go

@@ -266,7 +266,9 @@ func (sr *ServiceRestarter) monitorLoop(ctx context.Context, containerID, servic
 			return
 		default:
 			// Wait for container to exit
-			statusCh, errCh := sr.manager.GetClient().docker.ContainerWait(ctx, containerID, "not-running")
+			client := sr.manager.GetClient()
+			dockerClient := client.GetDockerClient()
+			statusCh, errCh := dockerClient.ContainerWait(ctx, containerID, "not-running")
 
 			select {
 			case err := <-errCh:

internal/recovery/restart_policy.go

@@ -14,6 +14,10 @@ type Client struct {
 	service *Service
 }
 
+func (c *Client) QueryRow(query string, args ...interface{}) *sql.Row {
+	return c.service.db.QueryRow(query, args...)
+}
+
 // NewClient creates a new database client
 func NewClient(service *Service) *Client {
 	return &Client{service: service}

internal/services/postgres/client.go

@@ -3,7 +3,6 @@ package pgvector
 
 import (
 	"context"
-	"database/sql"
 	"encoding/json"
 	"fmt"
 	"strings"
@@ -264,15 +263,15 @@ func (db *PgVectorDB) GetStats(ctx context.Context) (vectordb.Stats, error) {
 	stats.LastUpdated = time.Now()
 
 	// Count total documents
-	err := db.client.Query(
+	err := db.client.QueryRow(
 		"SELECT COUNT(DISTINCT document_id) FROM localcloud.embeddings",
 	).Scan(&stats.TotalDocuments)
 	if err != nil {
 		return stats, err
 	}
 
 	// Count total vectors
-	err = db.client.Query(
+	err = db.client.QueryRow(
 		"SELECT COUNT(*) FROM localcloud.embeddings",
 	).Scan(&stats.TotalVectors)
 	if err != nil {
@@ -301,7 +300,7 @@ func (db *PgVectorDB) ensureTables() error {
 	// This is already handled in postgres.go initialization
 	// Just verify the table exists
 	var exists bool
-	err := db.client.Query(`
+	err := db.client.QueryRow(`
 		SELECT EXISTS (
 			SELECT FROM information_schema.tables 
 			WHERE table_schema = 'localcloud'