fix syntax and types

This commit is contained in:
2026-04-24 18:23:02 +02:00
parent 948f8500be
commit c5b6b196b5
8 changed files with 21 additions and 25 deletions
+4 -3
View File
@@ -58,7 +58,9 @@ def count_collection(collection_name: str) -> str:
_, collection = _get_client_and_collection(collection_name)
count = collection.count()
return f"The '{collection_name}' collection contains [bold green]{count}[/] records."
return (
f"The '{collection_name}' collection contains [bold green]{count}[/] records."
)
def add_data(
@@ -71,8 +73,7 @@ def add_data(
_, collection = _get_client_and_collection(collection_name)
embeddings: list[Sequence[float]] = [record["embedding"]
for record in data]
embeddings: list[Sequence[float]] = [record["embedding"] for record in data]
collection.add(
ids=[str(uuid4()) for _ in data],