v0.5.0 changes

shuttleai · Aug 25, 2024 · 3eca4b6 · 3eca4b6
1 parent f4e7bc6
commit 3eca4b6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/shuttleai-version.json b/shuttleai-version.json
diff --git a/shuttleai/client/base.py b/shuttleai/client/base.py
@@ -47,8 +47,8 @@ def __init__(
 
         self._logger = logging.getLogger(__name__)
         self._default_chat_model = "shuttle-2.5"
-        self._default_image_model = "sdxl"
-        self._default_audio_speech_model = "eleven-labs"
+        self._default_image_model = "shuttle-2-diffusion"
+        self._default_audio_speech_model = "eleven_turbo_v2_5"
         self._version = __version__
 
         if "shuttleai.app" not in self.base_url:
@@ -66,8 +66,8 @@ def __init__(
                     If you wish to use the OpenAI API, consider using their SDK respectively. \
                     Otherwise, please use the official ShuttleAI API URL: https://api.shuttleai.app/v1"
                 )
-            self._default_chat_model = "gpt-3.5-turbo"
-            self._default_image_model = "dall-e-2"
+            self._default_chat_model = "gpt-4o-mini"
+            self._default_image_model = "dall-e-3"
             self._default_audio_speech_model = "whisper-1"
 
         self._logger.info(f"ShuttleAI API client initialized with base URL: {self._base_url}")
@@ -156,7 +156,7 @@ def _make_image_request(self, prompt: str, model: Optional[str] = None) -> Dict[
     def _make_audio_speech_request(
         self,
         input: str,
-        model: str = "eleven-labs",
+        model: str = "eleven_turbo_v2_5",
         voice: Optional[str] = None,
     ) -> Dict[str, Any]:
         request_data: Dict[str, Any] = {