modernize type hints
This commit is contained in:
+3
-2
@@ -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 [
|
||||
{
|
||||
|
||||
Reference in New Issue
Block a user