AtlasML: Add initial embedding model

AtlasMl/atlasml/ml/VectorEmbeddings/FallbackModel.py

@@ -0,0 +1,8 @@
+from sentence_transformers import SentenceTransformer
+
+sentence = ["This is an example sentence"]
+model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
+embeddings = model.encode(sentence)
+
+# Sanity check
+print(embeddings)

AtlasMl/atlasml/ml/VectorEmbeddings/MainEmbeddingModel.py

@@ -0,0 +1,12 @@
+from openai import OpenAI
+
+client = OpenAI()
+client.api_key = "" # Replace with OpenAI API Key
+
+response = client.embeddings.create(
+    input="Task Text",
+    model="text-embedding-3-small"
+)
+
+# Sanity check
+print(response.data[0].embedding)

AtlasMl/atlasml/ml/__init__.py

@@ -1,4 +1 @@
-from .ml_runner import ml_runner
-from .model import ml_model_handler
 
-__all__ = ["ml_runner", "ml_model_handler"]

AtlasMl/atlasml/models/__init__.py

@@ -0,0 +1,5 @@
+
+class ml_model_handler:
+    def __init__(self, model):
+        self.model = model
+