deepseek/deepseek-v4-pro · d2_pgvector · rep 2

echo "URL: $SUPABASE_URL" && echo "Ref: $SUPABASE_PROJECT_REF" && echo "DB URL: $SUPABASE_DB_URL"

URL: https://<ref>.supabase.co
Ref: <ref>
DB URL: <redacted:SUPABASE_DB_URL>

PGPASSWORD="$SUPABASE_DB_PASSWORD" psql "$SUPABASE_DB_URL" -c "SELECT * FROM public.documents;" 2>&1

 id |  title  |  embedding_json  
----+---------+------------------
  1 | alpha   | [10, 0.5, 0, 0]
  2 | bravo   | [0.9, 0.3, 0, 0]
  3 | delta   | [20, 60, 0, 0]
  4 | echo    | [0, 1, 0, 0]
  5 | foxtrot | [-5, 0.1, 0, 0]
  6 | golf    | [0.1, 0.1, 3, 4]
(6 rows)

PGPASSWORD="$SUPABASE_DB_PASSWORD" psql "$SUPABASE_DB_URL" -c "CREATE EXTENSION IF NOT EXISTS vector;" 2>&1

CREATE EXTENSION

PGPASSWORD="$SUPABASE_DB_PASSWORD" psql "$SUPABASE_DB_URL" -c "ALTER TABLE public.documents ADD COLUMN IF NOT EXISTS embedding vector(4);" 2>&1

ALTER TABLE

PGPASSWORD="$SUPABASE_DB_PASSWORD" psql "$SUPABASE_DB_URL" -c "UPDATE public.documents SET embedding = embedding_json::text::vector;" 2>&1

UPDATE 6

PGPASSWORD="$SUPABASE_DB_PASSWORD" psql "$SUPABASE_DB_URL" -c "CREATE INDEX IF NOT EXISTS documents_embedding_hnsw_idx ON public.documents USING hnsw (embedding vector_cosine_ops);" 2>&1

CREATE INDEX

PGPASSWORD="$SUPABASE_DB_PASSWORD" psql "$SUPABASE_DB_URL" -c "CREATE OR REPLACE FUNCTION public.match_documents(query_embedding vector(4), match_count int)
RETURNS TABLE (id int, title text, distance float8)
LANGUAGE sql STABLE
AS \$\$
  SELECT d.id, d.title, d.embedding <=> query_embedding AS distance
  FROM public.documents d
  ORDER BY d.embedding <=> query_embedding
  LIMIT match_count;
\$\$;" 2>&1

CREATE FUNCTION

PGPASSWORD="$SUPABASE_DB_PASSWORD" psql "$SUPABASE_DB_URL" -c "SELECT * FROM public.match_documents('[10, 0.5, 0, 0]'::vector, 3);" 2>&1

 id | title |      distance      
----+-------+--------------------
  1 | alpha |                  0
  2 | bravo | 0.0367087035723488
  3 | delta |  0.636791795933452
(3 rows)

curl -s -X POST "$SUPABASE_URL/rest/v1/rpc/match_documents" \
  -H "apikey: $SUPABASE_ANON_KEY" \
  -H "Authorization: Bearer $SUPABASE_ANON_KEY" \
  -H "Content-Type: application/json" \
  -d '{"query_embedding": [10, 0.5, 0, 0], "match_count": 3}' | python3 -m json.tool

[
    {
        "id": 1,
        "title": "alpha",
        "distance": 0
    },
    {
        "id": 2,
        "title": "bravo",
        "distance": 0.0367087035723488
    },
    {
        "id": 3,
        "title": "delta",
        "distance": 0.636791795933452
    }
]

curl -s -X POST "$SUPABASE_URL/rest/v1/rpc/match_documents" \
  -H "apikey: $SUPABASE_ANON_KEY" \
  -H "Authorization: Bearer $SUPABASE_ANON_KEY" \
  -H "Content-Type: application/json" \
  -d '{"query_embedding": [0, 1, 0, 0], "match_count": 3}' | python3 -m json.tool

[
    {
        "id": 4,
        "title": "echo",
        "distance": 0
    },
    {
        "id": 3,
        "title": "delta",
        "distance": 0.0513167019494862
    },
    {
        "id": 2,
        "title": "bravo",
        "distance": 0.683772217228812
    }
]