From d18459eae87add7b3f278d9ac98f590f1e87d8c6 Mon Sep 17 00:00:00 2001 From: Narcisse Date: Tue, 25 Feb 2025 21:16:40 +0100 Subject: [PATCH] add database param to text2sql --- jigsawstack/sql.py | 6 +++++- setup.py | 2 +- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/jigsawstack/sql.py b/jigsawstack/sql.py index 334acf8..69c224a 100644 --- a/jigsawstack/sql.py +++ b/jigsawstack/sql.py @@ -1,4 +1,4 @@ -from typing import Any, Dict, List, Union, cast +from typing import Any, Dict, List, Union, cast, Literal from typing_extensions import NotRequired, TypedDict from .request import Request, RequestConfig from .async_request import AsyncRequest @@ -22,6 +22,10 @@ class SQLParams(TypedDict): """ The key used to store the database schema on Jigsawstack file Storage. Not required if sql_schema is specified. """ + database: NotRequired[Literal["mysql", "postgresql", "sqlite"]] + """ + The type of database for the SQL query (mysql, postgresql, sqlite). + """ class SQLResponse(TypedDict): diff --git a/setup.py b/setup.py index feb3eb9..1219799 100644 --- a/setup.py +++ b/setup.py @@ -6,7 +6,7 @@ setup( name="jigsawstack", - version="0.1.28", + version="0.1.29", description="JigsawStack Python SDK", long_description=open("README.md", encoding="utf8").read(), long_description_content_type="text/markdown",