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

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.2.dist-info → zaturn-0.1.3.dist-info}/METADATA

@@ -1,12 +1,11 @@
 Metadata-Version: 2.4
 Name: zaturn
-Version: 0.1.2
+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: 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
@@ -31,10 +30,16 @@ Dynamic: license-file
 
 # 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 
@@ -149,7 +154,9 @@ 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

{zaturn-0.1.2.dist-info → zaturn-0.1.3.dist-info}/RECORD

@@ -1,12 +1,12 @@
 zaturn/__init__.py,sha256=v4t5fkRuIJFE-SBxCa5pBjZv0EoC0eWK75nU9iaa7Rg,267
 zaturn/config.py,sha256=gF5M6Agmixw2A4vpWqIF3ICVnClPeASA51dhp1bkk04,3221
-zaturn/core.py,sha256=sn2oliWVDqhYSAwSe5VbhNamaKAsnlrT3oc6uzffzs4,3267
+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.2.dist-info/licenses/LICENSE,sha256=mZSuFlbEBZGl0-8ULRMLdRDbhau5hrWRNQOjytYeaug,1070
-zaturn-0.1.2.dist-info/METADATA,sha256=fyeW0ljonWfL4krOO9Wfm_4etUXRRcl47JeZ_KphRI4,5731
-zaturn-0.1.2.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-zaturn-0.1.2.dist-info/entry_points.txt,sha256=N1UZC2zvod92_Brs4A2xZiAnt-iGLBNryglXfwhxfj4,43
-zaturn-0.1.2.dist-info/top_level.txt,sha256=KLUnwQwVZkfd5YCnnqR35MOOs8KLhanPGelvmRo2MVA,7
-zaturn-0.1.2.dist-info/RECORD,,
+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,,