modelparams 0.0.8__tar.gz → 0.0.10__tar.gz

{modelparams-0.0.8 → modelparams-0.0.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: modelparams
-Version: 0.0.8
+Version: 0.0.10
 Summary: Typed model parameters for Python, generated from the modelparams.dev catalog.
 Project-URL: Homepage, https://modelparams.dev
 Project-URL: Repository, https://github.com/mnfst/modelparameters.dev

{modelparams-0.0.8 → modelparams-0.0.10}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "modelparams"
-version = "0.0.8"
+version = "0.0.10"
 description = "Typed model parameters for Python, generated from the modelparams.dev catalog."
 readme = "README.md"
 requires-python = ">=3.10"

{modelparams-0.0.8 → modelparams-0.0.10}/src/modelparams/_generated/catalog.json

@@ -22,7 +22,7 @@
         "type": "number",
         "range": {
           "min": 0,
-          "max": 2,
+          "max": 1.9,
           "step": 0.1
         }
       },
@@ -82,7 +82,7 @@
         "type": "number",
         "range": {
           "min": 0,
-          "max": 2,
+          "max": 1.9,
           "step": 0.1
         }
       },
@@ -142,7 +142,7 @@
         "type": "number",
         "range": {
           "min": 0,
-          "max": 2,
+          "max": 1.9,
           "step": 0.1
         }
       },
@@ -194,7 +194,7 @@
         "type": "number",
         "range": {
           "min": 0,
-          "max": 2,
+          "max": 1.9,
           "step": 0.1
         }
       },
@@ -246,7 +246,7 @@
         "type": "number",
         "range": {
           "min": 0,
-          "max": 2,
+          "max": 1.9,
           "step": 0.1
         }
       },
@@ -306,7 +306,7 @@
         "type": "number",
         "range": {
           "min": 0,
-          "max": 2,
+          "max": 1.9,
           "step": 0.1
         }
       },
@@ -366,7 +366,7 @@
         "type": "number",
         "range": {
           "min": 0,
-          "max": 2,
+          "max": 1.9,
           "step": 0.1
         }
       },
@@ -426,7 +426,7 @@
         "type": "number",
         "range": {
           "min": 0,
-          "max": 2,
+          "max": 1.9,
           "step": 0.1
         }
       },
@@ -501,7 +501,7 @@
         "type": "number",
         "range": {
           "min": 0,
-          "max": 2,
+          "max": 1.9,
           "step": 0.1
         }
       },
@@ -576,7 +576,7 @@
         "type": "number",
         "range": {
           "min": 0,
-          "max": 2,
+          "max": 1.9,
           "step": 0.1
         }
       },
@@ -651,7 +651,7 @@
         "type": "number",
         "range": {
           "min": 0,
-          "max": 2,
+          "max": 1.9,
           "step": 0.1
         }
       },
@@ -726,7 +726,7 @@
         "type": "number",
         "range": {
           "min": 0,
-          "max": 2,
+          "max": 1.9,
           "step": 0.1
         }
       },
@@ -13267,6 +13267,94 @@
       }
     ]
   },
+  {
+    "provider": "openai",
+    "authType": "api_key",
+    "model": "gpt-3.5-turbo-16k",
+    "params": [
+      {
+        "path": "max_tokens",
+        "label": "Max tokens",
+        "description": "Maximum number of output tokens the model may generate.",
+        "group": "generation_length",
+        "type": "integer",
+        "default": 4096,
+        "range": {
+          "min": 1
+        }
+      },
+      {
+        "path": "temperature",
+        "label": "Temperature",
+        "description": "Controls randomness. Lower values make outputs more focused; higher values make them more varied.",
+        "group": "sampling",
+        "type": "number",
+        "default": 1,
+        "range": {
+          "min": 0,
+          "max": 2,
+          "step": 0.1
+        }
+      },
+      {
+        "path": "top_p",
+        "label": "Top P",
+        "description": "Controls nucleus sampling by limiting generation to tokens whose cumulative probability reaches this value.",
+        "group": "sampling",
+        "type": "number",
+        "default": 1,
+        "range": {
+          "min": 0,
+          "max": 1,
+          "step": 0.01
+        }
+      }
+    ]
+  },
+  {
+    "provider": "openai",
+    "authType": "api_key",
+    "model": "gpt-4-0613",
+    "params": [
+      {
+        "path": "max_tokens",
+        "label": "Max tokens",
+        "description": "Maximum number of output tokens the model may generate.",
+        "group": "generation_length",
+        "type": "integer",
+        "default": 4096,
+        "range": {
+          "min": 1
+        }
+      },
+      {
+        "path": "temperature",
+        "label": "Temperature",
+        "description": "Controls randomness. Lower values make outputs more focused; higher values make them more varied.",
+        "group": "sampling",
+        "type": "number",
+        "default": 1,
+        "range": {
+          "min": 0,
+          "max": 2,
+          "step": 0.1
+        }
+      },
+      {
+        "path": "top_p",
+        "label": "Top P",
+        "description": "Controls nucleus sampling by limiting generation to tokens whose cumulative probability reaches this value.",
+        "group": "sampling",
+        "type": "number",
+        "default": 1,
+        "range": {
+          "min": 0,
+          "max": 1,
+          "step": 0.01
+        }
+      }
+    ]
+  },
   {
     "provider": "openai",
     "authType": "api_key",

{modelparams-0.0.8 → modelparams-0.0.10}/src/modelparams/_generated/model_ids.py

@@ -160,6 +160,8 @@ ModelId = Literal[
     "nvidia/usdcode-llama-3.1-70b-instruct",
     "openai/chatgpt-4o-latest",
     "openai/gpt-3.5-turbo",
+    "openai/gpt-3.5-turbo-16k",
+    "openai/gpt-4-0613",
     "openai/gpt-4-turbo",
     "openai/gpt-4-turbo-2024-04-09",
     "openai/gpt-4.1",
@@ -409,6 +411,8 @@ MODEL_IDS: tuple[ModelId, ...] = (
     "nvidia/usdcode-llama-3.1-70b-instruct",
     "openai/chatgpt-4o-latest",
     "openai/gpt-3.5-turbo",
+    "openai/gpt-3.5-turbo-16k",
+    "openai/gpt-4-0613",
     "openai/gpt-4-turbo",
     "openai/gpt-4-turbo-2024-04-09",
     "openai/gpt-4.1",

{modelparams-0.0.8 → modelparams-0.0.10}/src/modelparams/_generated/registry.py

@@ -181,6 +181,8 @@ PARAM_TYPES: dict[ModelId, Any] = {
     "nvidia/usdcode-llama-3.1-70b-instruct": nvidia.Usdcode_Llama_3_1_70b_InstructParams,
     "openai/chatgpt-4o-latest": openai.Chatgpt_4o_LatestParams,
     "openai/gpt-3.5-turbo": openai.Gpt_3_5_TurboParams,
+    "openai/gpt-3.5-turbo-16k": openai.Gpt_3_5_Turbo_16kParams,
+    "openai/gpt-4-0613": openai.Gpt_4_0613Params,
     "openai/gpt-4-turbo": openai.Gpt_4_TurboParams,
     "openai/gpt-4-turbo-2024-04-09": openai.Gpt_4_Turbo_2024_04_09Params,
     "openai/gpt-4.1": openai.Gpt_4_1Params,

{modelparams-0.0.8 → modelparams-0.0.10}/src/modelparams/types/alibaba.py

@@ -14,7 +14,7 @@ Qwen_FlashParams = TypedDict(
     "Qwen_FlashParams",
     {
         "max_tokens": Annotated[int, Field(ge=1)],
-        "temperature": Annotated[float, Field(ge=0, le=2)],
+        "temperature": Annotated[float, Field(ge=0, le=1.9)],
         "top_p": Annotated[float, Field(ge=0, le=1)],
         "extra_body.top_k": Annotated[int, Field(ge=1)],
         "extra_body.chat_template_kwargs.enable_thinking": bool,
@@ -27,7 +27,7 @@ Qwen_PlusParams = TypedDict(
     "Qwen_PlusParams",
     {
         "max_tokens": Annotated[int, Field(ge=1)],
-        "temperature": Annotated[float, Field(ge=0, le=2)],
+        "temperature": Annotated[float, Field(ge=0, le=1.9)],
         "top_p": Annotated[float, Field(ge=0, le=1)],
         "extra_body.top_k": Annotated[int, Field(ge=1)],
         "extra_body.chat_template_kwargs.enable_thinking": bool,
@@ -40,7 +40,7 @@ Qwen3_Coder_FlashParams = TypedDict(
     "Qwen3_Coder_FlashParams",
     {
         "max_tokens": Annotated[int, Field(ge=1)],
-        "temperature": Annotated[float, Field(ge=0, le=2)],
+        "temperature": Annotated[float, Field(ge=0, le=1.9)],
         "top_p": Annotated[float, Field(ge=0, le=1)],
         "extra_body.top_k": Annotated[int, Field(ge=1)],
     },
@@ -52,7 +52,7 @@ Qwen3_Coder_PlusParams = TypedDict(
     "Qwen3_Coder_PlusParams",
     {
         "max_tokens": Annotated[int, Field(ge=1)],
-        "temperature": Annotated[float, Field(ge=0, le=2)],
+        "temperature": Annotated[float, Field(ge=0, le=1.9)],
         "top_p": Annotated[float, Field(ge=0, le=1)],
         "extra_body.top_k": Annotated[int, Field(ge=1)],
     },
@@ -64,7 +64,7 @@ Qwen3_MaxParams = TypedDict(
     "Qwen3_MaxParams",
     {
         "max_tokens": Annotated[int, Field(ge=1)],
-        "temperature": Annotated[float, Field(ge=0, le=2)],
+        "temperature": Annotated[float, Field(ge=0, le=1.9)],
         "top_p": Annotated[float, Field(ge=0, le=1)],
         "extra_body.top_k": Annotated[int, Field(ge=1)],
         "extra_body.chat_template_kwargs.enable_thinking": bool,
@@ -77,7 +77,7 @@ Qwen3_5Params = TypedDict(
     "Qwen3_5Params",
     {
         "max_tokens": Annotated[int, Field(ge=1)],
-        "temperature": Annotated[float, Field(ge=0, le=2)],
+        "temperature": Annotated[float, Field(ge=0, le=1.9)],
         "top_p": Annotated[float, Field(ge=0, le=1)],
         "extra_body.top_k": Annotated[int, Field(ge=1)],
         "extra_body.chat_template_kwargs.enable_thinking": bool,
@@ -90,7 +90,7 @@ Qwen3_5_FlashParams = TypedDict(
     "Qwen3_5_FlashParams",
     {
         "max_tokens": Annotated[int, Field(ge=1)],
-        "temperature": Annotated[float, Field(ge=0, le=2)],
+        "temperature": Annotated[float, Field(ge=0, le=1.9)],
         "top_p": Annotated[float, Field(ge=0, le=1)],
         "extra_body.top_k": Annotated[int, Field(ge=1)],
         "extra_body.chat_template_kwargs.enable_thinking": bool,
@@ -103,7 +103,7 @@ Qwen3_6_FlashParams = TypedDict(
     "Qwen3_6_FlashParams",
     {
         "max_completion_tokens": Annotated[int, Field(ge=1)],
-        "temperature": Annotated[float, Field(ge=0, le=2)],
+        "temperature": Annotated[float, Field(ge=0, le=1.9)],
         "top_p": Annotated[float, Field(ge=0, le=1)],
         "extra_body.top_k": Annotated[int, Field(ge=0)],
         "extra_body.enable_thinking": bool,
@@ -117,7 +117,7 @@ Qwen3_7_MaxParams = TypedDict(
     "Qwen3_7_MaxParams",
     {
         "max_completion_tokens": Annotated[int, Field(ge=1)],
-        "temperature": Annotated[float, Field(ge=0, le=2)],
+        "temperature": Annotated[float, Field(ge=0, le=1.9)],
         "top_p": Annotated[float, Field(ge=0, le=1)],
         "extra_body.top_k": Annotated[int, Field(ge=0)],
         "extra_body.enable_thinking": bool,
@@ -131,7 +131,7 @@ Qwen3_7_PlusParams = TypedDict(
     "Qwen3_7_PlusParams",
     {
         "max_completion_tokens": Annotated[int, Field(ge=1)],
-        "temperature": Annotated[float, Field(ge=0, le=2)],
+        "temperature": Annotated[float, Field(ge=0, le=1.9)],
         "top_p": Annotated[float, Field(ge=0, le=1)],
         "extra_body.top_k": Annotated[int, Field(ge=0)],
         "extra_body.enable_thinking": bool,
@@ -145,7 +145,7 @@ Qwen3_8_MaxParams = TypedDict(
     "Qwen3_8_MaxParams",
     {
         "max_completion_tokens": Annotated[int, Field(ge=1)],
-        "temperature": Annotated[float, Field(ge=0, le=2)],
+        "temperature": Annotated[float, Field(ge=0, le=1.9)],
         "top_p": Annotated[float, Field(ge=0, le=1)],
         "extra_body.top_k": Annotated[int, Field(ge=0)],
         "extra_body.enable_thinking": bool,
@@ -159,7 +159,7 @@ Qwq_PlusParams = TypedDict(
     "Qwq_PlusParams",
     {
         "max_tokens": Annotated[int, Field(ge=1)],
-        "temperature": Annotated[float, Field(ge=0, le=2)],
+        "temperature": Annotated[float, Field(ge=0, le=1.9)],
         "top_p": Annotated[float, Field(ge=0, le=1)],
         "extra_body.top_k": Annotated[int, Field(ge=1)],
     },

{modelparams-0.0.8 → modelparams-0.0.10}/src/modelparams/types/openai.py

@@ -32,6 +32,28 @@ Gpt_3_5_TurboParams = TypedDict(
 )
 setattr(Gpt_3_5_TurboParams, "__pydantic_config__", _PARAMS_CONFIG)
 
+Gpt_3_5_Turbo_16kParams = TypedDict(
+    "Gpt_3_5_Turbo_16kParams",
+    {
+        "max_tokens": Annotated[int, Field(ge=1)],
+        "temperature": Annotated[float, Field(ge=0, le=2)],
+        "top_p": Annotated[float, Field(ge=0, le=1)],
+    },
+    total=False,
+)
+setattr(Gpt_3_5_Turbo_16kParams, "__pydantic_config__", _PARAMS_CONFIG)
+
+Gpt_4_0613Params = TypedDict(
+    "Gpt_4_0613Params",
+    {
+        "max_tokens": Annotated[int, Field(ge=1)],
+        "temperature": Annotated[float, Field(ge=0, le=2)],
+        "top_p": Annotated[float, Field(ge=0, le=1)],
+    },
+    total=False,
+)
+setattr(Gpt_4_0613Params, "__pydantic_config__", _PARAMS_CONFIG)
+
 Gpt_4_TurboParams = TypedDict(
     "Gpt_4_TurboParams",
     {
@@ -558,6 +580,8 @@ setattr(O4_MiniParams, "__pydantic_config__", _PARAMS_CONFIG)
 __all__ = [
     "Chatgpt_4o_LatestParams",
     "Gpt_3_5_TurboParams",
+    "Gpt_3_5_Turbo_16kParams",
+    "Gpt_4_0613Params",
     "Gpt_4_TurboParams",
     "Gpt_4_Turbo_2024_04_09Params",
     "Gpt_4_1Params",