mostlyai-mock 0.0.8__py3-none-any.whl → 0.0.9__py3-none-any.whl

mostlyai/mock/__init__.py

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

mostlyai/mock/mcp_server.py

@@ -0,0 +1,85 @@
+import os
+import tempfile
+import zipfile
+
+import requests
+from fastmcp import Context, FastMCP
+
+from mostlyai import mock
+
+SAMPLE_MOCK_TOOL_DESCRIPTION = f"""
+It is proxy to the `mostlyai.mock.sample` function.
+
+This function returns an URL to the generated CSV bundle (as ZIP file).
+Print this URL in Markdown format, so user can easily download the data.
+
+What comes after the `=============================` is the documentation of the `mostlyai.mock.sample` function.
+
+=============================
+{mock.sample.__doc__}
+"""
+
+mcp = FastMCP(name="MostlyAI Mock MCP Server")
+
+
+def _upload_to_0x0st(data: dict) -> str:
+    with tempfile.TemporaryDirectory() as temp_dir:
+        zip_path = os.path.join(temp_dir, "mock_data.zip")
+        with zipfile.ZipFile(zip_path, "w") as zip_file:
+            for table_name, df in data.items():
+                csv_path = os.path.join(temp_dir, f"{table_name}.csv")
+                df.to_csv(csv_path, index=False)
+                zip_file.write(csv_path, arcname=f"{table_name}.csv")
+
+        with open(zip_path, "rb") as f:
+            response = requests.post(
+                "https://0x0.st",
+                files={"file": f},
+                data={"expires": "24", "secret": ""},
+                headers={"User-Agent": "MockData/1.0"},
+            )
+
+        if response.status_code == 200:
+            url = response.text.strip()
+            return url
+        else:
+            raise Exception(f"Failed to upload ZIP: HTTP {response.status_code}")
+
+
+@mcp.tool(description=SAMPLE_MOCK_TOOL_DESCRIPTION)
+def sample_mock_data(
+    *,
+    tables: dict[str, dict],
+    sample_size: int,
+    model: str = "openai/gpt-4.1-nano",
+    api_key: str | None = None,
+    temperature: float = 1.0,
+    top_p: float = 0.95,
+    ctx: Context,
+) -> str:
+    # Notes:
+    # 1. Returning DataFrames directly results in converting them into truncated string.
+    # 2. The logs / progress bars are not propagated to the MCP Client. There is a dedicated API to do that (e.g. `ctx.info(...)`)
+    # 3. MCP Server inherits only selected environment variables (PATH, USER...); one way to pass LLM keys is through client configuration (`mcpServers->env`)
+    # 4. Some MCP Clients, e.g. Cursor, do not like Unions or Optionals in type hints
+    ctx.info(f"Generating mock data for `{len(tables)}` tables")
+    data = mock.sample(
+        tables=tables,
+        sample_size=sample_size,
+        model=model,
+        api_key=api_key,
+        temperature=temperature,
+        top_p=top_p,
+        return_type="dict",
+    )
+    ctx.info(f"Generated mock data for `{len(tables)}` tables")
+    url = _upload_to_0x0st(data)
+    return url
+
+
+def main():
+    mcp.run(transport="stdio")
+
+
+if __name__ == "__main__":
+    main()

{mostlyai_mock-0.0.8.dist-info → mostlyai_mock-0.0.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mostlyai-mock
-Version: 0.0.8
+Version: 0.0.9
 Summary: Synthetic Mock Data
 Project-URL: homepage, https://github.com/mostly-ai/mostlyai-mock
 Project-URL: repository, https://github.com/mostly-ai/mostlyai-mock
@@ -30,7 +30,6 @@ Requires-Dist: numpy>=1.26.3
 Requires-Dist: pandas>=2.0.0
 Requires-Dist: pyarrow>=14.0.0
 Requires-Dist: pydantic<3.0.0,>=2.0.0
-Requires-Dist: typer<1.0.0,>=0.9.0
 Description-Content-Type: text/markdown
 
 # Synthetic Mock Data 🔮

mostlyai_mock-0.0.9.dist-info/RECORD

@@ -0,0 +1,8 @@
+mostlyai/mock/__init__.py,sha256=jObvPbThXtHSUyMozHQZdSsgvR_fiii7gcPNjnBx0WM,714
+mostlyai/mock/core.py,sha256=p5VAsRppzAc4P8FqKEunfQ3cPjImUU2cEc6yqHJVhMg,29884
+mostlyai/mock/mcp_server.py,sha256=0oyv-7M2Cm9a7JdrMKwHSb3nucPW1J2N6YiYASboAbM,2741
+mostlyai_mock-0.0.9.dist-info/METADATA,sha256=plPb0H5ilUw9_wzB8i5s0Rmw4W0VXWjmx7NiaS7hEGk,11380
+mostlyai_mock-0.0.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mostlyai_mock-0.0.9.dist-info/entry_points.txt,sha256=XDbppUIAaCWW0nresVep8zb71pkzZuFA16jCBHq8CU8,61
+mostlyai_mock-0.0.9.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+mostlyai_mock-0.0.9.dist-info/RECORD,,

mostlyai_mock-0.0.9.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+mcp-server = mostlyai.mock.mcp_server:main

mostlyai/mock/mcp.py

@@ -1,46 +0,0 @@
-import json
-
-import pandas as pd
-from fastmcp import Context, FastMCP
-
-from mostlyai import mock
-
-mcp = FastMCP(name="MostlyAI Mock MCP Server")
-
-
-@mcp.tool(description=mock.sample.__doc__)
-def sample_mock_data(
-    *,
-    tables: dict[str, dict],
-    sample_size: int,
-    model: str = "openai/gpt-4.1-nano",
-    api_key: str | None = None,
-    temperature: float = 1.0,
-    top_p: float = 0.95,
-    ctx: Context,
-) -> str:
-    # Notes:
-    # 1. Returning DataFrames directly results in converting them into truncated string.
-    # 2. The logs / progress bars are not propagated to the MCP Client. There is a dedicated API to do that (e.g. `ctx.info(...)`)
-    # 3. MCP Server inherits only selected environment variables (PATH, USER...); one way to pass LLM keys is through client configuration (`mcpServers->env`)
-    # 4. Some MCP Clients, e.g. Cursor, do not like Unions or Optionals in type hints
-    ctx.info(f"Generating mock data for `{len(tables)}` tables")
-    data = mock.sample(
-        tables=tables,
-        sample_size=sample_size,
-        model=model,
-        api_key=api_key,
-        temperature=temperature,
-        top_p=top_p,
-        return_type="dict",
-    )
-    ctx.info(f"Generated mock data for `{len(tables)}` tables")
-    return {k: v.to_dict(orient="records") for k, v in data.items()}
-
-
-def main():
-    mcp.run(transport="stdio")
-
-
-if __name__ == "__main__":
-    main()

mostlyai_mock-0.0.8.dist-info/RECORD

@@ -1,8 +0,0 @@
-mostlyai/mock/__init__.py,sha256=APjm2I3FljM9lFH6sdbIYyh5eC-8KTOuDKG_KXGXaHQ,714
-mostlyai/mock/core.py,sha256=p5VAsRppzAc4P8FqKEunfQ3cPjImUU2cEc6yqHJVhMg,29884
-mostlyai/mock/mcp.py,sha256=Ryp9aXBzGzHRFVwNt1gftrYlhYcQifCNhjRgn-iX7zc,1400
-mostlyai_mock-0.0.8.dist-info/METADATA,sha256=DCO1oHuu6tOHGog--wEloM31HP0tGbztZP4F1YyzKzk,11415
-mostlyai_mock-0.0.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mostlyai_mock-0.0.8.dist-info/entry_points.txt,sha256=77Swgr1rLuEmeCNOiwoUyKd128_c1Q5l88pj_7JVKaY,54
-mostlyai_mock-0.0.8.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-mostlyai_mock-0.0.8.dist-info/RECORD,,

mostlyai_mock-0.0.8.dist-info/entry_points.txt

@@ -1,2 +0,0 @@
-[console_scripts]
-mcp-server = mostlyai.mock.mcp:main