Dodawanie danych do kolekcji
Czas dodać filmy i seriale Netfliksa do kolekcji! Masz do dyspozycji listę identyfikatorów dokumentów i tekstów, zapisanych odpowiednio w zmiennych ids i documents, które zostały wyodrębnione z pliku netflix_titles.csv za pomocą następującego kodu:
ids = []
documents = []
with open('netflix_titles.csv') as csvfile:
reader = csv.DictReader(csvfile)
for i, row in enumerate(reader):
ids.append(row['show_id'])
text = f"Title: {row['title']} ({row['type']})\nDescription: {row['description']}\nCategories: {row['listed_in']}"
documents.append(text)
Aby zobaczyć, jakie informacje zostaną osadzone, oto pierwszy dokument ze zmiennej documents:
Title: Dick Johnson Is Dead (Movie)
Description: As her father nears the end of his life, filmmaker Kirsten Johnson stages his death in inventive and comical ways to help them both face the inevitable.
Categories: Documentaries
Wszystkie niezbędne funkcje i pakiety zostały już zaimportowane, a trwały klient został utworzony i przypisany do zmiennej client.
To ćwiczenie jest częścią kursu
Wprowadzenie do osadzeń z OpenAI API
Instrukcje do ćwiczenia
- Utwórz ponownie kolekcję
netflix_titles. - Dodaj dokumenty wraz z ich identyfikatorami do kolekcji.
- Wyświetl liczbę dokumentów w zmiennej
collectionoraz pierwsze dziesięć elementów.
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
# Recreate the netflix_titles collection
collection = client.____(
name="netflix_titles",
embedding_function=OpenAIEmbeddingFunction(model_name="text-embedding-3-small", api_key="")
)
# Add the documents and IDs to the collection
____
# Print the collection size and first ten items
print(f"No. of documents: {____}")
print(f"First ten documents: {____}")