Version 1.4.67

Author: gyanendra8

abacusai/__init__.py

@@ -21,6 +21,7 @@
 from .application_connector import ApplicationConnector
 from .audio_gen_settings import AudioGenSettings
 from .audio_url_result import AudioUrlResult
+from .audit_log import AuditLog
 from .batch_prediction import BatchPrediction
 from .batch_prediction_version import BatchPredictionVersion
 from .batch_prediction_version_logs import BatchPredictionVersionLogs
@@ -83,6 +84,7 @@
 from .document_retriever import DocumentRetriever
 from .document_retriever_lookup_result import DocumentRetrieverLookupResult
 from .document_retriever_version import DocumentRetrieverVersion
+from .document_retriever_version_logs import DocumentRetrieverVersionLogs
 from .drift_distribution import DriftDistribution
 from .drift_distributions import DriftDistributions
 from .eda import Eda
@@ -266,6 +268,7 @@
 from .use_case_requirements import UseCaseRequirements
 from .user import User
 from .user_exception import UserException
+from .user_group_object_permission import UserGroupObjectPermission
 from .video_gen_costs import VideoGenCosts
 from .video_gen_model import VideoGenModel
 from .video_gen_model_options import VideoGenModelOptions
@@ -282,4 +285,4 @@
 from .workflow_node_template import WorkflowNodeTemplate
 
 
-__version__ = "1.4.66"
+__version__ = "1.4.67"

abacusai/api_class/ai_agents.py

@@ -9,11 +9,15 @@
 MIN_AGENT_SLEEP_TIME = 300
 
 
-def validate_input_dict_param(dict_object, friendly_class_name, must_contain=[]):
+def validate_input_dict_param(dict_object, friendly_class_name, must_contain=[], schema={}):
     if not isinstance(dict_object, dict):
         raise ValueError(friendly_class_name, 'Invalid argument. Provided argument should be a dictionary.')
     if any(field not in dict_object for field in must_contain):
         raise ValueError(friendly_class_name, f'One or more keys are missing in the argument provided. Must contain keys - {must_contain}.')
+    for key, value_type in schema.items():
+        value = dict_object.get(key)
+        if value and not isinstance(value, value_type):
+            raise ValueError(friendly_class_name, f'Invalid parameter "{key}". It should be of type {value_type.__name__}.')
 
 
 @dataclasses.dataclass
@@ -766,7 +770,7 @@ def to_dict(self):
 
     @classmethod
     def from_dict(cls, graph: dict):
-        validate_input_dict_param(graph, friendly_class_name='workflow_graph')
+        validate_input_dict_param(graph, friendly_class_name='workflow_graph', schema={'nodes': list, 'edges': list, 'primary_start_node': str, 'common_source_code': str, 'specification_type': str})
         if graph.get('__return_filter'):
             for node in graph.get('nodes', []):
                 node['__return_filter'] = True

abacusai/api_class/model.py

@@ -486,6 +486,7 @@ class ChatLLMTrainingConfig(TrainingConfig):
         data_prompt_column_context (Dict[str, str]): Dict of 'table_name.column_name' and 'column_context' pairs to provide column context for some selected columns in the selected structured data table. This replaces the default auto-generated information about the column data.
         hide_sql_and_code (bool): When running data queries, this will hide the generated SQL and Code in the response.
         disable_data_summarization (bool): After executing a query summarize the reponse and reply back with only the table and query run.
+        disable_data_fetch_for_training (bool): Train using only table and column schema metadata without fetching sample data. This speeds up training but may result in less context for the model.
         data_columns_to_ignore (List[str]): Columns to ignore while encoding information about structured data tables in context for the LLM. A list of strings of format "<table_name>.<column_name>"
         search_score_cutoff (float): Minimum search score to consider a document as a valid search result.
         include_bm25_retrieval (bool): Combine BM25 search score with vector search using reciprocal rank fusion.
@@ -525,6 +526,7 @@ class ChatLLMTrainingConfig(TrainingConfig):
     data_prompt_column_context: Dict[str, str] = dataclasses.field(default=None)
     hide_sql_and_code: bool = dataclasses.field(default=None)
     disable_data_summarization: bool = dataclasses.field(default=None)
+    disable_data_fetch_for_training: bool = dataclasses.field(default=None)
     data_columns_to_ignore: List[str] = dataclasses.field(default=None)
     search_score_cutoff: float = dataclasses.field(default=None)
     include_bm25_retrieval: bool = dataclasses.field(default=None)

abacusai/api_endpoint.py

@@ -14,9 +14,10 @@ class ApiEndpoint(AbstractApiClass):
             externalChatEndpoint (str): The URI that can be used to access the external chat
             dashboardEndpoint (str):  The URI that the external chat will use to go back to the dashboard
             hostingDomain (str): The domain for hosted app deployments
+            publicHostingDomain (str): The domain for publicly hosted app deployments
     """
 
-    def __init__(self, client, apiEndpoint=None, predictEndpoint=None, proxyEndpoint=None, llmEndpoint=None, externalChatEndpoint=None, dashboardEndpoint=None, hostingDomain=None):
+    def __init__(self, client, apiEndpoint=None, predictEndpoint=None, proxyEndpoint=None, llmEndpoint=None, externalChatEndpoint=None, dashboardEndpoint=None, hostingDomain=None, publicHostingDomain=None):
         super().__init__(client, None)
         self.api_endpoint = apiEndpoint
         self.predict_endpoint = predictEndpoint
@@ -25,11 +26,12 @@ def __init__(self, client, apiEndpoint=None, predictEndpoint=None, proxyEndpoint
         self.external_chat_endpoint = externalChatEndpoint
         self.dashboard_endpoint = dashboardEndpoint
         self.hosting_domain = hostingDomain
+        self.public_hosting_domain = publicHostingDomain
         self.deprecated_keys = {}
 
     def __repr__(self):
-        repr_dict = {f'api_endpoint': repr(self.api_endpoint), f'predict_endpoint': repr(self.predict_endpoint), f'proxy_endpoint': repr(self.proxy_endpoint), f'llm_endpoint': repr(
-            self.llm_endpoint), f'external_chat_endpoint': repr(self.external_chat_endpoint), f'dashboard_endpoint': repr(self.dashboard_endpoint), f'hosting_domain': repr(self.hosting_domain)}
+        repr_dict = {f'api_endpoint': repr(self.api_endpoint), f'predict_endpoint': repr(self.predict_endpoint), f'proxy_endpoint': repr(self.proxy_endpoint), f'llm_endpoint': repr(self.llm_endpoint), f'external_chat_endpoint': repr(
+            self.external_chat_endpoint), f'dashboard_endpoint': repr(self.dashboard_endpoint), f'hosting_domain': repr(self.hosting_domain), f'public_hosting_domain': repr(self.public_hosting_domain)}
         class_name = "ApiEndpoint"
         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])
@@ -43,5 +45,5 @@ def to_dict(self):
             dict: The dict value representation of the class parameters
         """
         resp = {'api_endpoint': self.api_endpoint, 'predict_endpoint': self.predict_endpoint, 'proxy_endpoint': self.proxy_endpoint, 'llm_endpoint': self.llm_endpoint,
-                'external_chat_endpoint': self.external_chat_endpoint, 'dashboard_endpoint': self.dashboard_endpoint, 'hosting_domain': self.hosting_domain}
+                'external_chat_endpoint': self.external_chat_endpoint, 'dashboard_endpoint': self.dashboard_endpoint, 'hosting_domain': self.hosting_domain, 'public_hosting_domain': self.public_hosting_domain}
         return {key: value for key, value in resp.items() if value is not None and key not in self.deprecated_keys}

abacusai/audit_log.py

@@ -0,0 +1,47 @@
+from .return_class import AbstractApiClass
+
+
+class AuditLog(AbstractApiClass):
+    """
+        An audit log entry
+
+        Args:
+            client (ApiClient): An authenticated API Client instance
+            createdAt (str): The timestamp when the audit log entry was created
+            userId (str): The hashed ID of the user who performed the action
+            objectId (str): The hashed ID of the object that was affected by the action
+            action (str): The action performed (create, modify, start, stop, delete, share, hide, credential_change, login)
+            source (str): The source of the action (api, ui, pipeline, cli, system)
+            refreshPolicyId (str): The hashed ID of the refresh policy if applicable
+            pipelineId (str): The hashed ID of the pipeline if applicable
+    """
+
+    def __init__(self, client, createdAt=None, userId=None, objectId=None, action=None, source=None, refreshPolicyId=None, pipelineId=None):
+        super().__init__(client, None)
+        self.created_at = createdAt
+        self.user_id = userId
+        self.object_id = objectId
+        self.action = action
+        self.source = source
+        self.refresh_policy_id = refreshPolicyId
+        self.pipeline_id = pipelineId
+        self.deprecated_keys = {}
+
+    def __repr__(self):
+        repr_dict = {f'created_at': repr(self.created_at), f'user_id': repr(self.user_id), f'object_id': repr(self.object_id), f'action': repr(
+            self.action), f'source': repr(self.source), f'refresh_policy_id': repr(self.refresh_policy_id), f'pipeline_id': repr(self.pipeline_id)}
+        class_name = "AuditLog"
+        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 = {'created_at': self.created_at, 'user_id': self.user_id, 'object_id': self.object_id, 'action': self.action,
+                'source': self.source, 'refresh_policy_id': self.refresh_policy_id, 'pipeline_id': self.pipeline_id}
+        return {key: value for key, value in resp.items() if value is not None and key not in self.deprecated_keys}

abacusai/client.py

@@ -180,6 +180,7 @@
 from .use_case import UseCase
 from .use_case_requirements import UseCaseRequirements
 from .user import User
+from .user_group_object_permission import UserGroupObjectPermission
 from .web_page_response import WebPageResponse
 from .web_search_response import WebSearchResponse
 from .webhook import Webhook
@@ -218,17 +219,20 @@ async def sse_asynchronous_generator(endpoint: str, headers: dict, body: dict):
     async with aiohttp.request('POST', endpoint, json=body, headers=headers, timeout=aiohttp.ClientTimeout(total=0)) as response:
         async for line in response.content:
             if line:
-                streamed_responses = line.decode('utf-8').split('\n\n')
+                streamed_responses = line.decode('utf-8').split('\n')
                 for resp in streamed_responses:
                     if resp:
                         resp = resp.strip()
                         if resp:
-                            resp = json.loads(resp)
-                            resp = {snake_case(
-                                key): value for key, value in resp.items()}
-                            if 'ping' in resp:
-                                continue
-                            yield resp
+                            try:
+                                resp = json.loads(resp)
+                                resp = {snake_case(
+                                    key): value for key, value in resp.items()}
+                                if 'ping' in resp:
+                                    continue
+                                yield resp
+                            except Exception:
+                                pass
 
 
 def _requests_retry_session(retries=5, backoff_factor=0.1, status_forcelist=(502, 503, 504), session=None, retry_500: bool = False):
@@ -663,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.66'
+    client_version = '1.4.67'
 
     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
@@ -2777,6 +2781,14 @@ def get_deployment_conversation(self, deployment_conversation_id: str = None, ex
             DeploymentConversation: The deployment conversation."""
         return self._proxy_request('getDeploymentConversation', 'GET', query_params={'deploymentConversationId': deployment_conversation_id, 'externalSessionId': external_session_id, 'deploymentId': deployment_id, 'filterIntermediateConversationEvents': filter_intermediate_conversation_events, 'getUnusedDocumentUploads': get_unused_document_uploads, 'start': start, 'limit': limit, 'includeAllVersions': include_all_versions}, parse_type=DeploymentConversation, is_sync=True)
 
+    def get_deployment_conversation_file(self, deployment_conversation_id: str, file_path: str) -> io.BytesIO:
+        """Gets a deployment conversation file.
+
+        Args:
+            deployment_conversation_id (str): Unique ID of the conversation.
+            file_path (str): The path of the file to get."""
+        return self._proxy_request('getDeploymentConversationFile', 'GET', query_params={'deploymentConversationId': deployment_conversation_id, 'filePath': file_path}, is_sync=True, streamable_response=True)
+
     def list_deployment_conversations(self, deployment_id: str = None, external_application_id: str = None, conversation_type: Union[DeploymentConversationType, str] = None, fetch_last_llm_info: bool = False, limit: int = None, search: str = None) -> List[DeploymentConversation]:
         """Lists all conversations for the given deployment and current user.
 
@@ -2803,6 +2815,16 @@ def export_deployment_conversation(self, deployment_conversation_id: str = None,
             DeploymentConversationExport: The deployment conversation html export."""
         return self._proxy_request('exportDeploymentConversation', 'GET', query_params={'deploymentConversationId': deployment_conversation_id, 'externalSessionId': external_session_id}, parse_type=DeploymentConversationExport, is_sync=True)
 
+    def list_user_group_object_permissions(self) -> List[UserGroupObjectPermission]:
+        """List all user groups permissions associated with the objects in the organization.
+
+        If no associated user groups, the object is public among developers and admins.
+
+
+        Returns:
+            list[UserGroupObjectPermission]: List of user group object permissions in the organization."""
+        return self._call_api('listUserGroupObjectPermissions', 'GET', query_params={}, parse_type=UserGroupObjectPermission)
+
     def get_app_user_group(self, user_group_id: str) -> AppUserGroup:
         """Gets an App User Group.
 
@@ -7691,6 +7713,14 @@ def get_conversation_response_with_binary_data(self, deployment_id: str, deploym
             deployment_id, deployment_token) if deployment_token else None
         return self._call_api('getConversationResponseWithBinaryData', 'POST', query_params={'deploymentId': deployment_id, 'deploymentToken': deployment_token}, data={'message': json.dumps(message) if (message is not None and not isinstance(message, str)) else message, 'deploymentConversationId': json.dumps(deployment_conversation_id) if (deployment_conversation_id is not None and not isinstance(deployment_conversation_id, str)) else deployment_conversation_id, 'externalSessionId': json.dumps(external_session_id) if (external_session_id is not None and not isinstance(external_session_id, str)) else external_session_id, 'llmName': json.dumps(llm_name) if (llm_name is not None and not isinstance(llm_name, str)) else llm_name, 'numCompletionTokens': json.dumps(num_completion_tokens) if (num_completion_tokens is not None and not isinstance(num_completion_tokens, str)) else num_completion_tokens, 'systemMessage': json.dumps(system_message) if (system_message is not None and not isinstance(system_message, str)) else system_message, 'temperature': json.dumps(temperature) if (temperature is not None and not isinstance(temperature, str)) else temperature, 'filterKeyValues': json.dumps(filter_key_values) if (filter_key_values is not None and not isinstance(filter_key_values, str)) else filter_key_values, 'searchScoreCutoff': json.dumps(search_score_cutoff) if (search_score_cutoff is not None and not isinstance(search_score_cutoff, str)) else search_score_cutoff, 'chatConfig': json.dumps(chat_config) if (chat_config is not None and not isinstance(chat_config, str)) else chat_config}, files=attachments, server_override=prediction_url)
 
+    def get_deep_agent_response(self, message: str, deployment_conversation_id: str = None) -> Dict:
+        """Return a DeepAgent response with generated files if any, based on the input message.
+
+        Args:
+            message (str): The user's message/task for DeepAgent to complete
+            deployment_conversation_id (str): The unique identifier of a deployment conversation to continue. If not specified, a new one will be created."""
+        return self._proxy_request('getDeepAgentResponse', 'POST', query_params={}, body={'message': message, 'deploymentConversationId': deployment_conversation_id}, is_sync=True)
+
     def get_search_results(self, deployment_token: str, deployment_id: str, query_data: dict, num: int = 15) -> Dict:
         """Return the most relevant search results to the search query from the uploaded documents.
 

abacusai/deployment_conversation.py

@@ -118,6 +118,15 @@ def get(self, external_session_id: str = None, deployment_id: str = None, filter
         """
         return self.client.get_deployment_conversation(self.deployment_conversation_id, external_session_id, deployment_id, filter_intermediate_conversation_events, get_unused_document_uploads, start, limit, include_all_versions)
 
+    def get_file(self, file_path: str):
+        """
+        Gets a deployment conversation file.
+
+        Args:
+            file_path (str): The path of the file to get.
+        """
+        return self.client.get_deployment_conversation_file(self.deployment_conversation_id, file_path)
+
     def delete(self, deployment_id: str = None):
         """
         Delete a Deployment Conversation.

abacusai/document_retriever_version_logs.py

@@ -0,0 +1,33 @@
+from .return_class import AbstractApiClass
+
+
+class DocumentRetrieverVersionLogs(AbstractApiClass):
+    """
+        Logs from document retriever version.
+
+        Args:
+            client (ApiClient): An authenticated API Client instance
+            logs (list[str]): List of logs from document retriever version.
+    """
+
+    def __init__(self, client, logs=None):
+        super().__init__(client, None)
+        self.logs = logs
+        self.deprecated_keys = {}
+
+    def __repr__(self):
+        repr_dict = {f'logs': repr(self.logs)}
+        class_name = "DocumentRetrieverVersionLogs"
+        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 = {'logs': self.logs}
+        return {key: value for key, value in resp.items() if value is not None and key not in self.deprecated_keys}