openai_ops.py

"""
This module provides a class for interacting with the Azure OpenAI API.
It includes methods for generating embeddings and text responses.
"""

from typing import List

from openai import AzureOpenAI


class OpenAIOperations:
    def __init__(self, api_key: str):
        """
        Initialize the OpenAIOperations class.
        
        Args:
            api_key (str): The API key for Azure OpenAI.
        """
        self.client = AzureOpenAI(
            azure_endpoint="https://confmind-oai3.openai.azure.com/",     
            api_key=api_key,
            api_version="2024-02-15-preview",
        )

    def get_embedding(self, text: str) -> List[float]:
        """
        Generate an embedding for the given text.
        
        Args:
            text (str): The input text to embed.
        
        Returns:
            List[float]: The embedding vector.
        """
        response = self.client.embeddings.create(model="RAGProtoEmbed", input=text)
        return response.data[0].embedding

    def generate_response(self, context: str, question: str) -> str:
        """
        Generate a response to a question based on the given context.
        
        Args:
            context (str): The context information.
            question (str): The question to answer.
        
        Returns:
            str: The generated response.
        """
        prompt = f"Context: {context}\n\nQuestion: {question}\n\nAnswer:"
        response = self.client.chat.completions.create(
            model="RAGProto",
            messages=[
                {
                    "role": "system",
                    "content": "You are a helpful assistant that answers questions based on the given context.",
                },
                {"role": "user", "content": prompt},
            ],
            temperature=0.7,
            max_tokens=800,
            top_p=0.95,
            frequency_penalty=0,
            presence_penalty=0,
            stop=None,
        )
        return response.choices[0].message.content