Merge pull request #35 from JigsawStack/feat/text-2-sql

add database param to text2sql

Author: iamNarcisse

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):

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",