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

mostlyai/mock/__init__.py

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

mostlyai/mock/mcp_server.py

@@ -1,17 +1,21 @@
 import os
 import tempfile
-import zipfile
 
-import requests
+import pandas as pd
 from fastmcp import Context, FastMCP
 
 from mostlyai import mock
 
 SAMPLE_MOCK_TOOL_DESCRIPTION = f"""
-It is proxy to the `mostlyai.mock.sample` function.
+This tool is a proxy to the `mostlyai.mock.sample` function.
+It returns a dictionary. The keys are the table names, the values are the Paths to the generated CSV files.
+
+Present the result nicely to the user, in Markdown format. Example:
+
+Mock data can be found under the following paths:
+- `/tmp/tmpl41bwa6n/players.csv`
+- `/tmp/tmpl41bwa6n/seasons.csv`
 
-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.
 
@@ -22,28 +26,14 @@ What comes after the `=============================` is the documentation of the
 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}")
+def _store_locally(data: dict[str, pd.DataFrame]) -> dict[str, str]:
+    temp_dir = tempfile.mkdtemp()
+    locations = {}
+    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)
+        locations[table_name] = csv_path
+    return locations
 
 
 @mcp.tool(description=SAMPLE_MOCK_TOOL_DESCRIPTION)
@@ -56,7 +46,7 @@ def sample_mock_data(
     temperature: float = 1.0,
     top_p: float = 0.95,
     ctx: Context,
-) -> str:
+) -> dict[str, 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(...)`)
@@ -73,8 +63,8 @@ def sample_mock_data(
         return_type="dict",
     )
     ctx.info(f"Generated mock data for `{len(tables)}` tables")
-    url = _upload_to_0x0st(data)
-    return url
+    locations = _store_locally(data)
+    return locations
 
 
 def main():

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mostlyai-mock
-Version: 0.0.9
+Version: 0.0.11
 Summary: Synthetic Mock Data
 Project-URL: homepage, https://github.com/mostly-ai/mostlyai-mock
 Project-URL: repository, https://github.com/mostly-ai/mostlyai-mock
@@ -238,4 +238,34 @@ print(df)
 # 7            8     Lucas Romero        3                IT Manager
 # 8            9      Priya Desai        3    Lead Software Engineer
 # 9           10    Felix Bennett        3    Senior Systems Analyst
-```
+```
+
+## MCP Server
+
+This repo comes with MCP Server. It can be easily consumed by any MCP Client by providing the following configuration:
+
+```json
+{
+  "mcpServers": {
+      "mostlyai-mock-mcp": {
+          "command": "uvx",
+          "args": ["--from", "mostlyai-mock", "mcp-server"],
+          "env": {
+              "OPENAI_API_KEY": "PROVIDE YOUR KEY",
+              "GEMINI_API_KEY": "PROVIDE YOUR KEY",
+              "GROQ_API_KEY": "PROVIDE YOUR KEY",
+              "ANTHROPIC_API_KEY": "PROVIDE YOUR KEY"
+          }
+      }
+  }
+}
+```
+
+For example:
+- in Claude Desktop, go to "Settings" > "Developer" > "Edit Config" and paste the above into `claude_desktop_config.json`
+- in Cursor, go to "Settings" > "Cursor Settings" > "MCP" > "Add new global MCP server" and paste the above into `mcp.json`
+
+Troubleshooting:
+1. If the MCP Client fails to detect the MCP Server, provide the absolute path in the `command` field, for example: `/Users/johnsmith/.local/bin/uvx`
+2. To debug MCP Server issues, you can use MCP Inspector by running: `npx @modelcontextprotocol/inspector -- uvx --from mostlyai-mock mcp-server`
+3. In order to develop locally, modify the configuration by replacing `"command": "uv"` (or use the full path to `uv` if needed) and `"args": ["--from", "mostlyai-mock", "mcp-server"]`

mostlyai_mock-0.0.11.dist-info/RECORD

@@ -0,0 +1,8 @@
+mostlyai/mock/__init__.py,sha256=fDLeoKXZggBnsgC_lhC7-O_DltAbc-Wea0QL0xgsjtY,715
+mostlyai/mock/core.py,sha256=p5VAsRppzAc4P8FqKEunfQ3cPjImUU2cEc6yqHJVhMg,29884
+mostlyai/mock/mcp_server.py,sha256=F4O49tK2NJV-N53gxbJRNk24-lx5b2YgaUaojQhNAqQ,2318
+mostlyai_mock-0.0.11.dist-info/METADATA,sha256=tApgQujSIyaH_jutE-_m3-3L8sqm1nXrSrxmzOW-d6M,12680
+mostlyai_mock-0.0.11.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mostlyai_mock-0.0.11.dist-info/entry_points.txt,sha256=XDbppUIAaCWW0nresVep8zb71pkzZuFA16jCBHq8CU8,61
+mostlyai_mock-0.0.11.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+mostlyai_mock-0.0.11.dist-info/RECORD,,

mostlyai_mock-0.0.9.dist-info/RECORD

@@ -1,8 +0,0 @@
-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,,