zaturn 0.1.1__py3-none-any.whl → 0.1.3__py3-none-any.whl

zaturn/config.py

@@ -47,7 +47,7 @@ if not source_list:
     print("No data sources provided. Loading example dataset for demonstration.")
     print(f"\nTo load your datasets, add them to {SOURCES_FILE} (one source URL or full file path per line)")
     print("\nOr use command line args to specify data sources:")
-    print("uv run --directory /path/to/zaturn mcp_server.py sqlite:///path/to/mydata.db")
+    print("zaturn_mcp sqlite:///path/to/mydata.db /path/to/my_file.csv")
     print(f"\nNOTE: Sources in command line args will be ignored if sources are found in {SOURCES_FILE}")
     
 SOURCES = {}
@@ -88,7 +88,6 @@ for s in source_list:
 
 # Other Settings
 RETURN_IMAGES = not args.noimg
-print(SOURCES)
 
 
 

zaturn/core.py

@@ -20,15 +20,17 @@ def list_sources() -> str:
         
         result = "Available data sources:\n\n"
         for source in config.SOURCES:
-            result += f"- {source}\n"
+            tables = _list_tables(source)
+            if type(tables) is List:
+                tables = ', '.join(tables)
+            result += f"- {source}\nHas tables: {tables}\n"
     
         return result
     except Exception as e:
         return str(e)
 
 
-@mcp.tool()
-def list_tables(source_id: str):
+def _list_tables(source_id: str):
     """
     Lists names of all tables/datasets in a given data source.
     Use run_query with appropriate SQL query to determine table structure
@@ -67,6 +69,43 @@ def list_tables(source_id: str):
     except Exception as e:
         return str(e)
 
+@mcp.tool()
+def describe_table(source_id: str, table_name: str) -> str:
+    """
+    Lists columns and their types in the specified table of specified data source.
+
+    Args:
+        source_id: The data source
+        table_name: The table in the data source
+    """
+    try:
+        source = config.SOURCES.get(source_id)
+        if not source:
+            return f"Source {source_id} Not Found"
+
+        match source['type']:
+            case 'sqlite':
+                result = query_utils.execute_query(source,
+                    f"PRAGMA table_info({table_name});"
+                )
+                return result.to_markdown(index=False)
+                
+            case 'postgresql':
+                result = query_utils.execute_query(source,
+                    f"SELECT * FROM INFORMATION_SCHEMA.COLUMNS WHERE table_name = '{table_name}';"
+                )
+                return result.to_markdown(index=False)
+                
+            case "mysql" | "duckdb" | "csv" | "parquet":
+                result = query_utils.execute_query(source,
+                    f"DESCRIBE {table_name};"
+                )
+                return result.to_markdown(index=False)
+    
+    except Exception as e:
+        return str(e)
+            
+
 @mcp.tool()
 def run_query(source_id: str, query: str) -> str:
     """

{zaturn-0.1.1.dist-info → zaturn-0.1.3.dist-info}/METADATA

@@ -1,13 +1,11 @@
 Metadata-Version: 2.4
 Name: zaturn
-Version: 0.1.1
+Version: 0.1.3
 Summary: AI Data Analysis MCP
 Author-email: Karthik Devan <krtdvn@gmail.com>
 Maintainer-email: Karthik Devan <krtdvn@gmail.com>
 Project-URL: Homepage, https://github.com/kdqed/zaturn
-Project-URL: Changelog, https://github.com/kdqed/zaturn/releases
 Project-URL: Issues, https://github.com/kdqed/zaturn/issues
-Project-URL: CI, https://github.com/kdqed/zaturn/actions
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE
@@ -27,15 +25,21 @@ Requires-Dist: tabulate>=0.9.0
 Dynamic: license-file
 
 <p align="center">
-  <img src="brand/logo.png" width="128" height="128">
+  <img src="https://github.com/kdqed/zaturn/raw/main/brand/logo.png" width="128" height="128">
 </p>
 
 # Zaturn: Your Co-Pilot For Data Analytics & BI
 
+<a href="https://glama.ai/mcp/servers/@kdqed/zaturn">
+  <img width="380" height="200" src="https://glama.ai/mcp/servers/@kdqed/zaturn/badge" alt="Zaturn MCP server" />
+</a>
+
 https://github.com/user-attachments/assets/d42dc433-e5ec-4b3e-bef0-5cfc097396ab
 
 Zaturn is an open source, AI-powered data analysis/BI tool that can connect to your data sources, run SQL queries on it, and give you useful insights. Think of it like vibe data analysis, in the spirit of vibe coding. Currently Zaturn is available as an MCP (Model Context Protocol) Server that can be integrated into your favorite MCP Client (Claude, Cursor, etc.) A full fledged notebook interface is on the roadmap.
 
+[Join The Discord](https://discord.gg/K8mECeVzpQ)
+
 ## Features:
 
 ### Multiple Data Sources 
@@ -69,23 +73,19 @@ Using an MCP like Zaturn will keep your data where it is, and enable AI to draft
 ## Installation & Setup
 1. Install [uv](https://docs.astral.sh/uv/getting-started/installation/#installation-methods)
 
-2. Install Zaturn with uv:
+2. Install [Zaturn](https://pypi.org/project/zaturn/) with uv:
 ```bash
-$ uv tool install zaturn
+uv tool install zaturn
 ```
 
 3. Add to MCP config, with data sources:
 ```json
 "mcpServers": {
   "zaturn": {
-    "command": "uv",
+    "command": "zaturn_mcp",
     "args": [
-      "run",
-      "--directory",
-      "/path/to/downloaded/folder",
-      "mcp_server.py",
-      "mysql+pymysql://username:password@host:3306/dbname",
       "postgresql://username:password@host:port/dbname",
+      "mysql+pymysql://username:password@host:3306/dbname",
       "sqlite:////full/path/to/sample_dbs/northwind.db",
       "/full/path/to/sample_dbs/titanic.parquet",
       "/full/path/to/sample_dbs/ny_aq.csv",
@@ -95,10 +95,22 @@ $ uv tool install zaturn
 }
 ```
 
-If your MCP client does not support images, add the `--noimg` flag after `mcp_server.py`:
+OR add a `sources.txt` to the Zaturn config directory:
+```
+postgresql://username:password@host:port/dbname
+mysql+pymysql://username:password@host:3306/dbname
+sqlite:////full/path/to/sample_dbs/northwind.db
+/full/path/to/sample_dbs/titanic.parquet
+/full/path/to/sample_dbs/ny_aq.csv
+/full/path/to/sample_dbs/duckdb_sample.duckdb
+```
+
+This file needs to be at `~/.config/zaturn/sources.txt` on Linux/MacOS and at `%APPDATA%\zaturn\sources.txt` on Windows.
+
+If your MCP client does not support image rendering, add the `--noimg` argument:
 ```json
 ...
-      "mcp_server.py",
+    "args": [
       "--noimg",
       "mysql+pymysql://username:password@host:3306/dbname",
 ...
@@ -114,10 +126,6 @@ You are a helpful data analysis assistant. Use only the tool provided data sourc
 ```
 User: List the top 5 customers by revenue for Northwind
 AI: 
-[04/08/25 15:16:47] INFO     Processing request of type ListToolsRequest                                     server.py:534
-[04/08/25 15:16:51] INFO     Processing request of type CallToolRequest                                      server.py:534
-[04/08/25 15:16:53] INFO     Processing request of type CallToolRequest                                      server.py:534
-[04/08/25 15:16:55] INFO     Processing request of type CallToolRequest                                      server.py:534
 The top 5 customers by revenue for Northwind are:
 
 1. B's Beverages with a revenue of $6,154,115.34
@@ -146,9 +154,11 @@ Analyst:
 ```
 - A native notebook interface 
 
-If you have any specific requirements please feel free to raise an issue.
+## Support And Feedback
+
+[Raise an issue](https://github.com/kdqed/zaturn/issues) or [join the Discord](https://discord.gg/K8mECeVzpQ).
 
 
 ## Example Dataset Credits
 
-THe [pokemon dataset compiled by Sarah Taha and PokéAPI](https://www.kaggle.com/datasets/sarahtaha/1025-pokemon) has been included under the [CC BY-NC-SA 4.0](https://creativecommons.org/licenses/by-nc-sa/4.0/) license.
+The [pokemon dataset compiled by Sarah Taha and PokéAPI](https://www.kaggle.com/datasets/sarahtaha/1025-pokemon) has been included under the [CC BY-NC-SA 4.0](https://creativecommons.org/licenses/by-nc-sa/4.0/) license for demonstration purposes.

zaturn-0.1.3.dist-info/RECORD

@@ -0,0 +1,12 @@
+zaturn/__init__.py,sha256=v4t5fkRuIJFE-SBxCa5pBjZv0EoC0eWK75nU9iaa7Rg,267
+zaturn/config.py,sha256=gF5M6Agmixw2A4vpWqIF3ICVnClPeASA51dhp1bkk04,3221
+zaturn/core.py,sha256=9zcKb0FbkgGkDtWfBA6_O5NQf6GVKG69HrqOr5nhsLU,4641
+zaturn/query_utils.py,sha256=zyQjcRnPKGHZdf0XHzQeMxHw9vieZIwXhBbVGP87ml4,2801
+zaturn/visualizations.py,sha256=Dj08msgZ_DIMEVvJIlybT3cIA3GNkcXVOa9TJsHK2yo,4551
+zaturn/example_data/all_pokemon_data.csv,sha256=SUlGHHWbehuLg-ch1YUrQ6-xBtqHGw6rIkyn70fAgCk,130893
+zaturn-0.1.3.dist-info/licenses/LICENSE,sha256=mZSuFlbEBZGl0-8ULRMLdRDbhau5hrWRNQOjytYeaug,1070
+zaturn-0.1.3.dist-info/METADATA,sha256=BxES8Ifp0fusfDMeB2SNLpSUrZ2qsKWG8rgqhQnfD0Q,5961
+zaturn-0.1.3.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+zaturn-0.1.3.dist-info/entry_points.txt,sha256=N1UZC2zvod92_Brs4A2xZiAnt-iGLBNryglXfwhxfj4,43
+zaturn-0.1.3.dist-info/top_level.txt,sha256=KLUnwQwVZkfd5YCnnqR35MOOs8KLhanPGelvmRo2MVA,7
+zaturn-0.1.3.dist-info/RECORD,,

zaturn-0.1.1.dist-info/RECORD

@@ -1,12 +0,0 @@
-zaturn/__init__.py,sha256=v4t5fkRuIJFE-SBxCa5pBjZv0EoC0eWK75nU9iaa7Rg,267
-zaturn/config.py,sha256=Pq7mNScRnQ4PycDv3cqFVSK9Y9N9V9k3bdGgVo-P4Ns,3253
-zaturn/core.py,sha256=sn2oliWVDqhYSAwSe5VbhNamaKAsnlrT3oc6uzffzs4,3267
-zaturn/query_utils.py,sha256=zyQjcRnPKGHZdf0XHzQeMxHw9vieZIwXhBbVGP87ml4,2801
-zaturn/visualizations.py,sha256=Dj08msgZ_DIMEVvJIlybT3cIA3GNkcXVOa9TJsHK2yo,4551
-zaturn/example_data/all_pokemon_data.csv,sha256=SUlGHHWbehuLg-ch1YUrQ6-xBtqHGw6rIkyn70fAgCk,130893
-zaturn-0.1.1.dist-info/licenses/LICENSE,sha256=mZSuFlbEBZGl0-8ULRMLdRDbhau5hrWRNQOjytYeaug,1070
-zaturn-0.1.1.dist-info/METADATA,sha256=etCpT5a2AxDSP_hc7_ic6ecEd_aZYGsQNdvzTWtbQbU,5834
-zaturn-0.1.1.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-zaturn-0.1.1.dist-info/entry_points.txt,sha256=N1UZC2zvod92_Brs4A2xZiAnt-iGLBNryglXfwhxfj4,43
-zaturn-0.1.1.dist-info/top_level.txt,sha256=KLUnwQwVZkfd5YCnnqR35MOOs8KLhanPGelvmRo2MVA,7
-zaturn-0.1.1.dist-info/RECORD,,