mostlyai-mock 0.1.15__py3-none-any.whl → 0.1.16__py3-none-any.whl

mostlyai/mock/__init__.py

@@ -15,4 +15,4 @@
 from mostlyai.mock.core import sample
 
 __all__ = ["sample"]
-__version__ = "0.1.15"  # Do not set this manually. Use poetry version [params].
+__version__ = "0.1.16"  # Do not set this manually. Use poetry version [params].

mostlyai/mock/core.py

@@ -645,6 +645,24 @@ async def _worker(
                 "stream": True,
             }
 
+            # support for openai reasoning models
+            model_only = llm_config.model.split("/")[-1] if "/" in llm_config.model else llm_config.model
+            reasoning_effort = (
+                "low"
+                if (model_only.startswith("o") and (model_only[1:].isdigit() or model_only[1:].split("-")[0].isdigit()))
+                else "minimal"
+                if (
+                    model_only.startswith("gpt-")
+                    and model_only.split("-")[1].isdigit()
+                    and int(model_only.split("-")[1]) >= 5
+                )
+                else None
+            )
+
+            if reasoning_effort:
+                litellm_kwargs.pop("top_p")
+                litellm_kwargs["reasoning_effort"] = reasoning_effort
+
             # construct messages
             system_prompt = _create_system_prompt(llm_output_format)
             user_prompt = _create_table_prompt(
@@ -1116,7 +1134,7 @@ async def _sample_common(
     tables: dict[str, dict],
     sample_size: int | dict[str, int] = 4,
     existing_data: dict[str, pd.DataFrame] | None = None,
-    model: str = "openai/gpt-4.1-nano",
+    model: str = "openai/gpt-5-nano",
     api_key: str | None = None,
     temperature: float = 1.0,
     top_p: float = 0.95,
@@ -1164,7 +1182,7 @@ def sample(
     tables: dict[str, dict],
     sample_size: int | dict[str, int] = 4,
     existing_data: dict[str, pd.DataFrame] | None = None,
-    model: str = "openai/gpt-4.1-nano",
+    model: str = "openai/gpt-5-nano",
     api_key: str | None = None,
     temperature: float = 1.0,
     top_p: float = 0.95,
@@ -1194,9 +1212,9 @@ def sample(
             Default is None.
         model (str): The LiteLLM chat completion model to be used.
             Examples include:
-            - `openai/gpt-4.1-nano` (default; fast, and smart)
-            - `openai/gpt-4.1-mini` (slower, but smarter)
-            - `openai/gpt-4.1` (slowest, but smartest)
+            - `openai/gpt-5-nano` (default; fast, and smart)
+            - `openai/gpt-5-mini` (slower, but smarter)
+            - `openai/gpt-5` (slowest, but smartest)
             - `gemini/gemini-2.0-flash`
             - `gemini/gemini-2.5-flash-preview-04-17`
             - 'groq/gemma2-9b-it`
@@ -1234,7 +1252,7 @@ def sample(
             },
         }
     }
-    df = mock.sample(tables=tables, sample_size=10, model="openai/gpt-4.1-nano")
+    df = mock.sample(tables=tables, sample_size=10, model="openai/gpt-5-nano")
     ```
 
     Example of generating mock data for multiple tables (with PK/FK relationships):
@@ -1297,7 +1315,7 @@ def sample(
             ],
         },
     }
-    data = mock.sample(tables=tables, sample_size=2, model="openai/gpt-4.1")
+    data = mock.sample(tables=tables, sample_size=2, model="openai/gpt-5")
     df_customers = data["customers"]
     df_warehouses = data["warehouses"]
     df_orders = data["orders"]
@@ -1326,7 +1344,7 @@ def sample(
     enriched_df = mock.sample(
         tables=tables,
         existing_data={"patients": existing_df},
-        model="openai/gpt-4.1-nano"
+        model="openai/gpt-5-nano"
     )
     enriched_df
     ```
@@ -1381,7 +1399,7 @@ def sample(
             "customers": existing_customers,
             "orders": existing_orders,
         },
-        model="openai/gpt-4.1-nano"
+        model="openai/gpt-5-nano"
     )
     df_customers = data["customers"]
     df_orders = data["orders"]
@@ -1413,7 +1431,7 @@ async def _asample(
     tables: dict[str, dict],
     sample_size: int | dict[str, int] = 4,
     existing_data: dict[str, pd.DataFrame] | None = None,
-    model: str = "openai/gpt-4.1-nano",
+    model: str = "openai/gpt-5-nano",
     api_key: str | None = None,
     temperature: float = 1.0,
     top_p: float = 0.95,

mostlyai/mock/mcp_server.py

@@ -56,7 +56,7 @@ async def mock_data(
     *,
     tables: dict[str, dict],
     sample_size: int,
-    model: str = "openai/gpt-4.1-nano",
+    model: str = "openai/gpt-5-nano",
     api_key: str | None = None,
     temperature: float = 1.0,
     top_p: float = 0.95,

{mostlyai_mock-0.1.15.dist-info → mostlyai_mock-0.1.16.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mostlyai-mock
-Version: 0.1.15
+Version: 0.1.16
 Summary: Synthetic Mock Data
 Project-URL: homepage, https://github.com/mostly-ai/mostlyai-mock
 Project-URL: repository, https://github.com/mostly-ai/mostlyai-mock
@@ -95,7 +95,7 @@ tables = {
 df = mock.sample(
     tables=tables,   # provide table and column definitions
     sample_size=10,  # generate 10 records
-    model="openai/gpt-4.1-nano",  # select the LLM model (optional)
+    model="openai/gpt-5-nano",  # select the LLM model (optional)
 )
 print(df)
 #   nationality                 name  gender  age date_of_birth        checkin_time is_vip  price_per_night  room_number
@@ -176,7 +176,7 @@ tables = {
 data = mock.sample(
     tables=tables,
     sample_size=2,
-    model="openai/gpt-4.1",
+    model="openai/gpt-5",
     n_workers=1,
 )
 print(data["customers"])
@@ -232,7 +232,7 @@ tables = {
         ],
     }
 }
-df = mock.sample(tables=tables, sample_size=10, model="openai/gpt-4.1", n_workers=1)
+df = mock.sample(tables=tables, sample_size=10, model="openai/gpt-5", n_workers=1)
 print(df)
 #   employee_id              name boss_id                   role
 # 0        B0-1      Patricia Lee    <NA>              President
@@ -273,7 +273,7 @@ existing_guests = pd.DataFrame({
 df = mock.sample(
     tables=tables,
     existing_data={"guests": existing_guests},
-    model="openai/gpt-4.1-nano"
+    model="openai/gpt-5-nano"
 )
 print(df)
 #   guest_id           name nationality  gender  age  room_number is_vip

mostlyai_mock-0.1.16.dist-info/RECORD

@@ -0,0 +1,8 @@
+mostlyai/mock/__init__.py,sha256=XEezyGjkXQBReW_ORi83H2WEVhLolDDLbGjxA2g2yEs,715
+mostlyai/mock/core.py,sha256=FTF0BfJowxNHm_L0RpTk6BhS1mXzvjELP-3Z96aFVMQ,62454
+mostlyai/mock/mcp_server.py,sha256=uDLg0SeMPV2VZhXviM-F769W0xlmhGwlmQiQhY0Q-Ik,2365
+mostlyai_mock-0.1.16.dist-info/METADATA,sha256=CT6lcz2cAq5W-u3VjQLr_Dg8VbuEtU-JlvsXg5OsKTk,14297
+mostlyai_mock-0.1.16.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mostlyai_mock-0.1.16.dist-info/entry_points.txt,sha256=XDbppUIAaCWW0nresVep8zb71pkzZuFA16jCBHq8CU8,61
+mostlyai_mock-0.1.16.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+mostlyai_mock-0.1.16.dist-info/RECORD,,

mostlyai_mock-0.1.15.dist-info/RECORD

@@ -1,8 +0,0 @@
-mostlyai/mock/__init__.py,sha256=uv2DLnOleN6BNMfAMleXJCPcOZvM_tMTRy5njUIKDag,715
-mostlyai/mock/core.py,sha256=JDJ9nVpRR2WochxumSdQS96zak0OV1frkJOwlQsPVBw,61715
-mostlyai/mock/mcp_server.py,sha256=0Vn1jmrdNAvUZSviaaU7Lhn7L7iHFyd8kGFigM0-4s0,2367
-mostlyai_mock-0.1.15.dist-info/METADATA,sha256=OG3NRdCcH2qycRQ5HrzyLtJwtl74lRw5Py1JtqfB2YI,14305
-mostlyai_mock-0.1.15.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mostlyai_mock-0.1.15.dist-info/entry_points.txt,sha256=XDbppUIAaCWW0nresVep8zb71pkzZuFA16jCBHq8CU8,61
-mostlyai_mock-0.1.15.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-mostlyai_mock-0.1.15.dist-info/RECORD,,