Initial commit

backend/embeddings.py

@@ -0,0 +1,18 @@
+from config import Config
+import cohere
+import openai
+
+class Embeddings:
+    def __init__(self):
+        self.cohere = cohere.Client(Config.COHERE_API_KEY)
+    
+    def get_embedding(self, text: str, engine: str = "cohere"):
+        if engine == "cohere":
+            response = self.cohere.embed(texts=[text], model="small")
+            return response.embeddings[0]
+        else:  # OpenAI fallback
+            response = openai.Embedding.create(
+                input=[text],
+                model="text-embedding-ada-002"
+            )
+            return response.data[0].embedding