Simplifying embedding service

Author: tomsmoker

backend/src/app/services/embedding/openai_embedding_service.py

@@ -3,7 +3,7 @@
 import logging
 from typing import List
 
-from langchain_openai import OpenAIEmbeddings
+from openai import OpenAI
 
 from app.core.config import Settings
 from app.services.embedding.base import EmbeddingService
@@ -16,20 +16,22 @@ class OpenAIEmbeddingService(EmbeddingService):
 
     def __init__(self, settings: Settings) -> None:
         self.settings = settings
+        self.client = OpenAI(api_key=settings.openai_api_key)
+        self.model = settings.embedding_model
         if not settings.openai_api_key:
             raise ValueError("OpenAI API key is required but not set")
-        
-        self.embeddings = OpenAIEmbeddings(
-            api_key=settings.openai_api_key,
-            model=settings.embedding_model
-        )
 
     async def get_embeddings(self, texts: List[str]) -> List[List[float]]:
         """Get embeddings for text."""
-        if self.embeddings is None:
+        if self.client is None:
             logger.warning(
                 "OpenAI client is not initialized. Skipping embeddings."
             )
             return []
 
-        return self.embeddings.embed_documents(texts)
+        return [
+            embedding.embedding
+            for embedding in self.client.embeddings.create(
+                input=texts, model=self.model
+            ).data
+        ]