Version 1.4.74

Author: gyanendra8

abacusai/__init__.py

@@ -60,6 +60,7 @@
 from .custom_metric import CustomMetric
 from .custom_metric_version import CustomMetricVersion
 from .custom_train_function_info import CustomTrainFunctionInfo
+from .daemon_task_conversation import DaemonTaskConversation
 from .data_consistency_duplication import DataConsistencyDuplication
 from .data_metrics import DataMetrics
 from .data_prep_logs import DataPrepLogs
@@ -290,4 +291,4 @@
 from .workflow_node_template import WorkflowNodeTemplate
 
 
-__version__ = "1.4.73"
+__version__ = "1.4.74"

abacusai/api_class/ai_agents.py

@@ -743,13 +743,12 @@ def __init__(self, name: str, chatbot_deployment_id: str = None, chatbot_paramet
         output_mappings = [WorkflowNodeOutputMapping(name='chat_output', variable_type=enums.WorkflowNodeOutputType.DICT, description='The output of the chatbot')]
 
         # Determine function_name and create minimal source_code for parent validation
-        function_name = None
-        if chatbot_deployment_id:
-            function_name = self.get_function_name()
-        minimal_source_code = f"def {function_name or 'llm_agent_node'}(user_input):\n    pass"
+        # Always use get_function_name() which includes uid to ensure uniqueness
+        function_name = self.get_function_name()
+        minimal_source_code = f'def {function_name}(user_input):\n    pass'
 
         # Initialize parent class with minimal source code to satisfy validation
-        super().__init__(name=name, function_name=function_name or 'llm_agent_node', source_code=minimal_source_code, input_mappings=input_mappings, output_mappings=output_mappings, input_schema=input_schema or WorkflowNodeInputSchema.from_input_mappings(input_mappings), output_schema=output_schema or WorkflowNodeOutputSchema.from_fields_list(['chat_output']))
+        super().__init__(name=name, function_name=function_name, source_code=minimal_source_code, input_mappings=input_mappings, output_mappings=output_mappings, input_schema=input_schema or WorkflowNodeInputSchema.from_input_mappings(input_mappings), output_schema=output_schema or WorkflowNodeOutputSchema.from_fields_list(['chat_output']))
         self.node_type = 'llm_agent_node'
         self.source_code = None
 

abacusai/api_class/enums.py

@@ -827,6 +827,7 @@ class DeploymentConversationType(ApiEnum):
     TEST_AGENT = 'TEST_AGENT'
     SUPER_AGENT = 'SUPER_AGENT'
     CODE_LLM_NON_INTERACTIVE = 'CODE_LLM_NON_INTERACTIVE'
+    BROWSER_EXTENSION = 'BROWSER_EXTENSION'
 
 
 class AgentClientType(ApiEnum):

abacusai/api_class/segments.py

@@ -1,5 +1,6 @@
 import dataclasses
 import uuid
+import warnings
 from typing import Any, List
 
 from . import enums
@@ -81,9 +82,21 @@ class TextResponseSection(ResponseSection):
 
     segment: str
 
-    def __init__(self, text: str, section_key: str = None):
+    def __init__(self, segment: str = None, section_key: str = None, text: str = None):  # text is added for backward compatibility
+        if segment is not None and text is not None:
+            raise ValueError("Cannot specify both 'segment' and 'text' parameters. The 'text' parameter is deprecated, please use 'segment' only.")
+        if segment is None and text is None:
+            raise TypeError("'Segment' parameter is required.")
+
+        if text is not None:
+            warnings.warn(
+                "The 'text' parameter is deprecated and will be removed in future. Please use 'segment' instead.",
+                DeprecationWarning,
+                stacklevel=2
+            )
+
         super().__init__(type=enums.ResponseSectionType.TEXT, id=section_key)
-        self.segment = text
+        self.segment = segment if segment is not None else text
 
 
 @dataclasses.dataclass

abacusai/batch_prediction.py

@@ -84,7 +84,7 @@ def __init__(self, client, batchPredictionId=None, createdAt=None, name=None, de
             BatchPredictionArgs, globalPredictionArgs)
         self.batch_prediction_args = client._build_class(getattr(
             api_class, batchPredictionArgsType, BatchPredictionArgs) if batchPredictionArgsType else BatchPredictionArgs, batchPredictionArgs)
-        self.deprecated_keys = {'explanations', 'global_prediction_args'}
+        self.deprecated_keys = {'global_prediction_args', 'explanations'}
 
     def __repr__(self):
         repr_dict = {f'batch_prediction_id': repr(self.batch_prediction_id), f'created_at': repr(self.created_at), f'name': repr(self.name), f'deployment_id': repr(self.deployment_id), f'file_connector_output_location': repr(self.file_connector_output_location), f'database_connector_id': repr(self.database_connector_id), f'database_output_configuration': repr(self.database_output_configuration), f'file_output_format': repr(self.file_output_format), f'connector_type': repr(self.connector_type), f'legacy_input_location': repr(self.legacy_input_location), f'output_feature_group_id': repr(self.output_feature_group_id), f'feature_group_table_name': repr(self.feature_group_table_name), f'output_feature_group_table_name': repr(self.output_feature_group_table_name), f'summary_feature_group_table_name': repr(self.summary_feature_group_table_name), f'csv_input_prefix': repr(

abacusai/batch_prediction_version.py

@@ -100,7 +100,7 @@ def __init__(self, client, batchPredictionVersion=None, batchPredictionId=None,
             BatchPredictionArgs, globalPredictionArgs)
         self.batch_prediction_args = client._build_class(getattr(
             api_class, batchPredictionArgsType, BatchPredictionArgs) if batchPredictionArgsType else BatchPredictionArgs, batchPredictionArgs)
-        self.deprecated_keys = {'explanations', 'global_prediction_args'}
+        self.deprecated_keys = {'global_prediction_args', 'explanations'}
 
     def __repr__(self):
         repr_dict = {f'batch_prediction_version': repr(self.batch_prediction_version), f'batch_prediction_id': repr(self.batch_prediction_id), f'status': repr(self.status), f'drift_monitor_status': repr(self.drift_monitor_status), f'deployment_id': repr(self.deployment_id), f'model_id': repr(self.model_id), f'model_version': repr(self.model_version), f'predictions_started_at': repr(self.predictions_started_at), f'predictions_completed_at': repr(self.predictions_completed_at), f'database_output_error': repr(self.database_output_error), f'total_predictions': repr(self.total_predictions), f'failed_predictions': repr(self.failed_predictions), f'database_connector_id': repr(self.database_connector_id), f'database_output_configuration': repr(self.database_output_configuration), f'file_connector_output_location': repr(self.file_connector_output_location), f'file_output_format': repr(self.file_output_format), f'connector_type': repr(self.connector_type), f'legacy_input_location': repr(self.legacy_input_location), f'error': repr(self.error), f'drift_monitor_error': repr(self.drift_monitor_error), f'monitor_warnings': repr(self.monitor_warnings), f'csv_input_prefix': repr(

abacusai/chatllm_task.py

@@ -22,9 +22,10 @@ class ChatllmTask(AbstractApiClass):
             numUnreadTaskInstances (int): The number of unread task instances for the chatllm task.
             computePointsUsed (int): The compute points used for the chatllm task.
             displayMarkdown (str): The display markdown for the chatllm task.
+            requiresNewConversation (bool): Whether a new conversation is required for the chatllm task.
     """
 
-    def __init__(self, client, chatllmTaskId=None, daemonTaskId=None, taskType=None, name=None, instructions=None, lifecycle=None, scheduleInfo=None, externalApplicationId=None, deploymentConversationId=None, sourceDeploymentConversationId=None, enableEmailAlerts=None, email=None, numUnreadTaskInstances=None, computePointsUsed=None, displayMarkdown=None):
+    def __init__(self, client, chatllmTaskId=None, daemonTaskId=None, taskType=None, name=None, instructions=None, lifecycle=None, scheduleInfo=None, externalApplicationId=None, deploymentConversationId=None, sourceDeploymentConversationId=None, enableEmailAlerts=None, email=None, numUnreadTaskInstances=None, computePointsUsed=None, displayMarkdown=None, requiresNewConversation=None):
         super().__init__(client, chatllmTaskId)
         self.chatllm_task_id = chatllmTaskId
         self.daemon_task_id = daemonTaskId
@@ -41,11 +42,12 @@ def __init__(self, client, chatllmTaskId=None, daemonTaskId=None, taskType=None,
         self.num_unread_task_instances = numUnreadTaskInstances
         self.compute_points_used = computePointsUsed
         self.display_markdown = displayMarkdown
+        self.requires_new_conversation = requiresNewConversation
         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'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)}
+            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)}
         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])
@@ -58,6 +60,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, '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}
+        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, '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}
         return {key: value for key, value in resp.items() if value is not None and key not in self.deprecated_keys}

abacusai/client.py

@@ -667,7 +667,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.73'
+    client_version = '1.4.74'
 
     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/daemon_task_conversation.py

@@ -0,0 +1,43 @@
+from .return_class import AbstractApiClass
+
+
+class DaemonTaskConversation(AbstractApiClass):
+    """
+        Daemon Task Conversation
+
+        Args:
+            client (ApiClient): An authenticated API Client instance
+            deploymentConversationId (id): The ID of the deployment conversation
+            createdAt (str): The creation timestamp
+            updatedAt (str): The last update timestamp
+            deploymentConversationName (str): The name of the conversation
+            externalApplicationId (str): The external application ID
+    """
+
+    def __init__(self, client, deploymentConversationId=None, createdAt=None, updatedAt=None, deploymentConversationName=None, externalApplicationId=None):
+        super().__init__(client, None)
+        self.deployment_conversation_id = deploymentConversationId
+        self.created_at = createdAt
+        self.updated_at = updatedAt
+        self.deployment_conversation_name = deploymentConversationName
+        self.external_application_id = externalApplicationId
+        self.deprecated_keys = {}
+
+    def __repr__(self):
+        repr_dict = {f'deployment_conversation_id': repr(self.deployment_conversation_id), f'created_at': repr(self.created_at), f'updated_at': repr(
+            self.updated_at), f'deployment_conversation_name': repr(self.deployment_conversation_name), f'external_application_id': repr(self.external_application_id)}
+        class_name = "DaemonTaskConversation"
+        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])
+        return f"{class_name}({repr_str})"
+
+    def to_dict(self):
+        """
+        Get a dict representation of the parameters in this class
+
+        Returns:
+            dict: The dict value representation of the class parameters
+        """
+        resp = {'deployment_conversation_id': self.deployment_conversation_id, 'created_at': self.created_at, 'updated_at': self.updated_at,
+                'deployment_conversation_name': self.deployment_conversation_name, 'external_application_id': self.external_application_id}
+        return {key: value for key, value in resp.items() if value is not None and key not in self.deprecated_keys}