Launchlab / app /services /embeddings_service.py
MuhammadSaad16's picture
Upload 112 files
971b4ea verified
# app/services/embeddings_service.py
from openai import OpenAI
from app.config import settings
class EmbeddingsService:
def __init__(self):
self.client = OpenAI(
api_key=settings.OPENAI_API_KEY
)
self.model = settings.OPENAI_MODEL_EMBEDDING
def create_embedding(self, text: str):
"""Generate embedding for text using OpenAI API."""
response = self.client.embeddings.create(
model=self.model,
input=text
)
return response.data[0].embedding