databricks-sdk 0.64.0__py3-none-any.whl → 0.66.0__py3-none-any.whl

databricks/sdk/service/dashboards.py

@@ -263,6 +263,9 @@ class GenieAttachment:
     query: Optional[GenieQueryAttachment] = None
     """Query Attachment if Genie responds with a SQL query"""
 
+    suggested_questions: Optional[GenieSuggestedQuestionsAttachment] = None
+    """Follow-up questions suggested by Genie"""
+
     text: Optional[TextAttachment] = None
     """Text Attachment if Genie responds with text"""
 
@@ -273,6 +276,8 @@ class GenieAttachment:
             body["attachment_id"] = self.attachment_id
         if self.query:
             body["query"] = self.query.as_dict()
+        if self.suggested_questions:
+            body["suggested_questions"] = self.suggested_questions.as_dict()
         if self.text:
             body["text"] = self.text.as_dict()
         return body
@@ -284,6 +289,8 @@ class GenieAttachment:
             body["attachment_id"] = self.attachment_id
         if self.query:
             body["query"] = self.query
+        if self.suggested_questions:
+            body["suggested_questions"] = self.suggested_questions
         if self.text:
             body["text"] = self.text
         return body
@@ -294,6 +301,7 @@ class GenieAttachment:
         return cls(
             attachment_id=d.get("attachment_id", None),
             query=_from_dict(d, "query", GenieQueryAttachment),
+            suggested_questions=_from_dict(d, "suggested_questions", GenieSuggestedQuestionsAttachment),
             text=_from_dict(d, "text", TextAttachment),
         )
 
@@ -413,6 +421,33 @@ class GenieConversationSummary:
         )
 
 
+@dataclass
+class GenieFeedback:
+    """Feedback containing rating and optional comment"""
+
+    rating: Optional[GenieFeedbackRating] = None
+    """The feedback rating"""
+
+    def as_dict(self) -> dict:
+        """Serializes the GenieFeedback into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.rating is not None:
+            body["rating"] = self.rating.value
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the GenieFeedback into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.rating is not None:
+            body["rating"] = self.rating
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> GenieFeedback:
+        """Deserializes the GenieFeedback from a dictionary."""
+        return cls(rating=_enum(d, "rating", GenieFeedbackRating))
+
+
 class GenieFeedbackRating(Enum):
     """Feedback rating for Genie messages"""
 
@@ -572,6 +607,9 @@ class GenieMessage:
     error: Optional[MessageError] = None
     """Error message if Genie failed to respond to the message"""
 
+    feedback: Optional[GenieFeedback] = None
+    """User feedback for the message if provided"""
+
     last_updated_timestamp: Optional[int] = None
     """Timestamp when the message was last updated"""
 
@@ -597,6 +635,8 @@ class GenieMessage:
             body["created_timestamp"] = self.created_timestamp
         if self.error:
             body["error"] = self.error.as_dict()
+        if self.feedback:
+            body["feedback"] = self.feedback.as_dict()
         if self.id is not None:
             body["id"] = self.id
         if self.last_updated_timestamp is not None:
@@ -626,6 +666,8 @@ class GenieMessage:
             body["created_timestamp"] = self.created_timestamp
         if self.error:
             body["error"] = self.error
+        if self.feedback:
+            body["feedback"] = self.feedback
         if self.id is not None:
             body["id"] = self.id
         if self.last_updated_timestamp is not None:
@@ -651,6 +693,7 @@ class GenieMessage:
             conversation_id=d.get("conversation_id", None),
             created_timestamp=d.get("created_timestamp", None),
             error=_from_dict(d, "error", MessageError),
+            feedback=_from_dict(d, "feedback", GenieFeedback),
             id=d.get("id", None),
             last_updated_timestamp=d.get("last_updated_timestamp", None),
             message_id=d.get("message_id", None),
@@ -779,6 +822,9 @@ class GenieSpace:
     description: Optional[str] = None
     """Description of the Genie Space"""
 
+    warehouse_id: Optional[str] = None
+    """Warehouse associated with the Genie Space"""
+
     def as_dict(self) -> dict:
         """Serializes the GenieSpace into a dictionary suitable for use as a JSON request body."""
         body = {}
@@ -788,6 +834,8 @@ class GenieSpace:
             body["space_id"] = self.space_id
         if self.title is not None:
             body["title"] = self.title
+        if self.warehouse_id is not None:
+            body["warehouse_id"] = self.warehouse_id
         return body
 
     def as_shallow_dict(self) -> dict:
@@ -799,12 +847,19 @@ class GenieSpace:
             body["space_id"] = self.space_id
         if self.title is not None:
             body["title"] = self.title
+        if self.warehouse_id is not None:
+            body["warehouse_id"] = self.warehouse_id
         return body
 
     @classmethod
     def from_dict(cls, d: Dict[str, Any]) -> GenieSpace:
         """Deserializes the GenieSpace from a dictionary."""
-        return cls(description=d.get("description", None), space_id=d.get("space_id", None), title=d.get("title", None))
+        return cls(
+            description=d.get("description", None),
+            space_id=d.get("space_id", None),
+            title=d.get("title", None),
+            warehouse_id=d.get("warehouse_id", None),
+        )
 
 
 @dataclass
@@ -856,6 +911,33 @@ class GenieStartConversationResponse:
         )
 
 
+@dataclass
+class GenieSuggestedQuestionsAttachment:
+    """Follow-up questions suggested by Genie"""
+
+    questions: Optional[List[str]] = None
+    """The suggested follow-up questions"""
+
+    def as_dict(self) -> dict:
+        """Serializes the GenieSuggestedQuestionsAttachment into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.questions:
+            body["questions"] = [v for v in self.questions]
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the GenieSuggestedQuestionsAttachment into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.questions:
+            body["questions"] = self.questions
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> GenieSuggestedQuestionsAttachment:
+        """Deserializes the GenieSuggestedQuestionsAttachment from a dictionary."""
+        return cls(questions=d.get("questions", None))
+
+
 @dataclass
 class GetPublishedDashboardTokenInfoResponse:
     authorization_details: Optional[List[AuthorizationDetails]] = None
@@ -1063,6 +1145,8 @@ class MessageErrorType(Enum):
     GENERIC_CHAT_COMPLETION_SERVICE_EXCEPTION = "GENERIC_CHAT_COMPLETION_SERVICE_EXCEPTION"
     GENERIC_SQL_EXEC_API_CALL_EXCEPTION = "GENERIC_SQL_EXEC_API_CALL_EXCEPTION"
     ILLEGAL_PARAMETER_DEFINITION_EXCEPTION = "ILLEGAL_PARAMETER_DEFINITION_EXCEPTION"
+    INTERNAL_CATALOG_MISSING_UC_PATH_EXCEPTION = "INTERNAL_CATALOG_MISSING_UC_PATH_EXCEPTION"
+    INTERNAL_CATALOG_PATH_OVERLAP_EXCEPTION = "INTERNAL_CATALOG_PATH_OVERLAP_EXCEPTION"
     INVALID_CERTIFIED_ANSWER_FUNCTION_EXCEPTION = "INVALID_CERTIFIED_ANSWER_FUNCTION_EXCEPTION"
     INVALID_CERTIFIED_ANSWER_IDENTIFIER_EXCEPTION = "INVALID_CERTIFIED_ANSWER_IDENTIFIER_EXCEPTION"
     INVALID_CHAT_COMPLETION_ARGUMENTS_JSON_EXCEPTION = "INVALID_CHAT_COMPLETION_ARGUMENTS_JSON_EXCEPTION"
@@ -1921,8 +2005,8 @@ class GenieAPI:
         :param space_id: str
           The ID of the Genie space to retrieve conversations from.
         :param include_all: bool (optional)
-          Include all conversations in the space across all users. Requires "Can Manage" permission on the
-          space.
+          Include all conversations in the space across all users. Requires at least CAN MANAGE permission on
+          the space.
         :param page_size: int (optional)
           Maximum number of conversations to return per page
         :param page_token: str (optional)
@@ -1970,15 +2054,7 @@ class GenieAPI:
         res = self._api.do("GET", "/api/2.0/genie/spaces", query=query, headers=headers)
         return GenieListSpacesResponse.from_dict(res)
 
-    def send_message_feedback(
-        self,
-        space_id: str,
-        conversation_id: str,
-        message_id: str,
-        feedback_rating: GenieFeedbackRating,
-        *,
-        feedback_text: Optional[str] = None,
-    ):
+    def send_message_feedback(self, space_id: str, conversation_id: str, message_id: str, rating: GenieFeedbackRating):
         """Send feedback for a message.
 
         :param space_id: str
@@ -1987,18 +2063,14 @@ class GenieAPI:
           The ID associated with the conversation.
         :param message_id: str
           The ID associated with the message to provide feedback for.
-        :param feedback_rating: :class:`GenieFeedbackRating`
+        :param rating: :class:`GenieFeedbackRating`
           The rating (POSITIVE, NEGATIVE, or NONE).
-        :param feedback_text: str (optional)
-          Optional text feedback that will be stored as a comment.
 
 
         """
         body = {}
-        if feedback_rating is not None:
-            body["feedback_rating"] = feedback_rating.value
-        if feedback_text is not None:
-            body["feedback_text"] = feedback_text
+        if rating is not None:
+            body["rating"] = rating.value
         headers = {
             "Accept": "application/json",
             "Content-Type": "application/json",

databricks/sdk/service/database.py

@@ -125,38 +125,36 @@ class DatabaseInstance:
     creator: Optional[str] = None
     """The email of the creator of the instance."""
 
+    effective_capacity: Optional[str] = None
+    """Deprecated. The sku of the instance; this field will always match the value of capacity."""
+
+    effective_enable_pg_native_login: Optional[bool] = None
+    """Whether the instance has PG native password login enabled."""
+
     effective_enable_readable_secondaries: Optional[bool] = None
-    """xref AIP-129. `enable_readable_secondaries` is owned by the client, while
-    `effective_enable_readable_secondaries` is owned by the server. `enable_readable_secondaries`
-    will only be set in Create/Update response messages if and only if the user provides the field
-    via the request. `effective_enable_readable_secondaries` on the other hand will always bet set
-    in all response messages (Create/Update/Get/List)."""
+    """Whether secondaries serving read-only traffic are enabled. Defaults to false."""
 
     effective_node_count: Optional[int] = None
-    """xref AIP-129. `node_count` is owned by the client, while `effective_node_count` is owned by the
-    server. `node_count` will only be set in Create/Update response messages if and only if the user
-    provides the field via the request. `effective_node_count` on the other hand will always bet set
-    in all response messages (Create/Update/Get/List)."""
+    """The number of nodes in the instance, composed of 1 primary and 0 or more secondaries. Defaults
+    to 1 primary and 0 secondaries."""
 
     effective_retention_window_in_days: Optional[int] = None
-    """xref AIP-129. `retention_window_in_days` is owned by the client, while
-    `effective_retention_window_in_days` is owned by the server. `retention_window_in_days` will
-    only be set in Create/Update response messages if and only if the user provides the field via
-    the request. `effective_retention_window_in_days` on the other hand will always bet set in all
-    response messages (Create/Update/Get/List)."""
+    """The retention window for the instance. This is the time window in days for which the historical
+    data is retained."""
 
     effective_stopped: Optional[bool] = None
-    """xref AIP-129. `stopped` is owned by the client, while `effective_stopped` is owned by the
-    server. `stopped` will only be set in Create/Update response messages if and only if the user
-    provides the field via the request. `effective_stopped` on the other hand will always bet set in
-    all response messages (Create/Update/Get/List)."""
+    """Whether the instance is stopped."""
+
+    enable_pg_native_login: Optional[bool] = None
+    """Whether to enable PG native password login on the instance. Defaults to false."""
 
     enable_readable_secondaries: Optional[bool] = None
     """Whether to enable secondaries to serve read-only traffic. Defaults to false."""
 
     node_count: Optional[int] = None
     """The number of nodes in the instance, composed of 1 primary and 0 or more secondaries. Defaults
-    to 1 primary and 0 secondaries."""
+    to 1 primary and 0 secondaries. This field is input only, see effective_node_count for the
+    output."""
 
     parent_instance_ref: Optional[DatabaseInstanceRef] = None
     """The ref of the parent instance. This is only available if the instance is child instance. Input:
@@ -181,7 +179,7 @@ class DatabaseInstance:
     """The current state of the instance."""
 
     stopped: Optional[bool] = None
-    """Whether the instance is stopped."""
+    """Whether to stop the instance. An input only param, see effective_stopped for the output."""
 
     uid: Optional[str] = None
     """An immutable UUID identifier for the instance."""
@@ -197,6 +195,10 @@ class DatabaseInstance:
             body["creation_time"] = self.creation_time
         if self.creator is not None:
             body["creator"] = self.creator
+        if self.effective_capacity is not None:
+            body["effective_capacity"] = self.effective_capacity
+        if self.effective_enable_pg_native_login is not None:
+            body["effective_enable_pg_native_login"] = self.effective_enable_pg_native_login
         if self.effective_enable_readable_secondaries is not None:
             body["effective_enable_readable_secondaries"] = self.effective_enable_readable_secondaries
         if self.effective_node_count is not None:
@@ -205,6 +207,8 @@ class DatabaseInstance:
             body["effective_retention_window_in_days"] = self.effective_retention_window_in_days
         if self.effective_stopped is not None:
             body["effective_stopped"] = self.effective_stopped
+        if self.enable_pg_native_login is not None:
+            body["enable_pg_native_login"] = self.enable_pg_native_login
         if self.enable_readable_secondaries is not None:
             body["enable_readable_secondaries"] = self.enable_readable_secondaries
         if self.name is not None:
@@ -240,6 +244,10 @@ class DatabaseInstance:
             body["creation_time"] = self.creation_time
         if self.creator is not None:
             body["creator"] = self.creator
+        if self.effective_capacity is not None:
+            body["effective_capacity"] = self.effective_capacity
+        if self.effective_enable_pg_native_login is not None:
+            body["effective_enable_pg_native_login"] = self.effective_enable_pg_native_login
         if self.effective_enable_readable_secondaries is not None:
             body["effective_enable_readable_secondaries"] = self.effective_enable_readable_secondaries
         if self.effective_node_count is not None:
@@ -248,6 +256,8 @@ class DatabaseInstance:
             body["effective_retention_window_in_days"] = self.effective_retention_window_in_days
         if self.effective_stopped is not None:
             body["effective_stopped"] = self.effective_stopped
+        if self.enable_pg_native_login is not None:
+            body["enable_pg_native_login"] = self.enable_pg_native_login
         if self.enable_readable_secondaries is not None:
             body["enable_readable_secondaries"] = self.enable_readable_secondaries
         if self.name is not None:
@@ -280,10 +290,13 @@ class DatabaseInstance:
             child_instance_refs=_repeated_dict(d, "child_instance_refs", DatabaseInstanceRef),
             creation_time=d.get("creation_time", None),
             creator=d.get("creator", None),
+            effective_capacity=d.get("effective_capacity", None),
+            effective_enable_pg_native_login=d.get("effective_enable_pg_native_login", None),
             effective_enable_readable_secondaries=d.get("effective_enable_readable_secondaries", None),
             effective_node_count=d.get("effective_node_count", None),
             effective_retention_window_in_days=d.get("effective_retention_window_in_days", None),
             effective_stopped=d.get("effective_stopped", None),
+            enable_pg_native_login=d.get("enable_pg_native_login", None),
             enable_readable_secondaries=d.get("enable_readable_secondaries", None),
             name=d.get("name", None),
             node_count=d.get("node_count", None),
@@ -315,12 +328,9 @@ class DatabaseInstanceRef:
     provided as input to create a child instance."""
 
     effective_lsn: Optional[str] = None
-    """xref AIP-129. `lsn` is owned by the client, while `effective_lsn` is owned by the server. `lsn`
-    will only be set in Create/Update response messages if and only if the user provides the field
-    via the request. `effective_lsn` on the other hand will always bet set in all response messages
-    (Create/Update/Get/List). For a parent ref instance, this is the LSN on the parent instance from
-    which the instance was created. For a child ref instance, this is the LSN on the instance from
-    which the child instance was created."""
+    """For a parent ref instance, this is the LSN on the parent instance from which the instance was
+    created. For a child ref instance, this is the LSN on the instance from which the child instance
+    was created."""
 
     lsn: Optional[str] = None
     """User-specified WAL LSN of the ref database instance.
@@ -1591,12 +1601,8 @@ class DatabaseAPI:
           By default, a instance cannot be deleted if it has descendant instances created via PITR. If this
           flag is specified as true, all descendent instances will be deleted as well.
         :param purge: bool (optional)
-          Note purge=false is in development. If false, the database instance is soft deleted (implementation
-          pending). Soft deleted instances behave as if they are deleted, and cannot be used for CRUD
-          operations nor connected to. However they can be undeleted by calling the undelete API for a limited
-          time (implementation pending). If true, the database instance is hard deleted and cannot be
-          undeleted. For the time being, setting this value to true is required to delete an instance (soft
-          delete is not yet supported).
+          Deprecated. Omitting the field or setting it to true will result in the field being hard deleted.
+          Setting a value of false will throw a bad request.
 
 
         """