Skip to content

Commit d7bd6f8

Browse files
committed
implement abstract methods for RagasEmbeddingsWrapper
1 parent a698ec1 commit d7bd6f8

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

‎giskard/rag/metrics/ragas_metrics.py‎

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,11 @@ def embed_query(self, text: str) -> Sequence[float]:
6969
def embed_documents(self, texts: Sequence[str]) -> Sequence[Sequence[float]]:
7070
return self.embedding_model.embed(texts)
7171

72+
async def aembed_query(self, text: str) -> Sequence[float]:
73+
return await self.embedding_model.aembed([text])[0]
74+
75+
async def aembed_documents(self, texts: Sequence[str]) -> Sequence[Sequence[float]]:
76+
return await self.embedding_model.aembed(texts)
7277

7378
class RagasMetric(Metric):
7479
def __init__(

0 commit comments

Comments
 (0)