add embedding base class

Author: shahules786

ragas_annotator/_modidx.py

@@ -119,6 +119,16 @@
                                          'ragas_annotator.dataset.Dataset.pop': ('dataset.html#dataset.pop', 'ragas_annotator/dataset.py'),
                                          'ragas_annotator.dataset.Dataset.save': ( 'dataset.html#dataset.save',
                                                                                    'ragas_annotator/dataset.py')},
+            'ragas_annotator.embedding.base': { 'ragas_annotator.embedding.base.RagasEmbedding': ( 'embedding/base.html#ragasembedding',
+                                                                                                   'ragas_annotator/embedding/base.py'),
+                                                'ragas_annotator.embedding.base.RagasEmbedding.aembed_document': ( 'embedding/base.html#ragasembedding.aembed_document',
+                                                                                                                   'ragas_annotator/embedding/base.py'),
+                                                'ragas_annotator.embedding.base.RagasEmbedding.aembed_text': ( 'embedding/base.html#ragasembedding.aembed_text',
+                                                                                                               'ragas_annotator/embedding/base.py'),
+                                                'ragas_annotator.embedding.base.RagasEmbedding.embed_document': ( 'embedding/base.html#ragasembedding.embed_document',
+                                                                                                                  'ragas_annotator/embedding/base.py'),
+                                                'ragas_annotator.embedding.base.RagasEmbedding.embed_text': ( 'embedding/base.html#ragasembedding.embed_text',
+                                                                                                              'ragas_annotator/embedding/base.py')},
             'ragas_annotator.exceptions': { 'ragas_annotator.exceptions.DuplicateError': ( 'utils/exceptions.html#duplicateerror',
                                                                                            'ragas_annotator/exceptions.py'),
                                             'ragas_annotator.exceptions.NotFoundError': ( 'utils/exceptions.html#notfounderror',

ragas_annotator/embedding/__init__.py

@@ -0,0 +1,3 @@
+from ragas_annotator.embedding.base import RagasEmbedding
+
+__all__ = ['RagasEmbedding']

ragas_annotator/embedding/base.py

@@ -0,0 +1,32 @@
+# AUTOGENERATED! DO NOT EDIT! File to edit: ../../nbs/embedding/base.ipynb.
+
+# %% auto 0
+__all__ = ['RagasEmbedding']
+
+# %% ../../nbs/embedding/base.ipynb 2
+import typing as t
+from dataclasses import dataclass
+
+
+@dataclass
+class RagasEmbedding:
+    client: t.Any
+    model: str
+    
+    def embed_text(self,text:str,**kwargs: t.Any) -> t.List[float]:
+        
+        return self.client.embeddings.create(input=text, model=self.model, **kwargs).data[0].embedding
+        
+    async def aembed_text(self,text:str,**kwargs: t.Any):
+        
+        await self.client.embeddings.create(input=text, model=self.model, **kwargs).data[0].embedding
+    
+    
+    def embed_document(self,documents:t.List[str],**kwargs: t.Any) -> t.List[t.List[float]]:
+        embeddings = self.client.embeddings.create(input=documents, model=self.model, **kwargs)
+        return [embedding.embedding for embedding in embeddings.data]
+    
+    async def aembed_document(self,documents:t.List[str],**kwargs: t.Any)  -> t.List[t.List[float]]:
+        embeddings = await self.client.embeddings.create(input=documents, model=self.model, **kwargs)
+        return [embedding.embedding for embedding in embeddings.data]
+