Skip to content

Commit cb478d3

Browse files
authored
Merge pull request #2178 from tituslhy/main
implement abstract methods for RagasEmbeddingsWrapper
2 parents a698ec1 + e01b153 commit cb478d3

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

‎giskard/rag/metrics/ragas_metrics.py‎

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,12 @@ 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)
77+
7278

7379
class RagasMetric(Metric):
7480
def __init__(

0 commit comments

Comments
 (0)