modernize type hints

This commit is contained in:
Matteo Rosati
2026-04-22 17:19:14 +02:00
parent 33b46c2c21
commit 2dfaa68466
8 changed files with 23 additions and 9 deletions
+3 -2
View File
@@ -1,5 +1,6 @@
from __future__ import annotations
from collections.abc import Sequence
from typing import TypedDict
from chromadb.utils.embedding_functions import DefaultEmbeddingFunction
@@ -10,12 +11,12 @@ class EmbeddingRecord(TypedDict):
embedding: list[float]
def embed(chunks: list[str]) -> list[EmbeddingRecord]:
def embed(chunks: Sequence[str]) -> list[EmbeddingRecord]:
if not chunks:
return []
embedding_function = DefaultEmbeddingFunction()
embeddings = embedding_function(chunks)
embeddings = embedding_function(list(chunks))
return [
{