implement model config

open_notebook/models/speech_to_text_models.py

@@ -0,0 +1,62 @@
+"""
+Classes for supporting different transcription models
+"""
+
+from abc import ABC, abstractmethod
+from dataclasses import dataclass
+from typing import Optional
+
+from openai import OpenAI
+
+from open_notebook.domain.models import Model
+
+
+@dataclass
+class SpeechToTextModel(ABC):
+    """
+    Abstract base class for speech to text models.
+    """
+
+    model_name: Optional[str] = None
+
+    @abstractmethod
+    def transcribe(self, audio_file_path: str) -> str:
+        """
+        Generates a text transcription from audio
+        """
+        raise NotImplementedError
+
+
+@dataclass
+class OpenAISpeechToTextModel(SpeechToTextModel):
+    model_name: str
+
+    def transcribe(self, audio_file_path: str) -> str:
+        """
+        Transcribes an audio file into text
+        """
+        # todo: make this Singleton
+        client = OpenAI()
+        with open(audio_file_path, "rb") as audio:
+            transcription = client.audio.transcriptions.create(
+                model=self.model_name, file=audio
+            )
+            return transcription.text
+
+
+SPEECH_TO_TEXT_CLASS_MAP = {
+    "openai": OpenAISpeechToTextModel,
+}
+
+
+# todo: acho que dá pra juntar todos os get models em uma coisa só
+def get_speech_to_text_model(model_id):
+    assert model_id, "Model ID cannot be empty"
+    model = Model.get(model_id)
+    if not model:
+        raise ValueError(f"Model with ID {model_id} not found")
+    if model.provider not in SPEECH_TO_TEXT_CLASS_MAP.keys():
+        raise ValueError(
+            f"Provider {model.provider} not compatible with Embedding Models"
+        )
+    return SPEECH_TO_TEXT_CLASS_MAP[model.provider](model_name=model.name)