import requests
import json
import random

HOST = "https://rag-app-fa66b3d8eb83.hosted.ghaymah.systems"
N_DIM = 512

def random_vector():
    """Generates a random 512-dimensional vector."""
    return [random.random() for _ in range(N_DIM)]

# Create a sample 512-dimensional vector and a dummy payload
vectors_to_insert = [random_vector()]
payloads = [{"test_data": "This is a test payload for 512-dim vector."}]
print(vectors_to_insert)
print(f"Attempting to send a {N_DIM}-dimensional vector to {HOST}/insert...")

try:
    insert_resp = requests.post(
        f"{HOST}/insert",
        json={"vectors": vectors_to_insert, "payloads": payloads},
        headers={"Content-Type": "application/json"}
    )
    insert_resp.raise_for_status() # Raise an exception for bad status codes
    print("Response Status Code:", insert_resp.status_code)
    print("Response Body:", insert_resp.text)
except requests.exceptions.RequestException as e:
    print(f"An error occurred: {e}")