test_script/import_data_to_rag.py

import chromadb

chroma_client = chromadb.Client()

collection = chroma_client.create_collection(name="SpotifyLyrics")

collection.add(
    documents=[
        "This is a document about pineapple",
        "This is a document about oranges",
        "This is a query document about hawaii"
    ],
    ids=["id1", "id2", "id3"],
)
query_texts=["fruit"]

results = collection.query(
    query_texts=query_texts, 
    n_results=2 
)
print(results)