worqhat 3.8.0__py3-none-any.whl → 3.10.0__py3-none-any.whl

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
@@ -16,14 +16,20 @@ from .tables import (
     AsyncTablesResourceWithStreamingResponse,
 )
 from ...types import (
+    db_cluster_params,
+    db_recommend_params,
+    db_find_similar_params,
     db_execute_batch_params,
     db_execute_query_params,
+    db_hybrid_search_params,
     db_insert_record_params,
     db_delete_records_params,
     db_update_records_params,
+    db_semantic_search_params,
+    db_detect_anomalies_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
@@ -34,11 +40,17 @@ from ..._response import (
     async_to_streamed_response_wrapper,
 )
 from ..._base_client import make_request_options
+from ...types.db_cluster_response import DBClusterResponse
+from ...types.db_recommend_response import DBRecommendResponse
+from ...types.db_find_similar_response import DBFindSimilarResponse
 from ...types.db_execute_batch_response import DBExecuteBatchResponse
 from ...types.db_execute_query_response import DBExecuteQueryResponse
+from ...types.db_hybrid_search_response import DBHybridSearchResponse
 from ...types.db_insert_record_response import DBInsertRecordResponse
 from ...types.db_delete_records_response import DBDeleteRecordsResponse
 from ...types.db_update_records_response import DBUpdateRecordsResponse
+from ...types.db_semantic_search_response import DBSemanticSearchResponse
+from ...types.db_detect_anomalies_response import DBDetectAnomaliesResponse
 from ...types.db_process_nl_query_response import DBProcessNlQueryResponse
 
 __all__ = ["DBResource", "AsyncDBResource"]
@@ -68,6 +80,68 @@ class DBResource(SyncAPIResource):
         """
         return DBResourceWithStreamingResponse(self)
 
+    def cluster(
+        self,
+        *,
+        table: str,
+        environment: Literal["development", "staging", "production"] | Omit = omit,
+        generate_labels: bool | Omit = omit,
+        max_clusters: float | Omit = omit,
+        min_clusters: float | Omit = omit,
+        num_clusters: float | 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,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> DBClusterResponse:
+        """
+        Performs K-means clustering on vector embeddings to automatically group similar
+        records. Supports auto-detection of optimal cluster count and AI-generated
+        cluster labels. Returns cluster information including centroids, sample records,
+        and quality metrics.
+
+        Args:
+          table: Table to cluster
+
+          environment: Environment to cluster (development, staging, production)
+
+          generate_labels: Whether to generate AI labels for clusters
+
+          max_clusters: Maximum clusters for auto-detection
+
+          min_clusters: Minimum clusters for auto-detection
+
+          num_clusters: Number of clusters (auto-detected if not provided)
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return self._post(
+            "/db/cluster",
+            body=maybe_transform(
+                {
+                    "table": table,
+                    "environment": environment,
+                    "generate_labels": generate_labels,
+                    "max_clusters": max_clusters,
+                    "min_clusters": min_clusters,
+                    "num_clusters": num_clusters,
+                },
+                db_cluster_params.DBClusterParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=DBClusterResponse,
+        )
+
     def delete_records(
         self,
         *,
@@ -116,6 +190,64 @@ class DBResource(SyncAPIResource):
             cast_to=DBDeleteRecordsResponse,
         )
 
+    def detect_anomalies(
+        self,
+        *,
+        table: str,
+        environment: Literal["development", "staging", "production"] | Omit = omit,
+        k: float | Omit = omit,
+        limit: float | Omit = omit,
+        threshold: float | 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,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> DBDetectAnomaliesResponse:
+        """
+        Identifies anomalous or outlier records using K-nearest neighbors analysis on
+        vector embeddings. Useful for fraud detection, data quality checks, and
+        identifying unusual patterns. Returns anomaly scores and nearest neighbors for
+        each detected anomaly.
+
+        Args:
+          table: Table to analyze for anomalies
+
+          environment: Environment to analyze (development, staging, production)
+
+          k: Number of nearest neighbors to consider
+
+          limit: Maximum number of anomalies to return
+
+          threshold: Minimum anomaly score threshold
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return self._post(
+            "/db/detect-anomalies",
+            body=maybe_transform(
+                {
+                    "table": table,
+                    "environment": environment,
+                    "k": k,
+                    "limit": limit,
+                    "threshold": threshold,
+                },
+                db_detect_anomalies_params.DBDetectAnomaliesParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=DBDetectAnomaliesResponse,
+        )
+
     def execute_batch(
         self,
         *,
@@ -170,7 +302,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 +310,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
 
@@ -215,6 +348,138 @@ class DBResource(SyncAPIResource):
             cast_to=DBExecuteQueryResponse,
         )
 
+    def find_similar(
+        self,
+        *,
+        record_id: Union[str, float],
+        table: str,
+        environment: Literal["development", "staging", "production"] | Omit = omit,
+        exclude_self: bool | Omit = omit,
+        limit: float | Omit = omit,
+        target_table: str | Omit = omit,
+        threshold: float | 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,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> DBFindSimilarResponse:
+        """
+        Finds records similar to a specific existing record using vector embeddings.
+        Useful for "More like this" functionality and cross-table similarity search.
+        Returns similarity scores and supports excluding the source record.
+
+        Args:
+          record_id: ID of the source record
+
+          table: Table containing the source record
+
+          environment: Environment to search in (development, staging, production)
+
+          exclude_self: Whether to exclude the source record from results
+
+          limit: Maximum number of similar records to return
+
+          target_table: Different table to search in (optional)
+
+          threshold: Minimum similarity score threshold
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return self._post(
+            "/db/find-similar",
+            body=maybe_transform(
+                {
+                    "record_id": record_id,
+                    "table": table,
+                    "environment": environment,
+                    "exclude_self": exclude_self,
+                    "limit": limit,
+                    "target_table": target_table,
+                    "threshold": threshold,
+                },
+                db_find_similar_params.DBFindSimilarParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=DBFindSimilarResponse,
+        )
+
+    def hybrid_search(
+        self,
+        *,
+        query: str,
+        table: str,
+        environment: Literal["development", "staging", "production"] | Omit = omit,
+        keyword_weight: float | Omit = omit,
+        limit: float | Omit = omit,
+        semantic_weight: float | Omit = omit,
+        text_columns: SequenceNotStr[str] | 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,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> DBHybridSearchResponse:
+        """Combines semantic search using vector embeddings with traditional keyword
+        search.
+
+        Provides weighted scoring between semantic similarity and keyword
+        matching. Ideal for queries that need both contextual understanding and exact
+        term matching.
+
+        Args:
+          query: Search query combining natural language and keywords
+
+          table: Table to search in
+
+          environment: Environment to search in (development, staging, production)
+
+          keyword_weight: Weight for keyword matching score
+
+          limit: Maximum number of results to return
+
+          semantic_weight: Weight for semantic similarity score
+
+          text_columns: Columns to include in keyword search
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return self._post(
+            "/db/hybrid-search",
+            body=maybe_transform(
+                {
+                    "query": query,
+                    "table": table,
+                    "environment": environment,
+                    "keyword_weight": keyword_weight,
+                    "limit": limit,
+                    "semantic_weight": semantic_weight,
+                    "text_columns": text_columns,
+                },
+                db_hybrid_search_params.DBHybridSearchParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=DBHybridSearchResponse,
+        )
+
     def insert_record(
         self,
         *,
@@ -268,7 +533,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 +549,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 +566,6 @@ class DBResource(SyncAPIResource):
             body=maybe_transform(
                 {
                     "question": question,
-                    "table": table,
                     "context": context,
                     "environment": environment,
                 },
@@ -316,6 +577,137 @@ class DBResource(SyncAPIResource):
             cast_to=DBProcessNlQueryResponse,
         )
 
+    def recommend(
+        self,
+        *,
+        table: str,
+        environment: Literal["development", "staging", "production"] | Omit = omit,
+        exclude_ids: SequenceNotStr[str] | Omit = omit,
+        limit: float | Omit = omit,
+        record_id: Union[str, float] | Omit = omit,
+        strategy: Literal["similar", "diverse", "popular"] | Omit = omit,
+        user_history: SequenceNotStr[str] | 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,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> DBRecommendResponse:
+        """
+        Generates item recommendations using vector embeddings and collaborative
+        filtering. Supports multiple recommendation strategies including similar items,
+        diverse recommendations, and user history-based recommendations.
+
+        Args:
+          table: Table to generate recommendations from
+
+          environment: Environment to search in (development, staging, production)
+
+          exclude_ids: Record IDs to exclude from recommendations
+
+          limit: Maximum number of recommendations to return
+
+          record_id: Source item ID for item-to-item recommendations
+
+          strategy: Recommendation strategy to use
+
+          user_history: Array of record IDs the user has interacted with
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return self._post(
+            "/db/recommend",
+            body=maybe_transform(
+                {
+                    "table": table,
+                    "environment": environment,
+                    "exclude_ids": exclude_ids,
+                    "limit": limit,
+                    "record_id": record_id,
+                    "strategy": strategy,
+                    "user_history": user_history,
+                },
+                db_recommend_params.DBRecommendParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=DBRecommendResponse,
+        )
+
+    def semantic_search(
+        self,
+        *,
+        query: str,
+        environment: Literal["development", "staging", "production"] | Omit = omit,
+        filters: Dict[str, object] | Omit = omit,
+        limit: float | Omit = omit,
+        table: str | Omit = omit,
+        tables: SequenceNotStr[str] | Omit = omit,
+        threshold: float | 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,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> DBSemanticSearchResponse:
+        """
+        Performs semantic search across database tables using vector embeddings.
+        Supports both single table and cross-table searches with configurable similarity
+        thresholds. Returns records with similarity scores and metadata about matched
+        fields.
+
+        Args:
+          query: Natural language search query
+
+          environment: Environment to search in (development, staging, production)
+
+          filters: Additional WHERE conditions to apply
+
+          limit: Maximum number of results to return
+
+          table: Single table to search in (optional if tables is provided)
+
+          tables: Multiple tables to search across (optional if table is provided)
+
+          threshold: Minimum similarity score threshold
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return self._post(
+            "/db/semantic-search",
+            body=maybe_transform(
+                {
+                    "query": query,
+                    "environment": environment,
+                    "filters": filters,
+                    "limit": limit,
+                    "table": table,
+                    "tables": tables,
+                    "threshold": threshold,
+                },
+                db_semantic_search_params.DBSemanticSearchParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=DBSemanticSearchResponse,
+        )
+
     def update_records(
         self,
         *,
@@ -393,6 +785,68 @@ class AsyncDBResource(AsyncAPIResource):
         """
         return AsyncDBResourceWithStreamingResponse(self)
 
+    async def cluster(
+        self,
+        *,
+        table: str,
+        environment: Literal["development", "staging", "production"] | Omit = omit,
+        generate_labels: bool | Omit = omit,
+        max_clusters: float | Omit = omit,
+        min_clusters: float | Omit = omit,
+        num_clusters: float | 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,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> DBClusterResponse:
+        """
+        Performs K-means clustering on vector embeddings to automatically group similar
+        records. Supports auto-detection of optimal cluster count and AI-generated
+        cluster labels. Returns cluster information including centroids, sample records,
+        and quality metrics.
+
+        Args:
+          table: Table to cluster
+
+          environment: Environment to cluster (development, staging, production)
+
+          generate_labels: Whether to generate AI labels for clusters
+
+          max_clusters: Maximum clusters for auto-detection
+
+          min_clusters: Minimum clusters for auto-detection
+
+          num_clusters: Number of clusters (auto-detected if not provided)
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return await self._post(
+            "/db/cluster",
+            body=await async_maybe_transform(
+                {
+                    "table": table,
+                    "environment": environment,
+                    "generate_labels": generate_labels,
+                    "max_clusters": max_clusters,
+                    "min_clusters": min_clusters,
+                    "num_clusters": num_clusters,
+                },
+                db_cluster_params.DBClusterParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=DBClusterResponse,
+        )
+
     async def delete_records(
         self,
         *,
@@ -441,6 +895,64 @@ class AsyncDBResource(AsyncAPIResource):
             cast_to=DBDeleteRecordsResponse,
         )
 
+    async def detect_anomalies(
+        self,
+        *,
+        table: str,
+        environment: Literal["development", "staging", "production"] | Omit = omit,
+        k: float | Omit = omit,
+        limit: float | Omit = omit,
+        threshold: float | 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,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> DBDetectAnomaliesResponse:
+        """
+        Identifies anomalous or outlier records using K-nearest neighbors analysis on
+        vector embeddings. Useful for fraud detection, data quality checks, and
+        identifying unusual patterns. Returns anomaly scores and nearest neighbors for
+        each detected anomaly.
+
+        Args:
+          table: Table to analyze for anomalies
+
+          environment: Environment to analyze (development, staging, production)
+
+          k: Number of nearest neighbors to consider
+
+          limit: Maximum number of anomalies to return
+
+          threshold: Minimum anomaly score threshold
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return await self._post(
+            "/db/detect-anomalies",
+            body=await async_maybe_transform(
+                {
+                    "table": table,
+                    "environment": environment,
+                    "k": k,
+                    "limit": limit,
+                    "threshold": threshold,
+                },
+                db_detect_anomalies_params.DBDetectAnomaliesParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=DBDetectAnomaliesResponse,
+        )
+
     async def execute_batch(
         self,
         *,
@@ -495,7 +1007,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 +1015,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
 
@@ -540,6 +1053,138 @@ class AsyncDBResource(AsyncAPIResource):
             cast_to=DBExecuteQueryResponse,
         )
 
+    async def find_similar(
+        self,
+        *,
+        record_id: Union[str, float],
+        table: str,
+        environment: Literal["development", "staging", "production"] | Omit = omit,
+        exclude_self: bool | Omit = omit,
+        limit: float | Omit = omit,
+        target_table: str | Omit = omit,
+        threshold: float | 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,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> DBFindSimilarResponse:
+        """
+        Finds records similar to a specific existing record using vector embeddings.
+        Useful for "More like this" functionality and cross-table similarity search.
+        Returns similarity scores and supports excluding the source record.
+
+        Args:
+          record_id: ID of the source record
+
+          table: Table containing the source record
+
+          environment: Environment to search in (development, staging, production)
+
+          exclude_self: Whether to exclude the source record from results
+
+          limit: Maximum number of similar records to return
+
+          target_table: Different table to search in (optional)
+
+          threshold: Minimum similarity score threshold
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return await self._post(
+            "/db/find-similar",
+            body=await async_maybe_transform(
+                {
+                    "record_id": record_id,
+                    "table": table,
+                    "environment": environment,
+                    "exclude_self": exclude_self,
+                    "limit": limit,
+                    "target_table": target_table,
+                    "threshold": threshold,
+                },
+                db_find_similar_params.DBFindSimilarParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=DBFindSimilarResponse,
+        )
+
+    async def hybrid_search(
+        self,
+        *,
+        query: str,
+        table: str,
+        environment: Literal["development", "staging", "production"] | Omit = omit,
+        keyword_weight: float | Omit = omit,
+        limit: float | Omit = omit,
+        semantic_weight: float | Omit = omit,
+        text_columns: SequenceNotStr[str] | 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,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> DBHybridSearchResponse:
+        """Combines semantic search using vector embeddings with traditional keyword
+        search.
+
+        Provides weighted scoring between semantic similarity and keyword
+        matching. Ideal for queries that need both contextual understanding and exact
+        term matching.
+
+        Args:
+          query: Search query combining natural language and keywords
+
+          table: Table to search in
+
+          environment: Environment to search in (development, staging, production)
+
+          keyword_weight: Weight for keyword matching score
+
+          limit: Maximum number of results to return
+
+          semantic_weight: Weight for semantic similarity score
+
+          text_columns: Columns to include in keyword search
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return await self._post(
+            "/db/hybrid-search",
+            body=await async_maybe_transform(
+                {
+                    "query": query,
+                    "table": table,
+                    "environment": environment,
+                    "keyword_weight": keyword_weight,
+                    "limit": limit,
+                    "semantic_weight": semantic_weight,
+                    "text_columns": text_columns,
+                },
+                db_hybrid_search_params.DBHybridSearchParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=DBHybridSearchResponse,
+        )
+
     async def insert_record(
         self,
         *,
@@ -593,7 +1238,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 +1254,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 +1271,6 @@ class AsyncDBResource(AsyncAPIResource):
             body=await async_maybe_transform(
                 {
                     "question": question,
-                    "table": table,
                     "context": context,
                     "environment": environment,
                 },
@@ -641,6 +1282,137 @@ class AsyncDBResource(AsyncAPIResource):
             cast_to=DBProcessNlQueryResponse,
         )
 
+    async def recommend(
+        self,
+        *,
+        table: str,
+        environment: Literal["development", "staging", "production"] | Omit = omit,
+        exclude_ids: SequenceNotStr[str] | Omit = omit,
+        limit: float | Omit = omit,
+        record_id: Union[str, float] | Omit = omit,
+        strategy: Literal["similar", "diverse", "popular"] | Omit = omit,
+        user_history: SequenceNotStr[str] | 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,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> DBRecommendResponse:
+        """
+        Generates item recommendations using vector embeddings and collaborative
+        filtering. Supports multiple recommendation strategies including similar items,
+        diverse recommendations, and user history-based recommendations.
+
+        Args:
+          table: Table to generate recommendations from
+
+          environment: Environment to search in (development, staging, production)
+
+          exclude_ids: Record IDs to exclude from recommendations
+
+          limit: Maximum number of recommendations to return
+
+          record_id: Source item ID for item-to-item recommendations
+
+          strategy: Recommendation strategy to use
+
+          user_history: Array of record IDs the user has interacted with
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return await self._post(
+            "/db/recommend",
+            body=await async_maybe_transform(
+                {
+                    "table": table,
+                    "environment": environment,
+                    "exclude_ids": exclude_ids,
+                    "limit": limit,
+                    "record_id": record_id,
+                    "strategy": strategy,
+                    "user_history": user_history,
+                },
+                db_recommend_params.DBRecommendParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=DBRecommendResponse,
+        )
+
+    async def semantic_search(
+        self,
+        *,
+        query: str,
+        environment: Literal["development", "staging", "production"] | Omit = omit,
+        filters: Dict[str, object] | Omit = omit,
+        limit: float | Omit = omit,
+        table: str | Omit = omit,
+        tables: SequenceNotStr[str] | Omit = omit,
+        threshold: float | 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,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> DBSemanticSearchResponse:
+        """
+        Performs semantic search across database tables using vector embeddings.
+        Supports both single table and cross-table searches with configurable similarity
+        thresholds. Returns records with similarity scores and metadata about matched
+        fields.
+
+        Args:
+          query: Natural language search query
+
+          environment: Environment to search in (development, staging, production)
+
+          filters: Additional WHERE conditions to apply
+
+          limit: Maximum number of results to return
+
+          table: Single table to search in (optional if tables is provided)
+
+          tables: Multiple tables to search across (optional if table is provided)
+
+          threshold: Minimum similarity score threshold
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return await self._post(
+            "/db/semantic-search",
+            body=await async_maybe_transform(
+                {
+                    "query": query,
+                    "environment": environment,
+                    "filters": filters,
+                    "limit": limit,
+                    "table": table,
+                    "tables": tables,
+                    "threshold": threshold,
+                },
+                db_semantic_search_params.DBSemanticSearchParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=DBSemanticSearchResponse,
+        )
+
     async def update_records(
         self,
         *,
@@ -698,21 +1470,39 @@ class DBResourceWithRawResponse:
     def __init__(self, db: DBResource) -> None:
         self._db = db
 
+        self.cluster = to_raw_response_wrapper(
+            db.cluster,
+        )
         self.delete_records = to_raw_response_wrapper(
             db.delete_records,
         )
+        self.detect_anomalies = to_raw_response_wrapper(
+            db.detect_anomalies,
+        )
         self.execute_batch = to_raw_response_wrapper(
             db.execute_batch,
         )
         self.execute_query = to_raw_response_wrapper(
             db.execute_query,
         )
+        self.find_similar = to_raw_response_wrapper(
+            db.find_similar,
+        )
+        self.hybrid_search = to_raw_response_wrapper(
+            db.hybrid_search,
+        )
         self.insert_record = to_raw_response_wrapper(
             db.insert_record,
         )
         self.process_nl_query = to_raw_response_wrapper(
             db.process_nl_query,
         )
+        self.recommend = to_raw_response_wrapper(
+            db.recommend,
+        )
+        self.semantic_search = to_raw_response_wrapper(
+            db.semantic_search,
+        )
         self.update_records = to_raw_response_wrapper(
             db.update_records,
         )
@@ -726,21 +1516,39 @@ class AsyncDBResourceWithRawResponse:
     def __init__(self, db: AsyncDBResource) -> None:
         self._db = db
 
+        self.cluster = async_to_raw_response_wrapper(
+            db.cluster,
+        )
         self.delete_records = async_to_raw_response_wrapper(
             db.delete_records,
         )
+        self.detect_anomalies = async_to_raw_response_wrapper(
+            db.detect_anomalies,
+        )
         self.execute_batch = async_to_raw_response_wrapper(
             db.execute_batch,
         )
         self.execute_query = async_to_raw_response_wrapper(
             db.execute_query,
         )
+        self.find_similar = async_to_raw_response_wrapper(
+            db.find_similar,
+        )
+        self.hybrid_search = async_to_raw_response_wrapper(
+            db.hybrid_search,
+        )
         self.insert_record = async_to_raw_response_wrapper(
             db.insert_record,
         )
         self.process_nl_query = async_to_raw_response_wrapper(
             db.process_nl_query,
         )
+        self.recommend = async_to_raw_response_wrapper(
+            db.recommend,
+        )
+        self.semantic_search = async_to_raw_response_wrapper(
+            db.semantic_search,
+        )
         self.update_records = async_to_raw_response_wrapper(
             db.update_records,
         )
@@ -754,21 +1562,39 @@ class DBResourceWithStreamingResponse:
     def __init__(self, db: DBResource) -> None:
         self._db = db
 
+        self.cluster = to_streamed_response_wrapper(
+            db.cluster,
+        )
         self.delete_records = to_streamed_response_wrapper(
             db.delete_records,
         )
+        self.detect_anomalies = to_streamed_response_wrapper(
+            db.detect_anomalies,
+        )
         self.execute_batch = to_streamed_response_wrapper(
             db.execute_batch,
         )
         self.execute_query = to_streamed_response_wrapper(
             db.execute_query,
         )
+        self.find_similar = to_streamed_response_wrapper(
+            db.find_similar,
+        )
+        self.hybrid_search = to_streamed_response_wrapper(
+            db.hybrid_search,
+        )
         self.insert_record = to_streamed_response_wrapper(
             db.insert_record,
         )
         self.process_nl_query = to_streamed_response_wrapper(
             db.process_nl_query,
         )
+        self.recommend = to_streamed_response_wrapper(
+            db.recommend,
+        )
+        self.semantic_search = to_streamed_response_wrapper(
+            db.semantic_search,
+        )
         self.update_records = to_streamed_response_wrapper(
             db.update_records,
         )
@@ -782,21 +1608,39 @@ class AsyncDBResourceWithStreamingResponse:
     def __init__(self, db: AsyncDBResource) -> None:
         self._db = db
 
+        self.cluster = async_to_streamed_response_wrapper(
+            db.cluster,
+        )
         self.delete_records = async_to_streamed_response_wrapper(
             db.delete_records,
         )
+        self.detect_anomalies = async_to_streamed_response_wrapper(
+            db.detect_anomalies,
+        )
         self.execute_batch = async_to_streamed_response_wrapper(
             db.execute_batch,
         )
         self.execute_query = async_to_streamed_response_wrapper(
             db.execute_query,
         )
+        self.find_similar = async_to_streamed_response_wrapper(
+            db.find_similar,
+        )
+        self.hybrid_search = async_to_streamed_response_wrapper(
+            db.hybrid_search,
+        )
         self.insert_record = async_to_streamed_response_wrapper(
             db.insert_record,
         )
         self.process_nl_query = async_to_streamed_response_wrapper(
             db.process_nl_query,
         )
+        self.recommend = async_to_streamed_response_wrapper(
+            db.recommend,
+        )
+        self.semantic_search = async_to_streamed_response_wrapper(
+            db.semantic_search,
+        )
         self.update_records = async_to_streamed_response_wrapper(
             db.update_records,
         )