Version 1.4.84

Author: gyanendra8

abacusai/__init__.py

@@ -303,4 +303,4 @@
 from .workflow_node_template import WorkflowNodeTemplate
 
 
-__version__ = "1.4.83"
+__version__ = "1.4.84"

abacusai/api_class/enums.py

@@ -562,6 +562,7 @@ class LLMName(ApiEnum):
     CLAUDE_V4_SONNET = 'CLAUDE_V4_SONNET'
     CLAUDE_V4_OPUS = 'CLAUDE_V4_OPUS'
     CLAUDE_V4_5_SONNET = 'CLAUDE_V4_5_SONNET'
+    CLAUDE_V4_6_SONNET = 'CLAUDE_V4_6_SONNET'
     CLAUDE_V4_5_OPUS = 'CLAUDE_V4_5_OPUS'
     CLAUDE_V4_6_OPUS = 'CLAUDE_V4_6_OPUS'
     GEMINI_1_5_PRO = 'GEMINI_1_5_PRO'
@@ -570,6 +571,7 @@ class LLMName(ApiEnum):
     GEMINI_2_FLASH = 'GEMINI_2_FLASH'
     GEMINI_2_5_PRO = 'GEMINI_2_5_PRO'
     GEMINI_3_PRO = 'GEMINI_3_PRO'
+    GEMINI_3_1_PRO = 'GEMINI_3_1_PRO'
     GEMINI_2_5_FLASH = 'GEMINI_2_5_FLASH'
     GEMINI_3_FLASH = 'GEMINI_3_FLASH'
     XAI_GROK = 'XAI_GROK'

abacusai/api_class/model.py

@@ -522,7 +522,7 @@ class ChatLLMTrainingConfig(TrainingConfig):
     keyword_requirement_instructions: str = dataclasses.field(default=None)
     query_rewrite_instructions: str = dataclasses.field(default=None)
     max_search_results: int = dataclasses.field(default=None)
-    max_page_images: int = dataclasses.field(default=5)
+    max_page_images: int = dataclasses.field(default=None)
     data_feature_group_ids: List[str] = dataclasses.field(default=None)
     data_prompt_context: str = dataclasses.field(default=None)
     data_prompt_table_context: Dict[str, str] = dataclasses.field(default=None)

abacusai/api_client_utils.py

@@ -33,7 +33,7 @@ def avro_to_pandas_dtype(avro_type):
         'bytes': 'object',
         'null': 'object',
         'date': 'object',
-        'datetime': 'datetime',
+        'datetime': None,
     }
 
     if isinstance(avro_type, dict):
@@ -177,14 +177,14 @@ def load_as_pandas_from_avro_fd(fd: IO):
         [r for r in reader], columns=col_dtypes.keys())
 
     for col in df_part.columns:
-        if col_dtypes[col] == 'datetime':
+        if col_dtypes[col] is None:
             df_part[col] = pd.to_datetime(
                 df_part[col], errors='coerce')
 
-        if pd.core.dtypes.common.is_datetime64_ns_dtype(df_part[col]):
+        if pd.api.types.is_datetime64_any_dtype(df_part[col]):
             df_part[col] = df_part[col].dt.tz_localize(
                 None)
-        elif str(df_part[col].dtype).lower() != str(col_dtypes[col]).lower():
+        elif col_dtypes[col] is not None and str(df_part[col].dtype).lower() != str(col_dtypes[col]).lower():
             df_part[col] = df_part[col].astype(
                 col_dtypes[col])
     return df_part

abacusai/chatllm_task.py

@@ -29,11 +29,13 @@ class ChatllmTask(AbstractApiClass):
             executionMode (str): The execution mode of the chatllm task.
             taskDefinition (dict): The task definition (for web_service_trigger tasks).
             webAppHostname (str): The hostname of the web app associated with the daemon task.
+            triggerType (str): The trigger type of the daemon task (scheduled or event_based).
+            webhookUrl (str): The webhook URL for event-based daemon tasks.
             hostedDatabase (HostedDatabase): The hosted database for the daemon task.
             latestDaemonTaskInstance (DaemonTaskInstance): The latest task instance for daemon tasks.
     """
 
-    def __init__(self, client, chatllmTaskId=None, daemonTaskId=None, taskType=None, name=None, instructions=None, description=None, lifecycle=None, scheduleInfo=None, externalApplicationId=None, deploymentConversationId=None, sourceDeploymentConversationId=None, enableEmailAlerts=None, email=None, numUnreadTaskInstances=None, computePointsUsed=None, displayMarkdown=None, requiresNewConversation=None, executionMode=None, taskDefinition=None, webAppHostname=None, hostedDatabase={}, latestDaemonTaskInstance={}):
+    def __init__(self, client, chatllmTaskId=None, daemonTaskId=None, taskType=None, name=None, instructions=None, description=None, lifecycle=None, scheduleInfo=None, externalApplicationId=None, deploymentConversationId=None, sourceDeploymentConversationId=None, enableEmailAlerts=None, email=None, numUnreadTaskInstances=None, computePointsUsed=None, displayMarkdown=None, requiresNewConversation=None, executionMode=None, taskDefinition=None, webAppHostname=None, triggerType=None, webhookUrl=None, hostedDatabase={}, latestDaemonTaskInstance={}):
         super().__init__(client, chatllmTaskId)
         self.chatllm_task_id = chatllmTaskId
         self.daemon_task_id = daemonTaskId
@@ -55,15 +57,17 @@ def __init__(self, client, chatllmTaskId=None, daemonTaskId=None, taskType=None,
         self.execution_mode = executionMode
         self.task_definition = taskDefinition
         self.web_app_hostname = webAppHostname
+        self.trigger_type = triggerType
+        self.webhook_url = webhookUrl
         self.hosted_database = client._build_class(
             HostedDatabase, hostedDatabase)
         self.latest_daemon_task_instance = client._build_class(
             DaemonTaskInstance, latestDaemonTaskInstance)
         self.deprecated_keys = {}
 
     def __repr__(self):
-        repr_dict = {f'chatllm_task_id': repr(self.chatllm_task_id), f'daemon_task_id': repr(self.daemon_task_id), f'task_type': repr(self.task_type), f'name': repr(self.name), f'instructions': repr(self.instructions), f'description': repr(self.description), f'lifecycle': repr(self.lifecycle), f'schedule_info': repr(self.schedule_info), f'external_application_id': repr(self.external_application_id), f'deployment_conversation_id': repr(self.deployment_conversation_id), f'source_deployment_conversation_id': repr(self.source_deployment_conversation_id), f'enable_email_alerts': repr(
-            self.enable_email_alerts), f'email': repr(self.email), f'num_unread_task_instances': repr(self.num_unread_task_instances), f'compute_points_used': repr(self.compute_points_used), f'display_markdown': repr(self.display_markdown), f'requires_new_conversation': repr(self.requires_new_conversation), f'execution_mode': repr(self.execution_mode), f'task_definition': repr(self.task_definition), f'web_app_hostname': repr(self.web_app_hostname), f'hosted_database': repr(self.hosted_database), f'latest_daemon_task_instance': repr(self.latest_daemon_task_instance)}
+        repr_dict = {f'chatllm_task_id': repr(self.chatllm_task_id), f'daemon_task_id': repr(self.daemon_task_id), f'task_type': repr(self.task_type), f'name': repr(self.name), f'instructions': repr(self.instructions), f'description': repr(self.description), f'lifecycle': repr(self.lifecycle), f'schedule_info': repr(self.schedule_info), f'external_application_id': repr(self.external_application_id), f'deployment_conversation_id': repr(self.deployment_conversation_id), f'source_deployment_conversation_id': repr(self.source_deployment_conversation_id), f'enable_email_alerts': repr(self.enable_email_alerts), f'email': repr(
+            self.email), f'num_unread_task_instances': repr(self.num_unread_task_instances), f'compute_points_used': repr(self.compute_points_used), f'display_markdown': repr(self.display_markdown), f'requires_new_conversation': repr(self.requires_new_conversation), f'execution_mode': repr(self.execution_mode), f'task_definition': repr(self.task_definition), f'web_app_hostname': repr(self.web_app_hostname), f'trigger_type': repr(self.trigger_type), f'webhook_url': repr(self.webhook_url), f'hosted_database': repr(self.hosted_database), f'latest_daemon_task_instance': repr(self.latest_daemon_task_instance)}
         class_name = "ChatllmTask"
         repr_str = ',\n  '.join([f'{key}={value}' for key, value in repr_dict.items(
         ) if getattr(self, key, None) is not None and key not in self.deprecated_keys])
@@ -76,6 +80,6 @@ def to_dict(self):
         Returns:
             dict: The dict value representation of the class parameters
         """
-        resp = {'chatllm_task_id': self.chatllm_task_id, 'daemon_task_id': self.daemon_task_id, 'task_type': self.task_type, 'name': self.name, 'instructions': self.instructions, 'description': self.description, 'lifecycle': self.lifecycle, 'schedule_info': self.schedule_info, 'external_application_id': self.external_application_id, 'deployment_conversation_id': self.deployment_conversation_id, 'source_deployment_conversation_id': self.source_deployment_conversation_id, 'enable_email_alerts': self.enable_email_alerts,
-                'email': self.email, 'num_unread_task_instances': self.num_unread_task_instances, 'compute_points_used': self.compute_points_used, 'display_markdown': self.display_markdown, 'requires_new_conversation': self.requires_new_conversation, 'execution_mode': self.execution_mode, 'task_definition': self.task_definition, 'web_app_hostname': self.web_app_hostname, 'hosted_database': self._get_attribute_as_dict(self.hosted_database), 'latest_daemon_task_instance': self._get_attribute_as_dict(self.latest_daemon_task_instance)}
+        resp = {'chatllm_task_id': self.chatllm_task_id, 'daemon_task_id': self.daemon_task_id, 'task_type': self.task_type, 'name': self.name, 'instructions': self.instructions, 'description': self.description, 'lifecycle': self.lifecycle, 'schedule_info': self.schedule_info, 'external_application_id': self.external_application_id, 'deployment_conversation_id': self.deployment_conversation_id, 'source_deployment_conversation_id': self.source_deployment_conversation_id, 'enable_email_alerts': self.enable_email_alerts, 'email': self.email,
+                'num_unread_task_instances': self.num_unread_task_instances, 'compute_points_used': self.compute_points_used, 'display_markdown': self.display_markdown, 'requires_new_conversation': self.requires_new_conversation, 'execution_mode': self.execution_mode, 'task_definition': self.task_definition, 'web_app_hostname': self.web_app_hostname, 'trigger_type': self.trigger_type, 'webhook_url': self.webhook_url, 'hosted_database': self._get_attribute_as_dict(self.hosted_database), 'latest_daemon_task_instance': self._get_attribute_as_dict(self.latest_daemon_task_instance)}
         return {key: value for key, value in resp.items() if value is not None and key not in self.deprecated_keys}

abacusai/client.py

@@ -716,7 +716,7 @@ class BaseApiClient:
         client_options (ClientOptions): Optional API client configurations
         skip_version_check (bool): If true, will skip checking the server's current API version on initializing the client
     """
-    client_version = '1.4.83'
+    client_version = '1.4.84'
 
     def __init__(self, api_key: str = None, server: str = None, client_options: ClientOptions = None, skip_version_check: bool = False, include_tb: bool = False):
         self.api_key = api_key

abacusai/llm_artifact.py

@@ -19,12 +19,13 @@ class LlmArtifact(AbstractApiClass):
             hasDatabase (bool): Whether the app associated with the artifact has a database
             hasStorage (bool): Whether the app associated with the artifact has a storage
             projectMetadata (dict): The metadata of the web app project
+            hasSeparateProdDatabase (bool): Whether the artifact has a separate prod database
             hasDatabaseSnapshot (bool): Whether the artifact database has a snapshot
             supportsPreviewRestore (bool): Whether the artifact supports preview restore
             accessLevel (str): The access level of the artifact
     """
 
-    def __init__(self, client, llmArtifactId=None, info=None, description=None, createdAt=None, webAppDeploymentId=None, deploymentStatus=None, isLatest=None, deploymentConversationId=None, webAppProjectId=None, hasDatabase=None, hasStorage=None, projectMetadata=None, hasDatabaseSnapshot=None, supportsPreviewRestore=None, accessLevel=None):
+    def __init__(self, client, llmArtifactId=None, info=None, description=None, createdAt=None, webAppDeploymentId=None, deploymentStatus=None, isLatest=None, deploymentConversationId=None, webAppProjectId=None, hasDatabase=None, hasStorage=None, projectMetadata=None, hasSeparateProdDatabase=None, hasDatabaseSnapshot=None, supportsPreviewRestore=None, accessLevel=None):
         super().__init__(client, llmArtifactId)
         self.llm_artifact_id = llmArtifactId
         self.info = info
@@ -38,14 +39,15 @@ def __init__(self, client, llmArtifactId=None, info=None, description=None, crea
         self.has_database = hasDatabase
         self.has_storage = hasStorage
         self.project_metadata = projectMetadata
+        self.has_separate_prod_database = hasSeparateProdDatabase
         self.has_database_snapshot = hasDatabaseSnapshot
         self.supports_preview_restore = supportsPreviewRestore
         self.access_level = accessLevel
         self.deprecated_keys = {}
 
     def __repr__(self):
-        repr_dict = {f'llm_artifact_id': repr(self.llm_artifact_id), f'info': repr(self.info), f'description': repr(self.description), f'created_at': repr(self.created_at), f'web_app_deployment_id': repr(self.web_app_deployment_id), f'deployment_status': repr(self.deployment_status), f'is_latest': repr(self.is_latest), f'deployment_conversation_id': repr(
-            self.deployment_conversation_id), f'web_app_project_id': repr(self.web_app_project_id), f'has_database': repr(self.has_database), f'has_storage': repr(self.has_storage), f'project_metadata': repr(self.project_metadata), f'has_database_snapshot': repr(self.has_database_snapshot), f'supports_preview_restore': repr(self.supports_preview_restore), f'access_level': repr(self.access_level)}
+        repr_dict = {f'llm_artifact_id': repr(self.llm_artifact_id), f'info': repr(self.info), f'description': repr(self.description), f'created_at': repr(self.created_at), f'web_app_deployment_id': repr(self.web_app_deployment_id), f'deployment_status': repr(self.deployment_status), f'is_latest': repr(self.is_latest), f'deployment_conversation_id': repr(self.deployment_conversation_id), f'web_app_project_id': repr(
+            self.web_app_project_id), f'has_database': repr(self.has_database), f'has_storage': repr(self.has_storage), f'project_metadata': repr(self.project_metadata), f'has_separate_prod_database': repr(self.has_separate_prod_database), f'has_database_snapshot': repr(self.has_database_snapshot), f'supports_preview_restore': repr(self.supports_preview_restore), f'access_level': repr(self.access_level)}
         class_name = "LlmArtifact"
         repr_str = ',\n  '.join([f'{key}={value}' for key, value in repr_dict.items(
         ) if getattr(self, key, None) is not None and key not in self.deprecated_keys])
@@ -58,6 +60,6 @@ def to_dict(self):
         Returns:
             dict: The dict value representation of the class parameters
         """
-        resp = {'llm_artifact_id': self.llm_artifact_id, 'info': self.info, 'description': self.description, 'created_at': self.created_at, 'web_app_deployment_id': self.web_app_deployment_id, 'deployment_status': self.deployment_status, 'is_latest': self.is_latest, 'deployment_conversation_id': self.deployment_conversation_id,
-                'web_app_project_id': self.web_app_project_id, 'has_database': self.has_database, 'has_storage': self.has_storage, 'project_metadata': self.project_metadata, 'has_database_snapshot': self.has_database_snapshot, 'supports_preview_restore': self.supports_preview_restore, 'access_level': self.access_level}
+        resp = {'llm_artifact_id': self.llm_artifact_id, 'info': self.info, 'description': self.description, 'created_at': self.created_at, 'web_app_deployment_id': self.web_app_deployment_id, 'deployment_status': self.deployment_status, 'is_latest': self.is_latest, 'deployment_conversation_id': self.deployment_conversation_id, 'web_app_project_id':
+                self.web_app_project_id, 'has_database': self.has_database, 'has_storage': self.has_storage, 'project_metadata': self.project_metadata, 'has_separate_prod_database': self.has_separate_prod_database, 'has_database_snapshot': self.has_database_snapshot, 'supports_preview_restore': self.supports_preview_restore, 'access_level': self.access_level}
         return {key: value for key, value in resp.items() if value is not None and key not in self.deprecated_keys}

abacusai/vertical.py

@@ -8,22 +8,24 @@ class Vertical(AbstractApiClass):
         Args:
             client (ApiClient): An authenticated API Client instance
             externalApplicationId (str): The ID of the vertical's external application.
+            deploymentId (str): The ID of the vertical's deployment.
             name (str): The name of the vertical.
             description (str): The description of the vertical.
             verticalType (str): The type of vertical (e.g., HEALTH).
     """
 
-    def __init__(self, client, externalApplicationId=None, name=None, description=None, verticalType=None):
+    def __init__(self, client, externalApplicationId=None, deploymentId=None, name=None, description=None, verticalType=None):
         super().__init__(client, None)
         self.external_application_id = externalApplicationId
+        self.deployment_id = deploymentId
         self.name = name
         self.description = description
         self.vertical_type = verticalType
         self.deprecated_keys = {}
 
     def __repr__(self):
-        repr_dict = {f'external_application_id': repr(self.external_application_id), f'name': repr(
-            self.name), f'description': repr(self.description), f'vertical_type': repr(self.vertical_type)}
+        repr_dict = {f'external_application_id': repr(self.external_application_id), f'deployment_id': repr(
+            self.deployment_id), f'name': repr(self.name), f'description': repr(self.description), f'vertical_type': repr(self.vertical_type)}
         class_name = "Vertical"
         repr_str = ',\n  '.join([f'{key}={value}' for key, value in repr_dict.items(
         ) if getattr(self, key, None) is not None and key not in self.deprecated_keys])
@@ -36,6 +38,6 @@ def to_dict(self):
         Returns:
             dict: The dict value representation of the class parameters
         """
-        resp = {'external_application_id': self.external_application_id, 'name': self.name,
-                'description': self.description, 'vertical_type': self.vertical_type}
+        resp = {'external_application_id': self.external_application_id, 'deployment_id': self.deployment_id,
+                'name': self.name, 'description': self.description, 'vertical_type': self.vertical_type}
         return {key: value for key, value in resp.items() if value is not None and key not in self.deprecated_keys}