Merge branch 'main' into willtai/hybrid-retriever

neo4j · Apr 26, 2024 · 126aa48 · 126aa48
2 parents 267fb72 + ec557f2
commit 126aa48
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 7 deletions.
diff --git a/src/neo4j_genai/embedder.py b/src/neo4j_genai/embedder.py
@@ -14,19 +14,18 @@
 #  limitations under the License.
 
 from abc import ABC, abstractmethod
-from .types import EmbeddingVector
 
 
 class Embedder(ABC):
     """Interface for embedding models."""
 
     @abstractmethod
-    def embed_query(self, text: str) -> EmbeddingVector:
+    def embed_query(self, text: str) -> list[float]:
         """Embed query text.
 
         Args:
             text (str): Text to convert to vector embedding
 
         Returns:
-            EmbeddingVector: A vector embedding.
+            list[float]: A vector embedding.
         """
diff --git a/src/neo4j_genai/types.py b/src/neo4j_genai/types.py
@@ -23,10 +23,6 @@ class VectorSearchRecord(BaseModel):
     score: float
 
 
-class EmbeddingVector(BaseModel):
-    vector: list[float]
-
-
 class IndexModel(BaseModel):
     driver: Any