Додавання даних до колекції

Час додати ці фільми й серіали Netflix до вашої колекції! Вам надано список ID документів і текстів, збережених у змінних ids і documents відповідно. Їх витягнуто з netflix_titles.csv за допомогою такого коду:

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)

Як приклад того, яку інформацію буде вкладено у вектори, ось перший документ із 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

Усі потрібні функції та пакети вже імпортовано, а також створено постійний клієнт і присвоєно його змінній client.

Ця вправа є частиною курсу

Вступ до Embeddings з OpenAI API

Переглянути курс

Інструкції до вправи

Відтворіть вашу колекцію netflix_titles.
Додайте документи та їхні ID до колекції.
Виведіть кількість документів у collection і перші десять елементів.

Інтерактивна практична вправа

Спробуйте виконати цю вправу, доповнивши цей зразок коду.

# 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: {____}")

Редагувати та запускати код