feat: add typing for groupby-only queries

This commit adds typing to the groupby only queries.
dbt-labs · Nov 12, 2024 · bacafba · bacafba
1 parent b583741
commit bacafba
Show file tree

Hide file tree

Showing 4 changed files with 72 additions and 0 deletions.
diff --git a/dbtsl/api/graphql/client/asyncio.pyi b/dbtsl/api/graphql/client/asyncio.pyi
@@ -55,6 +55,15 @@ class AsyncGraphQLClient:
         read_cache: bool = True,
     ) -> str: ...
     @overload
+    async def compile_sql(
+        self,
+        group_by: List[str],
+        limit: Optional[int] = None,
+        order_by: Optional[List[Union[str, OrderByGroupBy]]] = None,
+        where: Optional[List[str]] = None,
+        read_cache: bool = True,
+    ) -> str: ...
+    @overload
     async def compile_sql(
         self,
         saved_query: str,
@@ -78,6 +87,15 @@ class AsyncGraphQLClient:
         read_cache: bool = True,
     ) -> "pa.Table": ...
     @overload
+    async def query(
+        self,
+        group_by: List[str],
+        limit: Optional[int] = None,
+        order_by: Optional[List[Union[str, OrderByGroupBy]]] = None,
+        where: Optional[List[str]] = None,
+        read_cache: bool = True,
+    ) -> "pa.Table": ...
+    @overload
     async def query(
         self,
         saved_query: str,

diff --git a/dbtsl/api/graphql/client/sync.pyi b/dbtsl/api/graphql/client/sync.pyi
@@ -55,6 +55,15 @@ class SyncGraphQLClient:
         read_cache: bool = True,
     ) -> str: ...
     @overload
+    def compile_sql(
+        self,
+        group_by: List[str],
+        limit: Optional[int] = None,
+        order_by: Optional[List[Union[str, OrderByGroupBy]]] = None,
+        where: Optional[List[str]] = None,
+        read_cache: bool = True,
+    ) -> str: ...
+    @overload
     def compile_sql(
         self,
         saved_query: str,
@@ -78,6 +87,15 @@ class SyncGraphQLClient:
         read_cache: bool = True,
     ) -> "pa.Table": ...
     @overload
+    def query(
+        self,
+        group_by: List[str],
+        limit: Optional[int] = None,
+        order_by: Optional[List[Union[str, OrderByGroupBy]]] = None,
+        where: Optional[List[str]] = None,
+        read_cache: bool = True,
+    ) -> "pa.Table": ...
+    @overload
     def query(
         self,
         saved_query: str,

diff --git a/dbtsl/client/asyncio.pyi b/dbtsl/client/asyncio.pyi
@@ -25,6 +25,15 @@ class AsyncSemanticLayerClient:
         read_cache: bool = True,
     ) -> str: ...
     @overload
+    async def compile_sql(
+        self,
+        group_by: List[str],
+        limit: Optional[int] = None,
+        order_by: Optional[List[Union[str, OrderByGroupBy]]] = None,
+        where: Optional[List[str]] = None,
+        read_cache: bool = True,
+    ) -> str: ...
+    @overload
     async def compile_sql(
         self,
         saved_query: str,
@@ -48,6 +57,15 @@ class AsyncSemanticLayerClient:
         read_cache: bool = True,
     ) -> "pa.Table": ...
     @overload
+    async def query(
+        self,
+        group_by: List[str],
+        limit: Optional[int] = None,
+        order_by: Optional[List[Union[str, OrderByGroupBy]]] = None,
+        where: Optional[List[str]] = None,
+        read_cache: bool = True,
+    ) -> "pa.Table": ...
+    @overload
     async def query(
         self,
         saved_query: str,

diff --git a/dbtsl/client/sync.pyi b/dbtsl/client/sync.pyi
@@ -25,6 +25,15 @@ class SyncSemanticLayerClient:
         read_cache: bool = True,
     ) -> str: ...
     @overload
+    def compile_sql(
+        self,
+        group_by: List[str],
+        limit: Optional[int] = None,
+        order_by: Optional[List[Union[str, OrderByGroupBy]]] = None,
+        where: Optional[List[str]] = None,
+        read_cache: bool = True,
+    ) -> str: ...
+    @overload
     def compile_sql(
         self,
         saved_query: str,
@@ -48,6 +57,15 @@ class SyncSemanticLayerClient:
         read_cache: bool = True,
     ) -> "pa.Table": ...
     @overload
+    def query(
+        self,
+        group_by: List[str],
+        limit: Optional[int] = None,
+        order_by: Optional[List[Union[str, OrderByGroupBy]]] = None,
+        where: Optional[List[str]] = None,
+        read_cache: bool = True,
+    ) -> "pa.Table": ...
+    @overload
     def query(
         self,
         saved_query: str,