cutecutecat cutecutecat

AI infra, Kubernetes, LLM

cutecutecat / citus_pgvectors.py

Last active September 5, 2024 14:07

Citus with pgvecto.rs

	import numpy as np
	from pgvecto_rs.psycopg import register_vector
	import psycopg

	# generate random data
	rows = 100000
	dimensions = 128
	embeddings = np.random.rand(rows, dimensions)
	categories = np.random.randint(100, size=rows).tolist()
	queries = np.random.rand(10, dimensions)