zaturn 0.2.1__py3-none-any.whl → 0.2.2__py3-none-any.whl

zaturn/mcp/__init__.py

@@ -1,10 +1,11 @@
 import argparse
+import importlib.resources
 import os
 import platformdirs
-import pkg_resources
 import sys
 
 from fastmcp import FastMCP
+from fastmcp.tools.tool import Tool
 
 from zaturn.tools import ZaturnTools
 
@@ -30,19 +31,17 @@ if not source_list:
     source_list = args.sources
 
 if not source_list:
-    source_list = [
-        pkg_resources.resource_filename(
-            'zaturn',
-            os.path.join('mcp', 'example_data', 'all_pokemon_data.csv')
-        )
-    ]
+    with importlib.resources.path(
+        'zaturn.tools.example_data', 'all_pokemon_data.csv'
+    ) as source_path:
+        source_list = [str(source_path)]
+    
     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("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 = {}
 for s in source_list:
     source = s.lower()
@@ -90,8 +89,8 @@ for s in source_list:
 def ZaturnMCP(sources):
     zaturn_tools = ZaturnTools(sources)
     zaturn_mcp = FastMCP()
-    for tool in zaturn_tools.tools:
-        zaturn_mcp.add_tool(tool)
+    for tool_function in zaturn_tools.tools:
+        zaturn_mcp.add_tool(Tool.from_function(tool_function))
 
     return zaturn_mcp
 

zaturn/tools/config.py

@@ -1,8 +1,5 @@
-import argparse
 import os
 import platformdirs
-import pkg_resources
-import sys
 
 # Basic Setup
 USER_DATA_DIR = platformdirs.user_data_dir('zaturn', 'zaturn')
@@ -13,83 +10,3 @@ SOURCES_FILE = os.path.join(USER_DATA_DIR, 'sources.txt')
 os.makedirs(QUERIES_DIR, exist_ok=True)
 os.makedirs(VISUALS_DIR, exist_ok=True)
 
-# Parse command line args
-parser = argparse.ArgumentParser(
-    description="Zaturn: A read-only BI tool for analyzing various data sources"
-)
-parser.add_argument('--noimg', action='store_const', 
-    const=True, default=False, 
-    help='Return image file paths instead of images for visuals. Use when MCP client cannot render images.',
-)
-parser.add_argument('sources', nargs=argparse.REMAINDER, default=[], 
-    help='Data source (can be specified multiple times). Can be SQLite, MySQL, PostgreSQL connection string, or a path to CSV, Parquet, or DuckDB file.'
-)
-args = parser.parse_args()
-
-# Read and parse sources
-source_list = []
-if os.path.exists(SOURCES_FILE):
-    with open(SOURCES_FILE) as f:
-        source_list = [line.strip('\n') for line in f.readlines() if line.strip('\n')]
-
-if not source_list:
-    source_list = args.sources
-
-if not source_list:
-    source_list = [
-        pkg_resources.resource_filename(
-            'zaturn',
-            os.path.join('mcp', 'example_data', 'all_pokemon_data.csv')
-        )
-    ]
-    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("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}")
-    
-CLI_SOURCES = {}
-for s in source_list:
-    source = s.lower()
-    if source.startswith('sqlite://'):
-        source_type = 'sqlite'
-        source_name = source.split('/')[-1].split('?')[0].split('.db')[0]
-    elif source.startswith('postgresql://'):
-        source_type = 'postgresql'
-        source_name = source.split('/')[-1].split('?')[0]
-    elif source.startswith("mysql://") or source.startswith("mysql+pymysql://"):
-        source_type = 'mysql'
-        s = s.replace('mysql://', 'mysql+pymysql://')
-        source_name = source.split('/')[-1].split('?')[0]
-    elif source.startswith('clickhouse://'):
-        source_type = 'clickhouse'
-        source_name = source.split('/')[-1].split('?')[0]
-    elif source.endswith(".duckdb"):
-        source_type = "duckdb"
-        source_name = source.split('/')[-1].split('.')[0]
-    elif source.endswith(".csv"):
-        source_type = "csv"
-        source_name = source.split('/')[-1].split('.')[0]
-    elif source.endswith(".parquet") or source.endswith(".pq"):
-        source_type = "parquet"
-        source_name = source.split('/')[-1].split('.')[0]
-    else:
-        continue
-
-    source_id = f'{source_name}-{source_type}'
-    if source_id in CLI_SOURCES:
-        i = 2
-        while True:
-            source_id = f'{source_name}{i}-{source_type}'
-            if source_id not in SOURCES:
-                break
-            i += 1
-
-    CLI_SOURCES[source_id] = {'url': s, 'type': source_type}
-
-
-# Other Settings
-CLI_RETURN_IMAGES = not args.noimg
-
-
-

{zaturn-0.2.1.dist-info → zaturn-0.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: zaturn
-Version: 0.2.1
+Version: 0.2.2
 Summary: AI Data Analysis MCP & Studio
 Author-email: Karthik Devan <krtdvn@gmail.com>
 Maintainer-email: Karthik Devan <krtdvn@gmail.com>
@@ -78,7 +78,7 @@ More visualization capabilities are being added.
 
 ## Installation & Setup
 
-See [https://zaturn.pro/install](https://zaturn.pro/install)
+See [https://zaturn.pro/docs/install](https://zaturn.pro/docs/install)
 
 
 ## Roadmap
@@ -98,7 +98,6 @@ Analyst:
   I can get it done right now. Actually, you can do it right now.
   You know what? The boss can do it right now.
 ```
-- A native notebook interface 
 
 ## Help And Feedback
 

{zaturn-0.2.1.dist-info → zaturn-0.2.2.dist-info}/RECORD

@@ -1,4 +1,4 @@
-zaturn/mcp/__init__.py,sha256=lBfs5Ztpih-Dn5vuTxJUTzjD2SM2icAZZM9GgRVozYU,3399
+zaturn/mcp/__init__.py,sha256=8pTcOyJ3tK7TJcP4HBDVQA_MVuMrfYJwa8sWKfCvXIE,3466
 zaturn/studio/__init__.py,sha256=A1QxWydtsE0VmsJlvF7O9Fx1-WKjV3MBPje7knFSDzw,83
 zaturn/studio/agent_wrapper.py,sha256=4mEMRfE1Tfv63D7OIabqUbEJdIgO2yo8UXO6PKvNrmI,4603
 zaturn/studio/app.py,sha256=LEoSjs5wpZGX1B7pcJ2e_6CaxOFjbO6JitxlkJv__30,9208
@@ -26,14 +26,14 @@ zaturn/studio/templates/setup_prompt.html,sha256=K8l3ewlEyrUFiMmw2qPyDFbFNkq1RCB
 zaturn/studio/templates/user_message.html,sha256=WbvdVr4AVzVx6MG0eQ_ulMeHB9rpScZc-EZlCoYFEgw,90
 zaturn/studio/templates/css/style.css,sha256=cKImLAjLp6XjHdT16v9yJS0DHG2wWx8djR18SHfMqvE,6264
 zaturn/tools/__init__.py,sha256=vOcHneRuZll9k-2e3sra7m8qfdc4IoAFYc2g_xK79HE,262
-zaturn/tools/config.py,sha256=8YCwFp1emhEfbbJd_KTQ_Nrl4yt9tJoDZHgMXatJGjk,3382
+zaturn/tools/config.py,sha256=Ia-WzK_G6vQqB7norvaffb3fKLjtmXupVqAWsxdu6Gs,354
 zaturn/tools/core.py,sha256=4fHxSIbO22cTIejJk02GzUBfWimh1Ql9mZ3JgI1qLBY,2893
 zaturn/tools/query_utils.py,sha256=OnXbONmEgsHcScrJ76b7P3BRLOKgi8TMkhxBcLQK9cY,5306
 zaturn/tools/visualizations.py,sha256=aPXCN1YIjCz6uiwJjduKCiLLtnBFX92hTvjy5EqQYk8,9509
 zaturn/tools/example_data/all_pokemon_data.csv,sha256=SUlGHHWbehuLg-ch1YUrQ6-xBtqHGw6rIkyn70fAgCk,130893
-zaturn-0.2.1.dist-info/licenses/LICENSE,sha256=mZSuFlbEBZGl0-8ULRMLdRDbhau5hrWRNQOjytYeaug,1070
-zaturn-0.2.1.dist-info/METADATA,sha256=NsCeZRR_EKj5TekqsM4R4fgicuUbwWf29PxHm5PQ_7k,3999
-zaturn-0.2.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-zaturn-0.2.1.dist-info/entry_points.txt,sha256=MWMWX0dE_ZQM4StGKCLxJym_D91F82RcBi2LBj0hEho,82
-zaturn-0.2.1.dist-info/top_level.txt,sha256=KLUnwQwVZkfd5YCnnqR35MOOs8KLhanPGelvmRo2MVA,7
-zaturn-0.2.1.dist-info/RECORD,,
+zaturn-0.2.2.dist-info/licenses/LICENSE,sha256=mZSuFlbEBZGl0-8ULRMLdRDbhau5hrWRNQOjytYeaug,1070
+zaturn-0.2.2.dist-info/METADATA,sha256=BSnrqH9k7MNfqmRVPEO6q3884y-8Ym1X6ctklZz6NZc,3978
+zaturn-0.2.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+zaturn-0.2.2.dist-info/entry_points.txt,sha256=MWMWX0dE_ZQM4StGKCLxJym_D91F82RcBi2LBj0hEho,82
+zaturn-0.2.2.dist-info/top_level.txt,sha256=KLUnwQwVZkfd5YCnnqR35MOOs8KLhanPGelvmRo2MVA,7
+zaturn-0.2.2.dist-info/RECORD,,