import sys
import faiss
import numpy as np
import openai
import sqlite3

openai.api_key = "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"

# Load FAISS index
index = faiss.read_index("faiss_index.bin")
id_map = np.load("id_map.npy", allow_pickle=True).item()

# Convert query to vector
query = sys.argv[1]
query_vector = openai.Embedding.create(
    input=query,
    model="text-embedding-ada-002"
)["data"][0]["embedding"]
query_vector = np.array([query_vector]).astype("float32")

# Search FAISS
_, result_indices = index.search(query_vector, 1)
best_match_id = id_map[result_indices[0][0]]

print(best_match_id)