File size: 542 Bytes
971b4ea
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
# 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