sutro 0.1.14__py3-none-any.whl → 0.1.15__py3-none-any.whl

sutro/sdk.py

@@ -30,6 +30,25 @@ def is_jupyter() -> bool:
 YASPIN_COLOR = None if is_jupyter() else "blue"
 SPINNER = Spinners.dots14
 
+# Models available for inference.  Keep in sync with the backend configuration
+# so users get helpful autocompletion when selecting a model.
+ModelOptions = Literal[
+    "llama-3.2-3b",
+    "llama-3.1-8b",
+    "llama-3.3-70b-8k",
+    "llama-3.3-70b-64k",
+    "qwen-qwq-32b-8k",
+    "qwen-3-4b",
+    "qwen-3-32b",
+    "qwen-3-4b-thinking",
+    "qwen-3-32b-thinking",
+    "gemma-3-4b-it",
+    "gemma-3-27b-it-16k",
+    "gemma-3-27b-it-128k",
+    "multilingual-e5-large-instruct",
+    "gte-qwen2-7b-instruct",
+]
+
 
 def to_colored_text(
     text: str, state: Optional[Literal["success", "fail"]] = None
@@ -156,7 +175,7 @@ class Sutro:
     def infer(
         self,
         data: Union[List, pd.DataFrame, pl.DataFrame, str],
-        model: str = "llama-3.1-8b",
+        model: ModelOptions = "llama-3.1-8b",
         column: str = None,
         output_column: str = "inference_result",
         job_priority: int = 0,
@@ -176,7 +195,7 @@ class Sutro:
 
         Args:
             data (Union[List, pd.DataFrame, pl.DataFrame, str]): The data to run inference on.
-            model (str, optional): The model to use for inference. Defaults to "llama-3.1-8b".
+            model (ModelOptions, optional): The model to use for inference. Defaults to "llama-3.1-8b".
             column (str, optional): The column name to use for inference. Required if data is a DataFrame, file path, or dataset.
             output_column (str, optional): The column name to store the inference results in if the input is a DataFrame. Defaults to "inference_result".
             job_priority (int, optional): The priority of the job. Defaults to 0.

{sutro-0.1.14.dist-info → sutro-0.1.15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sutro
-Version: 0.1.14
+Version: 0.1.15
 Summary: Sutro Python SDK
 Project-URL: Homepage, https://sutro.sh
 Project-URL: Documentation, https://docs.sutro.sh

sutro-0.1.15.dist-info/RECORD

@@ -0,0 +1,8 @@
+sutro/__init__.py,sha256=yUiVwcZ8QamSqDdRHgzoANyTZ-x3cPzlt2Fs5OllR_w,402
+sutro/cli.py,sha256=6Qy9Vwaaho92HeO8YA_z1De4zp1dEFkSX3bEnLvdbkE,13203
+sutro/sdk.py,sha256=fCujFW5Z4wQF9696ABW4bYpUxEZGX8bq5oJjmtpsSRA,42966
+sutro-0.1.15.dist-info/METADATA,sha256=VpTGz6j0cWzAvVdBlRXd3je73aN4zUhNdYEOGFPg5Rk,669
+sutro-0.1.15.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+sutro-0.1.15.dist-info/entry_points.txt,sha256=eXvr4dvMV4UmZgR0zmrY8KOmNpo64cJkhNDywiadRFM,40
+sutro-0.1.15.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+sutro-0.1.15.dist-info/RECORD,,

sutro-0.1.14.dist-info/RECORD

@@ -1,8 +0,0 @@
-sutro/__init__.py,sha256=yUiVwcZ8QamSqDdRHgzoANyTZ-x3cPzlt2Fs5OllR_w,402
-sutro/cli.py,sha256=6Qy9Vwaaho92HeO8YA_z1De4zp1dEFkSX3bEnLvdbkE,13203
-sutro/sdk.py,sha256=K86B8TNkNTSc1k0I6SVJldt6QE-ctnZmc8dMiOjK-PA,42439
-sutro-0.1.14.dist-info/METADATA,sha256=ZpPrBjBQz3CCiXtvvBnJgFyz4cekr9gzi_ETtWtWc-8,669
-sutro-0.1.14.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-sutro-0.1.14.dist-info/entry_points.txt,sha256=eXvr4dvMV4UmZgR0zmrY8KOmNpo64cJkhNDywiadRFM,40
-sutro-0.1.14.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-sutro-0.1.14.dist-info/RECORD,,