18 lines
498 B
Python
18 lines
498 B
Python
|
from typing import List
|
||
|
|
import numpy as np
|
||
|
|
from pydantic import BaseModel
|
||
|
|
from langchain.embeddings.base import Embeddings
|
||
|
|
|
||
|
|
|
||
|
|
class MyEmbeddings(Embeddings, BaseModel):
|
||
|
|
size: int
|
||
|
|
|
||
|
|
def _get_embedding(self) -> List[float]:
|
||
|
|
return list(np.random.normal(size=self.size))
|
||
|
|
|
||
|
|
def embed_documents(self, texts: List[str]) -> List[List[float]]:
|
||
|
|
return [self._get_embedding() for _ in texts]
|
||
|
|
|
||
|
|
def embed_query(self, text: str) -> List[float]:
|
||
|
|
return self._get_embedding()
|