mostlyai-mock 0.0.8__py3-none-any.whl → 0.0.10__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.10"  # Do not set this manually. Use poetry version [params].

mostlyai/mock/{mcp.py → mcp_server.py}

@@ -1,14 +1,42 @@
-import json
+import os
+import tempfile
+import zipfile
 
-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 returns an URL or a Path to the generated CSV bundle (as ZIP file).
+Present the result nicely to the user, in Markdown format. Some examples:
+
+"Mock data is ready to download: [Mock Data](https://example.com/mock_data.zip)" (if result is a link)
+"Mock data can be found in `/tmp/tmpl41bwa6n/mock_data.zip`" (if result is a path)
+
+What comes after the `=============================` is the documentation of the `mostlyai.mock.sample` function.
+
+=============================
+{mock.sample.__doc__}
+"""
+
 mcp = FastMCP(name="MostlyAI Mock MCP Server")
 
 
-@mcp.tool(description=mock.sample.__doc__)
+def _store_locally(data: dict) -> str:
+    temp_dir = tempfile.mkdtemp()
+    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")
+
+    return os.path.abspath(zip_path)
+
+
+@mcp.tool(description=SAMPLE_MOCK_TOOL_DESCRIPTION)
 def sample_mock_data(
     *,
     tables: dict[str, dict],
@@ -35,7 +63,8 @@ def sample_mock_data(
         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()}
+    url = _store_locally(data)
+    return url
 
 
 def main():

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mostlyai-mock
-Version: 0.0.8
+Version: 0.0.10
 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 🔮
@@ -239,4 +238,33 @@ 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:
+
+```yaml
+{
+  "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 MCP Server is not picked up by the MCP Client, specify full path in `command`, e.g. `/Users/johnsmith/.local/bin/uvx`
+2. MCP Inspector can be used for debugging: `npx @modelcontextprotocol/inspector -- uvx --from mostlyai-mock mcp-server`

mostlyai_mock-0.0.10.dist-info/RECORD

@@ -0,0 +1,8 @@
+mostlyai/mock/__init__.py,sha256=grEzN1CxhOdKf-ow8fWVNTzUrB2Ogg8IBQIKqyaOc7I,715
+mostlyai/mock/core.py,sha256=p5VAsRppzAc4P8FqKEunfQ3cPjImUU2cEc6yqHJVhMg,29884
+mostlyai/mock/mcp_server.py,sha256=juy5n6-Xo-ZVC3u5o2zylLgN1CaAwzIaS_bk2qXxpcU,2439
+mostlyai_mock-0.0.10.dist-info/METADATA,sha256=OHHgoJQ6_RL20Ze5ofFvMscXK5oBOXTfYyEYy0XwsCU,12443
+mostlyai_mock-0.0.10.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mostlyai_mock-0.0.10.dist-info/entry_points.txt,sha256=XDbppUIAaCWW0nresVep8zb71pkzZuFA16jCBHq8CU8,61
+mostlyai_mock-0.0.10.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+mostlyai_mock-0.0.10.dist-info/RECORD,,

mostlyai_mock-0.0.10.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+mcp-server = mostlyai.mock.mcp_server: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