worqhat 3.8.0__tar.gz → 3.9.0__tar.gz

worqhat-3.9.0/.release-please-manifest.json

@@ -0,0 +1,3 @@
+{
+  ".": "3.9.0"
+}

{worqhat-3.8.0 → worqhat-3.9.0}/CHANGELOG.md

@@ -1,5 +1,14 @@
 # Changelog
 
+## 3.9.0 (2025-10-24)
+
+Full Changelog: [v3.8.0...v3.9.0](https://github.com/WorqHat/worqhat-python-sdk/compare/v3.8.0...v3.9.0)
+
+### Features
+
+* **api:** manual updates ([586bf23](https://github.com/WorqHat/worqhat-python-sdk/commit/586bf230bbe49f665d3dfc5a5c01be30c6ae56c4))
+* **api:** manual updates ([af79b06](https://github.com/WorqHat/worqhat-python-sdk/commit/af79b065dff0f6d7d41312f7c197f433a7ab5489))
+
 ## 3.8.0 (2025-10-19)
 
 Full Changelog: [v3.7.0...v3.8.0](https://github.com/WorqHat/worqhat-python-sdk/compare/v3.7.0...v3.8.0)

{worqhat-3.8.0 → worqhat-3.9.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: worqhat
-Version: 3.8.0
+Version: 3.9.0
 Summary: The official Python library for the worqhat API
 Project-URL: Homepage, https://github.com/WorqHat/worqhat-python-sdk
 Project-URL: Repository, https://github.com/WorqHat/worqhat-python-sdk

{worqhat-3.8.0 → worqhat-3.9.0}/api.md

@@ -82,14 +82,12 @@ Methods:
 - <code title="post /flows/file/{flowId}">client.flows.<a href="./src/worqhat/resources/flows.py">trigger_with_file</a>(flow_id, \*\*<a href="src/worqhat/types/flow_trigger_with_file_params.py">params</a>) -> <a href="./src/worqhat/types/flow_trigger_with_file_response.py">FlowTriggerWithFileResponse</a></code>
 - <code title="post /flows/trigger/{flowId}">client.flows.<a href="./src/worqhat/resources/flows.py">trigger_with_payload</a>(flow_id, \*\*<a href="src/worqhat/types/flow_trigger_with_payload_params.py">params</a>) -> <a href="./src/worqhat/types/flow_trigger_with_payload_response.py">FlowTriggerWithPayloadResponse</a></code>
 
-# Client
-
-## Storage
+# Storage
 
 Types:
 
 ```python
-from worqhat.types.client import (
+from worqhat.types import (
     StorageDeleteFileByIDResponse,
     StorageRetrieveFileByIDResponse,
     StorageRetrieveFileByPathResponse,
@@ -99,7 +97,7 @@ from worqhat.types.client import (
 
 Methods:
 
-- <code title="delete /storage/delete/{fileId}">client.client.storage.<a href="./src/worqhat/resources/client/storage.py">delete_file_by_id</a>(file_id) -> <a href="./src/worqhat/types/client/storage_delete_file_by_id_response.py">StorageDeleteFileByIDResponse</a></code>
-- <code title="get /storage/fetch/{fileId}">client.client.storage.<a href="./src/worqhat/resources/client/storage.py">retrieve_file_by_id</a>(file_id) -> <a href="./src/worqhat/types/client/storage_retrieve_file_by_id_response.py">StorageRetrieveFileByIDResponse</a></code>
-- <code title="get /storage/fetch-by-path">client.client.storage.<a href="./src/worqhat/resources/client/storage.py">retrieve_file_by_path</a>(\*\*<a href="src/worqhat/types/client/storage_retrieve_file_by_path_params.py">params</a>) -> <a href="./src/worqhat/types/client/storage_retrieve_file_by_path_response.py">StorageRetrieveFileByPathResponse</a></code>
-- <code title="post /storage/upload">client.client.storage.<a href="./src/worqhat/resources/client/storage.py">upload_file</a>(\*\*<a href="src/worqhat/types/client/storage_upload_file_params.py">params</a>) -> <a href="./src/worqhat/types/client/storage_upload_file_response.py">StorageUploadFileResponse</a></code>
+- <code title="delete /storage/delete/{fileId}">client.storage.<a href="./src/worqhat/resources/storage.py">delete_file_by_id</a>(file_id) -> <a href="./src/worqhat/types/storage_delete_file_by_id_response.py">StorageDeleteFileByIDResponse</a></code>
+- <code title="get /storage/fetch/{fileId}">client.storage.<a href="./src/worqhat/resources/storage.py">retrieve_file_by_id</a>(file_id) -> <a href="./src/worqhat/types/storage_retrieve_file_by_id_response.py">StorageRetrieveFileByIDResponse</a></code>
+- <code title="get /storage/fetch-by-path">client.storage.<a href="./src/worqhat/resources/storage.py">retrieve_file_by_path</a>(\*\*<a href="src/worqhat/types/storage_retrieve_file_by_path_params.py">params</a>) -> <a href="./src/worqhat/types/storage_retrieve_file_by_path_response.py">StorageRetrieveFileByPathResponse</a></code>
+- <code title="post /storage/upload">client.storage.<a href="./src/worqhat/resources/storage.py">upload_file</a>(\*\*<a href="src/worqhat/types/storage_upload_file_params.py">params</a>) -> <a href="./src/worqhat/types/storage_upload_file_response.py">StorageUploadFileResponse</a></code>

{worqhat-3.8.0 → worqhat-3.9.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "worqhat"
-version = "3.8.0"
+version = "3.9.0"
 description = "The official Python library for the worqhat API"
 dynamic = ["readme"]
 license = "Apache-2.0"

{worqhat-3.8.0 → worqhat-3.9.0}/src/worqhat/_client.py

@@ -30,7 +30,7 @@ from ._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from .resources import flows, health
+from .resources import flows, health, storage
 from ._streaming import Stream as Stream, AsyncStream as AsyncStream
 from ._exceptions import WorqhatError, APIStatusError
 from ._base_client import (
@@ -40,7 +40,6 @@ from ._base_client import (
     make_request_options,
 )
 from .resources.db import db
-from .resources.client import client
 from .types.get_server_info_response import GetServerInfoResponse
 
 __all__ = ["Timeout", "Transport", "ProxiesTypes", "RequestOptions", "Worqhat", "AsyncWorqhat", "Client", "AsyncClient"]
@@ -50,7 +49,7 @@ class Worqhat(SyncAPIClient):
     db: db.DBResource
     health: health.HealthResource
     flows: flows.FlowsResource
-    client: client.ClientResource
+    storage: storage.StorageResource
     with_raw_response: WorqhatWithRawResponse
     with_streaming_response: WorqhatWithStreamedResponse
 
@@ -111,7 +110,7 @@ class Worqhat(SyncAPIClient):
         self.db = db.DBResource(self)
         self.health = health.HealthResource(self)
         self.flows = flows.FlowsResource(self)
-        self.client = client.ClientResource(self)
+        self.storage = storage.StorageResource(self)
         self.with_raw_response = WorqhatWithRawResponse(self)
         self.with_streaming_response = WorqhatWithStreamedResponse(self)
 
@@ -243,7 +242,7 @@ class AsyncWorqhat(AsyncAPIClient):
     db: db.AsyncDBResource
     health: health.AsyncHealthResource
     flows: flows.AsyncFlowsResource
-    client: client.AsyncClientResource
+    storage: storage.AsyncStorageResource
     with_raw_response: AsyncWorqhatWithRawResponse
     with_streaming_response: AsyncWorqhatWithStreamedResponse
 
@@ -304,7 +303,7 @@ class AsyncWorqhat(AsyncAPIClient):
         self.db = db.AsyncDBResource(self)
         self.health = health.AsyncHealthResource(self)
         self.flows = flows.AsyncFlowsResource(self)
-        self.client = client.AsyncClientResource(self)
+        self.storage = storage.AsyncStorageResource(self)
         self.with_raw_response = AsyncWorqhatWithRawResponse(self)
         self.with_streaming_response = AsyncWorqhatWithStreamedResponse(self)
 
@@ -437,7 +436,7 @@ class WorqhatWithRawResponse:
         self.db = db.DBResourceWithRawResponse(client.db)
         self.health = health.HealthResourceWithRawResponse(client.health)
         self.flows = flows.FlowsResourceWithRawResponse(client.flows)
-        self.client = client.ClientResourceWithRawResponse(client.client)
+        self.storage = storage.StorageResourceWithRawResponse(client.storage)
 
         self.get_server_info = to_raw_response_wrapper(
             client.get_server_info,
@@ -449,7 +448,7 @@ class AsyncWorqhatWithRawResponse:
         self.db = db.AsyncDBResourceWithRawResponse(client.db)
         self.health = health.AsyncHealthResourceWithRawResponse(client.health)
         self.flows = flows.AsyncFlowsResourceWithRawResponse(client.flows)
-        self.client = client.AsyncClientResourceWithRawResponse(client.client)
+        self.storage = storage.AsyncStorageResourceWithRawResponse(client.storage)
 
         self.get_server_info = async_to_raw_response_wrapper(
             client.get_server_info,
@@ -461,7 +460,7 @@ class WorqhatWithStreamedResponse:
         self.db = db.DBResourceWithStreamingResponse(client.db)
         self.health = health.HealthResourceWithStreamingResponse(client.health)
         self.flows = flows.FlowsResourceWithStreamingResponse(client.flows)
-        self.client = client.ClientResourceWithStreamingResponse(client.client)
+        self.storage = storage.StorageResourceWithStreamingResponse(client.storage)
 
         self.get_server_info = to_streamed_response_wrapper(
             client.get_server_info,
@@ -473,7 +472,7 @@ class AsyncWorqhatWithStreamedResponse:
         self.db = db.AsyncDBResourceWithStreamingResponse(client.db)
         self.health = health.AsyncHealthResourceWithStreamingResponse(client.health)
         self.flows = flows.AsyncFlowsResourceWithStreamingResponse(client.flows)
-        self.client = client.AsyncClientResourceWithStreamingResponse(client.client)
+        self.storage = storage.AsyncStorageResourceWithStreamingResponse(client.storage)
 
         self.get_server_info = async_to_streamed_response_wrapper(
             client.get_server_info,

{worqhat-3.8.0 → worqhat-3.9.0}/src/worqhat/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "worqhat"
-__version__ = "3.8.0"  # x-release-please-version
+__version__ = "3.9.0"  # x-release-please-version

{worqhat-3.8.0 → worqhat-3.9.0}/src/worqhat/resources/__init__.py

@@ -16,14 +16,6 @@ from .flows import (
     FlowsResourceWithStreamingResponse,
     AsyncFlowsResourceWithStreamingResponse,
 )
-from .client import (
-    ClientResource,
-    AsyncClientResource,
-    ClientResourceWithRawResponse,
-    AsyncClientResourceWithRawResponse,
-    ClientResourceWithStreamingResponse,
-    AsyncClientResourceWithStreamingResponse,
-)
 from .health import (
     HealthResource,
     AsyncHealthResource,
@@ -32,6 +24,14 @@ from .health import (
     HealthResourceWithStreamingResponse,
     AsyncHealthResourceWithStreamingResponse,
 )
+from .storage import (
+    StorageResource,
+    AsyncStorageResource,
+    StorageResourceWithRawResponse,
+    AsyncStorageResourceWithRawResponse,
+    StorageResourceWithStreamingResponse,
+    AsyncStorageResourceWithStreamingResponse,
+)
 
 __all__ = [
     "DBResource",
@@ -52,10 +52,10 @@ __all__ = [
     "AsyncFlowsResourceWithRawResponse",
     "FlowsResourceWithStreamingResponse",
     "AsyncFlowsResourceWithStreamingResponse",
-    "ClientResource",
-    "AsyncClientResource",
-    "ClientResourceWithRawResponse",
-    "AsyncClientResourceWithRawResponse",
-    "ClientResourceWithStreamingResponse",
-    "AsyncClientResourceWithStreamingResponse",
+    "StorageResource",
+    "AsyncStorageResource",
+    "StorageResourceWithRawResponse",
+    "AsyncStorageResourceWithRawResponse",
+    "StorageResourceWithStreamingResponse",
+    "AsyncStorageResourceWithStreamingResponse",
 ]

{worqhat-3.8.0 → worqhat-3.9.0}/src/worqhat/resources/db/db.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import Dict, Iterable
+from typing import Dict, Union, Iterable
 from typing_extensions import Literal
 
 import httpx
@@ -23,7 +23,7 @@ from ...types import (
     db_update_records_params,
     db_process_nl_query_params,
 )
-from ..._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
+from ..._types import Body, Omit, Query, Headers, NotGiven, SequenceNotStr, omit, not_given
 from ..._utils import maybe_transform, async_maybe_transform
 from ..._compat import cached_property
 from ..._resource import SyncAPIResource, AsyncAPIResource
@@ -170,7 +170,7 @@ class DBResource(SyncAPIResource):
         *,
         query: str,
         environment: Literal["development", "staging", "production"] | Omit = omit,
-        params: Dict[str, object] | Omit = omit,
+        params: Union[Dict[str, object], SequenceNotStr[Union[str, float, bool]]] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -178,18 +178,19 @@ class DBResource(SyncAPIResource):
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> DBExecuteQueryResponse:
-        """Executes a raw SQL query directly against ClickHouse (WorqDB).
+        """Executes a raw SQL query directly against the database.
 
-        This endpoint
-        provides direct SQL access with security guardrails to prevent destructive
-        operations.
+        Supports both named
+        parameters ({param}) and positional parameters ($1, $2). Provides security
+        guardrails to prevent destructive operations.
 
         Args:
-          query: SQL query to execute
+          query: SQL query to execute. Supports both named parameters ({param}) and positional
+              parameters ($1, $2)
 
           environment: Environment to query (development, staging, production)
 
-          params: Optional query parameters
+          params: Named parameters for queries with {param} syntax
 
           extra_headers: Send extra headers
 
@@ -268,7 +269,6 @@ class DBResource(SyncAPIResource):
         self,
         *,
         question: str,
-        table: str,
         context: Dict[str, object] | Omit = omit,
         environment: Literal["development", "staging", "production"] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -285,8 +285,6 @@ class DBResource(SyncAPIResource):
         Args:
           question: Natural language question
 
-          table: Table name to query
-
           context: Optional context for the query
 
           environment: Environment to query (development, staging, production)
@@ -304,7 +302,6 @@ class DBResource(SyncAPIResource):
             body=maybe_transform(
                 {
                     "question": question,
-                    "table": table,
                     "context": context,
                     "environment": environment,
                 },
@@ -495,7 +492,7 @@ class AsyncDBResource(AsyncAPIResource):
         *,
         query: str,
         environment: Literal["development", "staging", "production"] | Omit = omit,
-        params: Dict[str, object] | Omit = omit,
+        params: Union[Dict[str, object], SequenceNotStr[Union[str, float, bool]]] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -503,18 +500,19 @@ class AsyncDBResource(AsyncAPIResource):
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> DBExecuteQueryResponse:
-        """Executes a raw SQL query directly against ClickHouse (WorqDB).
+        """Executes a raw SQL query directly against the database.
 
-        This endpoint
-        provides direct SQL access with security guardrails to prevent destructive
-        operations.
+        Supports both named
+        parameters ({param}) and positional parameters ($1, $2). Provides security
+        guardrails to prevent destructive operations.
 
         Args:
-          query: SQL query to execute
+          query: SQL query to execute. Supports both named parameters ({param}) and positional
+              parameters ($1, $2)
 
           environment: Environment to query (development, staging, production)
 
-          params: Optional query parameters
+          params: Named parameters for queries with {param} syntax
 
           extra_headers: Send extra headers
 
@@ -593,7 +591,6 @@ class AsyncDBResource(AsyncAPIResource):
         self,
         *,
         question: str,
-        table: str,
         context: Dict[str, object] | Omit = omit,
         environment: Literal["development", "staging", "production"] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -610,8 +607,6 @@ class AsyncDBResource(AsyncAPIResource):
         Args:
           question: Natural language question
 
-          table: Table name to query
-
           context: Optional context for the query
 
           environment: Environment to query (development, staging, production)
@@ -629,7 +624,6 @@ class AsyncDBResource(AsyncAPIResource):
             body=await async_maybe_transform(
                 {
                     "question": question,
-                    "table": table,
                     "context": context,
                     "environment": environment,
                 },

{worqhat-3.8.0/src/worqhat/resources/client → worqhat-3.9.0/src/worqhat/resources}/storage.py

@@ -6,22 +6,22 @@ from typing import Mapping, cast
 
 import httpx
 
-from ..._types import Body, Omit, Query, Headers, NotGiven, FileTypes, omit, not_given
-from ..._utils import extract_files, maybe_transform, deepcopy_minimal, async_maybe_transform
-from ..._compat import cached_property
-from ..._resource import SyncAPIResource, AsyncAPIResource
-from ..._response import (
+from ..types import storage_upload_file_params, storage_retrieve_file_by_path_params
+from .._types import Body, Omit, Query, Headers, NotGiven, FileTypes, omit, not_given
+from .._utils import extract_files, maybe_transform, deepcopy_minimal, async_maybe_transform
+from .._compat import cached_property
+from .._resource import SyncAPIResource, AsyncAPIResource
+from .._response import (
     to_raw_response_wrapper,
     to_streamed_response_wrapper,
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ..._base_client import make_request_options
-from ...types.client import storage_upload_file_params, storage_retrieve_file_by_path_params
-from ...types.client.storage_upload_file_response import StorageUploadFileResponse
-from ...types.client.storage_delete_file_by_id_response import StorageDeleteFileByIDResponse
-from ...types.client.storage_retrieve_file_by_id_response import StorageRetrieveFileByIDResponse
-from ...types.client.storage_retrieve_file_by_path_response import StorageRetrieveFileByPathResponse
+from .._base_client import make_request_options
+from ..types.storage_upload_file_response import StorageUploadFileResponse
+from ..types.storage_delete_file_by_id_response import StorageDeleteFileByIDResponse
+from ..types.storage_retrieve_file_by_id_response import StorageRetrieveFileByIDResponse
+from ..types.storage_retrieve_file_by_path_response import StorageRetrieveFileByPathResponse
 
 __all__ = ["StorageResource", "AsyncStorageResource"]
 

{worqhat-3.8.0 → worqhat-3.9.0}/src/worqhat/types/__init__.py

@@ -17,8 +17,16 @@ from .flow_get_metrics_response import FlowGetMetricsResponse as FlowGetMetricsR
 from .db_delete_records_response import DBDeleteRecordsResponse as DBDeleteRecordsResponse
 from .db_process_nl_query_params import DBProcessNlQueryParams as DBProcessNlQueryParams
 from .db_update_records_response import DBUpdateRecordsResponse as DBUpdateRecordsResponse
+from .storage_upload_file_params import StorageUploadFileParams as StorageUploadFileParams
 from .db_process_nl_query_response import DBProcessNlQueryResponse as DBProcessNlQueryResponse
+from .storage_upload_file_response import StorageUploadFileResponse as StorageUploadFileResponse
 from .flow_trigger_with_file_params import FlowTriggerWithFileParams as FlowTriggerWithFileParams
 from .flow_trigger_with_file_response import FlowTriggerWithFileResponse as FlowTriggerWithFileResponse
 from .flow_trigger_with_payload_params import FlowTriggerWithPayloadParams as FlowTriggerWithPayloadParams
 from .flow_trigger_with_payload_response import FlowTriggerWithPayloadResponse as FlowTriggerWithPayloadResponse
+from .storage_delete_file_by_id_response import StorageDeleteFileByIDResponse as StorageDeleteFileByIDResponse
+from .storage_retrieve_file_by_id_response import StorageRetrieveFileByIDResponse as StorageRetrieveFileByIDResponse
+from .storage_retrieve_file_by_path_params import StorageRetrieveFileByPathParams as StorageRetrieveFileByPathParams
+from .storage_retrieve_file_by_path_response import (
+    StorageRetrieveFileByPathResponse as StorageRetrieveFileByPathResponse,
+)

{worqhat-3.8.0 → worqhat-3.9.0}/src/worqhat/types/db_execute_query_params.py

@@ -2,18 +2,23 @@
 
 from __future__ import annotations
 
-from typing import Dict
+from typing import Dict, Union
 from typing_extensions import Literal, Required, TypedDict
 
+from .._types import SequenceNotStr
+
 __all__ = ["DBExecuteQueryParams"]
 
 
 class DBExecuteQueryParams(TypedDict, total=False):
     query: Required[str]
-    """SQL query to execute"""
+    """SQL query to execute.
+
+    Supports both named parameters ({param}) and positional parameters ($1, $2)
+    """
 
     environment: Literal["development", "staging", "production"]
     """Environment to query (development, staging, production)"""
 
-    params: Dict[str, object]
-    """Optional query parameters"""
+    params: Union[Dict[str, object], SequenceNotStr[Union[str, float, bool]]]
+    """Named parameters for queries with {param} syntax"""

{worqhat-3.8.0 → worqhat-3.9.0}/src/worqhat/types/db_process_nl_query_params.py

@@ -12,9 +12,6 @@ class DBProcessNlQueryParams(TypedDict, total=False):
     question: Required[str]
     """Natural language question"""
 
-    table: Required[str]
-    """Table name to query"""
-
     context: Dict[str, object]
     """Optional context for the query"""
 

{worqhat-3.8.0/src/worqhat/types/client → worqhat-3.9.0/src/worqhat/types}/storage_delete_file_by_id_response.py

@@ -5,7 +5,7 @@ from datetime import datetime
 
 from pydantic import Field as FieldInfo
 
-from ..._models import BaseModel
+from .._models import BaseModel
 
 __all__ = ["StorageDeleteFileByIDResponse"]
 

{worqhat-3.8.0/src/worqhat/types/client → worqhat-3.9.0/src/worqhat/types}/storage_retrieve_file_by_id_response.py

@@ -5,7 +5,7 @@ from datetime import datetime
 
 from pydantic import Field as FieldInfo
 
-from ..._models import BaseModel
+from .._models import BaseModel
 
 __all__ = ["StorageRetrieveFileByIDResponse", "File"]
 

{worqhat-3.8.0/src/worqhat/types/client → worqhat-3.9.0/src/worqhat/types}/storage_retrieve_file_by_path_response.py

@@ -5,7 +5,7 @@ from datetime import datetime
 
 from pydantic import Field as FieldInfo
 
-from ..._models import BaseModel
+from .._models import BaseModel
 
 __all__ = ["StorageRetrieveFileByPathResponse", "File"]
 

{worqhat-3.8.0/src/worqhat/types/client → worqhat-3.9.0/src/worqhat/types}/storage_upload_file_params.py

@@ -4,7 +4,7 @@ from __future__ import annotations
 
 from typing_extensions import Required, TypedDict
 
-from ..._types import FileTypes
+from .._types import FileTypes
 
 __all__ = ["StorageUploadFileParams"]
 

{worqhat-3.8.0/src/worqhat/types/client → worqhat-3.9.0/src/worqhat/types}/storage_upload_file_response.py

@@ -5,7 +5,7 @@ from datetime import datetime
 
 from pydantic import Field as FieldInfo
 
-from ..._models import BaseModel
+from .._models import BaseModel
 
 __all__ = ["StorageUploadFileResponse", "File"]
 

{worqhat-3.8.0 → worqhat-3.9.0}/tests/api_resources/test_db.py

@@ -144,7 +144,7 @@ class TestDB:
     @parametrize
     def test_method_execute_query(self, client: Worqhat) -> None:
         db = client.db.execute_query(
-            query="SELECT * FROM users WHERE status = 'active' LIMIT 10",
+            query="SELECT * FROM users WHERE slug = {slug}",
         )
         assert_matches_type(DBExecuteQueryResponse, db, path=["response"])
 
@@ -152,9 +152,9 @@ class TestDB:
     @parametrize
     def test_method_execute_query_with_all_params(self, client: Worqhat) -> None:
         db = client.db.execute_query(
-            query="SELECT * FROM users WHERE status = 'active' LIMIT 10",
+            query="SELECT * FROM users WHERE slug = {slug}",
             environment="production",
-            params={"foo": "bar"},
+            params={"slug": "bar"},
         )
         assert_matches_type(DBExecuteQueryResponse, db, path=["response"])
 
@@ -162,7 +162,7 @@ class TestDB:
     @parametrize
     def test_raw_response_execute_query(self, client: Worqhat) -> None:
         response = client.db.with_raw_response.execute_query(
-            query="SELECT * FROM users WHERE status = 'active' LIMIT 10",
+            query="SELECT * FROM users WHERE slug = {slug}",
         )
 
         assert response.is_closed is True
@@ -174,7 +174,7 @@ class TestDB:
     @parametrize
     def test_streaming_response_execute_query(self, client: Worqhat) -> None:
         with client.db.with_streaming_response.execute_query(
-            query="SELECT * FROM users WHERE status = 'active' LIMIT 10",
+            query="SELECT * FROM users WHERE slug = {slug}",
         ) as response:
             assert not response.is_closed
             assert response.http_request.headers.get("X-Stainless-Lang") == "python"
@@ -248,7 +248,6 @@ class TestDB:
     def test_method_process_nl_query(self, client: Worqhat) -> None:
         db = client.db.process_nl_query(
             question="How many active users do we have?",
-            table="users",
         )
         assert_matches_type(DBProcessNlQueryResponse, db, path=["response"])
 
@@ -257,7 +256,6 @@ class TestDB:
     def test_method_process_nl_query_with_all_params(self, client: Worqhat) -> None:
         db = client.db.process_nl_query(
             question="How many active users do we have?",
-            table="users",
             context={"foo": "bar"},
             environment="production",
         )
@@ -268,7 +266,6 @@ class TestDB:
     def test_raw_response_process_nl_query(self, client: Worqhat) -> None:
         response = client.db.with_raw_response.process_nl_query(
             question="How many active users do we have?",
-            table="users",
         )
 
         assert response.is_closed is True
@@ -281,7 +278,6 @@ class TestDB:
     def test_streaming_response_process_nl_query(self, client: Worqhat) -> None:
         with client.db.with_streaming_response.process_nl_query(
             question="How many active users do we have?",
-            table="users",
         ) as response:
             assert not response.is_closed
             assert response.http_request.headers.get("X-Stainless-Lang") == "python"
@@ -468,7 +464,7 @@ class TestAsyncDB:
     @parametrize
     async def test_method_execute_query(self, async_client: AsyncWorqhat) -> None:
         db = await async_client.db.execute_query(
-            query="SELECT * FROM users WHERE status = 'active' LIMIT 10",
+            query="SELECT * FROM users WHERE slug = {slug}",
         )
         assert_matches_type(DBExecuteQueryResponse, db, path=["response"])
 
@@ -476,9 +472,9 @@ class TestAsyncDB:
     @parametrize
     async def test_method_execute_query_with_all_params(self, async_client: AsyncWorqhat) -> None:
         db = await async_client.db.execute_query(
-            query="SELECT * FROM users WHERE status = 'active' LIMIT 10",
+            query="SELECT * FROM users WHERE slug = {slug}",
             environment="production",
-            params={"foo": "bar"},
+            params={"slug": "bar"},
         )
         assert_matches_type(DBExecuteQueryResponse, db, path=["response"])
 
@@ -486,7 +482,7 @@ class TestAsyncDB:
     @parametrize
     async def test_raw_response_execute_query(self, async_client: AsyncWorqhat) -> None:
         response = await async_client.db.with_raw_response.execute_query(
-            query="SELECT * FROM users WHERE status = 'active' LIMIT 10",
+            query="SELECT * FROM users WHERE slug = {slug}",
         )
 
         assert response.is_closed is True
@@ -498,7 +494,7 @@ class TestAsyncDB:
     @parametrize
     async def test_streaming_response_execute_query(self, async_client: AsyncWorqhat) -> None:
         async with async_client.db.with_streaming_response.execute_query(
-            query="SELECT * FROM users WHERE status = 'active' LIMIT 10",
+            query="SELECT * FROM users WHERE slug = {slug}",
         ) as response:
             assert not response.is_closed
             assert response.http_request.headers.get("X-Stainless-Lang") == "python"
@@ -572,7 +568,6 @@ class TestAsyncDB:
     async def test_method_process_nl_query(self, async_client: AsyncWorqhat) -> None:
         db = await async_client.db.process_nl_query(
             question="How many active users do we have?",
-            table="users",
         )
         assert_matches_type(DBProcessNlQueryResponse, db, path=["response"])
 
@@ -581,7 +576,6 @@ class TestAsyncDB:
     async def test_method_process_nl_query_with_all_params(self, async_client: AsyncWorqhat) -> None:
         db = await async_client.db.process_nl_query(
             question="How many active users do we have?",
-            table="users",
             context={"foo": "bar"},
             environment="production",
         )
@@ -592,7 +586,6 @@ class TestAsyncDB:
     async def test_raw_response_process_nl_query(self, async_client: AsyncWorqhat) -> None:
         response = await async_client.db.with_raw_response.process_nl_query(
             question="How many active users do we have?",
-            table="users",
         )
 
         assert response.is_closed is True
@@ -605,7 +598,6 @@ class TestAsyncDB:
     async def test_streaming_response_process_nl_query(self, async_client: AsyncWorqhat) -> None:
         async with async_client.db.with_streaming_response.process_nl_query(
             question="How many active users do we have?",
-            table="users",
         ) as response:
             assert not response.is_closed
             assert response.http_request.headers.get("X-Stainless-Lang") == "python"