Uppdatera och ta bort objekt i en samling

Att dokumenten lagrats i en vektordatabas betyder inte att du är låst – du kan fortfarande lägga till nya objekt eller uppdatera befintliga.

I den här övningen har du fått två nya Netflix-titlar lagrade i new_data:

[{"id": "s1001", "document": "Title: Cats & Dogs (Movie)\nDescription: A look at the top-secret, high-tech espionage war going on between cats and dogs, of which their human owners are blissfully unaware."},
 {"id": "s6884", "document": 'Title: Goosebumps 2: Haunted Halloween (Movie)\nDescription: Three teens spend their Halloween trying to stop a magical book, which brings characters from the "Goosebumps" novels to life.\nCategories: Children & Family Movies, Comedies'}]

Beroende på om ID:na redan finns i samlingen kommer du antingen att lägga till eller uppdatera dem i databasen.

Den här övningen är en del av kursen

Introduktion till inbäddningar med OpenAI API

Visa kurs

Övningsinstruktioner

Extrahera ID:n och dokument från new_data, och använd en enda metod för att uppdatera dem i samlingen netflix_titles om de redan finns, eller lägga till dem om de inte gör det.
Ta bort objektet med ID:t 's95' när du har lagt till eller uppdaterat objekten.

Interaktiv övning med praktiskt arbete

Testa den här övningen genom att slutföra den här exempelkoden.

collection = client.get_collection(
  name="netflix_titles",
  embedding_function=OpenAIEmbeddingFunction(model_name="text-embedding-3-small", api_key="")
)

# Update or add the new documents
____(
    ids=____,
    documents=____
)

# Delete the item with ID "s95"
____

result = collection.query(
    query_texts=["films about dogs"],
    n_results=3
)
print(result)

Redigera och kör kod