openai_query.py

import os
import openai

embedding  = openai.Embedding.create(input="test query", model="text-embedding-ada-002").data[0].embedding
pipeline = [
			{
	        "$vectorSearch": {
	            "index":'bson_vector_index',
	            "path": "plot_embedding",
	            "queryVector": embedding,
	            "limit": 10,
	            "numCandidates": 100,
	        },
	        },
	        {
    		"$project": {"plot_embedding": 0}
	        }
	    ]

docs = list(coll.aggregate(pipeline))