modelparams 0.0.8__tar.gz → 0.0.9__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: modelparams
-Version: 0.0.8
+Version: 0.0.9
 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.9}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "modelparams"
-version = "0.0.8"
+version = "0.0.9"
 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.9}/src/modelparams/_generated/catalog.json

@@ -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.9}/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.9}/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.9}/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",