use name parameter

rafalposwiata · Apr 17, 2023 · 4dd1d58 · 4dd1d58
1 parent a5c6996
commit 4dd1d58
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/6_team.py b/6_team.py
@@ -7,7 +7,7 @@
 
 def create_embedding_store(name):
     chroma_client = chromadb.Client()
-    return chroma_client.create_collection("newspieces")
+    return chroma_client.create_collection(name)
 
 def query_pages(collection, urls, questions):
     docs = TrafilaturaWebReader().load_data(urls)