add to base sl client && plumbing

Author: marcodamore

dbtsl/api/graphql/client/asyncio.pyi

@@ -22,6 +22,7 @@ class AsyncGraphQLClient:
         auth_token: str,
         url_format: Optional[str] = None,
         timeout: Optional[Union[TimeoutOptions, float, int]] = None,
+        client_partner_source: Optional[str] = None,
     ) -> None: ...
     def session(self) -> AbstractAsyncContextManager[AsyncIterator[Self]]: ...
     @property

dbtsl/api/graphql/client/base.py

@@ -146,6 +146,7 @@ def __call__(
         auth_token: str,
         url_format: Optional[str] = None,
         timeout: Optional[Union[TimeoutOptions, float, int]] = None,
+        client_partner_source: Optional[str] = None,
     ) -> TClient:
         """Initialize the Semantic Layer client.
 
@@ -155,5 +156,6 @@ def __call__(
             auth_token: the API auth token
             url_format: the URL format string to construct the final URL with
             timeout: `TimeoutOptions` or total timeout
+            client_partner_source: Pass a dbt partner source header for traffic source tracking
         """
         pass

dbtsl/api/graphql/client/sync.pyi

@@ -22,6 +22,7 @@ class SyncGraphQLClient:
         auth_token: str,
         url_format: Optional[str] = None,
         timeout: Optional[Union[TimeoutOptions, float, int]] = None,
+        client_partner_source: Optional[str] = None,
     ) -> None: ...
     def session(self) -> AbstractContextManager[Iterator[Self]]: ...
     @property

dbtsl/client/asyncio.py

@@ -26,6 +26,7 @@ def __init__(
         auth_token: str,
         host: str,
         timeout: Optional[Union[TimeoutOptions, float, int]] = None,
+        client_partner_source: Optional[str] = None,
     ) -> None:
         """Initialize the Semantic Layer client.
 

dbtsl/client/asyncio.pyi

@@ -15,6 +15,7 @@ class AsyncSemanticLayerClient:
         auth_token: str,
         host: str,
         timeout: Optional[Union[TimeoutOptions, float, int]] = None,
+        client_partner_source: Optional[str] = None,
     ) -> None: ...
     @overload
     async def compile_sql(

dbtsl/client/base.py

@@ -40,7 +40,7 @@ def __init__(
         gql_factory: GraphQLClientFactory[TGQLClient],
         adbc_factory: ADBCClientFactory[TADBCClient],
         timeout: Optional[Union[TimeoutOptions, float, int]] = None,
-        requests_headers: Optional[dict[str, str]] = None,
+        client_partner_source: Optional[str] = None,
     ) -> None:
         """Initialize the Semantic Layer client.
 
@@ -51,7 +51,7 @@ def __init__(
             gql_factory: class of the underlying GQL client
             adbc_factory: class of the underlying ADBC client
             timeout: `TimeoutOptions` or total timeout for the underlying GraphQL client.
-            requests_headers: additional headers to pass to the requests, optional
+            client_partner_source: Pass a dbt partner source header for traffic source tracking
         """
         self._has_session = False
 
@@ -63,6 +63,7 @@ def __init__(
             auth_token=auth_token,
             url_format=env.GRAPHQL_URL_FORMAT,
             timeout=timeout,
+            client_partner_source=client_partner_source,
         )
         self._adbc = adbc_factory(
             server_host=host,

dbtsl/client/sync.py

@@ -26,6 +26,7 @@ def __init__(
         auth_token: str,
         host: str,
         timeout: Optional[Union[TimeoutOptions, float, int]] = None,
+        client_partner_source: Optional[str] = None,
     ) -> None:
         """Initialize the Semantic Layer client.
 

dbtsl/client/sync.pyi

@@ -15,6 +15,7 @@ class SyncSemanticLayerClient:
         auth_token: str,
         host: str,
         timeout: Optional[Union[TimeoutOptions, float, int]] = None,
+        client_partner_source: Optional[str] = None,
     ) -> None: ...
     @overload
     def compile_sql(