mirror of
https://github.com/imartinez/privateGPT.git
synced 2025-06-24 22:42:18 +00:00
a579c9bdc5
* Update the version of llama_index used to fix transient openai errors * Update poetry.lock file * Make `local` mode the default mode by default
31 lines
924 B
Python
31 lines
924 B
Python
from typing import Literal
|
|
|
|
from injector import inject, singleton
|
|
from pydantic import BaseModel, Field
|
|
|
|
from private_gpt.components.embedding.embedding_component import EmbeddingComponent
|
|
|
|
|
|
class Embedding(BaseModel):
|
|
index: int
|
|
object: Literal["embedding"]
|
|
embedding: list[float] = Field(examples=[[0.0023064255, -0.009327292]])
|
|
|
|
|
|
@singleton
|
|
class EmbeddingsService:
|
|
@inject
|
|
def __init__(self, embedding_component: EmbeddingComponent) -> None:
|
|
self.embedding_model = embedding_component.embedding_model
|
|
|
|
def texts_embeddings(self, texts: list[str]) -> list[Embedding]:
|
|
texts_embeddings = self.embedding_model.get_text_embedding_batch(texts)
|
|
return [
|
|
Embedding(
|
|
index=texts_embeddings.index(embedding),
|
|
object="embedding",
|
|
embedding=embedding,
|
|
)
|
|
for embedding in texts_embeddings
|
|
]
|