abacusai 1.3.9__tar.gz → 1.4.0__tar.gz

{abacusai-1.3.9/abacusai.egg-info → abacusai-1.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: abacusai
-Version: 1.3.9
+Version: 1.4.0
 Summary: Abacus.AI Python Client Library
 Home-page: https://github.com/abacusai/api-python
 Author: Abacus.AI

{abacusai-1.3.9 → abacusai-1.4.0}/abacusai/__init__.py

@@ -4,4 +4,4 @@ from .prediction_client import PredictionClient
 from .streaming_client import StreamingClient
 
 
-__version__ = "1.3.9"
+__version__ = "1.4.0"

{abacusai-1.3.9 → abacusai-1.4.0}/abacusai/api_class/abstract.py

@@ -1,3 +1,4 @@
+import ast
 import dataclasses
 import datetime
 import inspect
@@ -52,6 +53,56 @@ def get_clean_function_source_code(func: Callable):
     return source_code
 
 
+def get_clean_function_source_code_for_agent(func: Callable):
+    sample_lambda = (lambda: 0)
+    if isinstance(func, type(sample_lambda)) and func.__name__ == sample_lambda.__name__:
+        raise ValueError('Lambda function not allowed.')
+    source_code = get_source_code(func)
+    # If function source code has some initial indentation, remove it (Ex - can happen if the functor was defined inside a function)
+    source_code = dedent(source_code)
+    return source_code
+
+
+def get_source_code(func: Callable):
+    main_function_name = func.__name__
+    source_code = inspect.getsource(func)
+    source_code = dedent(source_code)
+
+    function_globals = func.__globals__
+    tree = ast.parse(source_code)
+    call_nodes = []
+    for node in ast.walk(tree):
+        if isinstance(node, ast.Call):
+            if isinstance(node.func, ast.Name):
+                call_nodes.append(node.func.id)
+
+    functions_included = {}
+
+    while True:
+        if len(call_nodes) == 0:
+            break
+        new_call_nodes = []
+        for function_name in call_nodes:
+            if function_name not in functions_included and function_name in function_globals:
+                function_callable = function_globals[function_name]
+                if inspect.isfunction(function_callable) and function_callable.__module__ == '__main__':
+                    cur_node_source_code = inspect.getsource(function_globals[function_name])
+                    cur_node_source_code = dedent(cur_node_source_code)
+                    functions_included[function_name] = cur_node_source_code
+                    cur_node_tree = ast.parse(cur_node_source_code)
+                    cur_node_call_nodes = []
+                    for node in ast.walk(cur_node_tree):
+                        if isinstance(node, ast.Call):
+                            if isinstance(node.func, ast.Name):
+                                cur_node_call_nodes.append(node.func.id)
+                    new_call_nodes.extend(cur_node_call_nodes)
+        call_nodes = new_call_nodes
+
+    functions_included.pop(main_function_name, None)
+    final_source_code = '\n\n'.join([value for value in functions_included.values()]) + '\n\n' + source_code
+    return final_source_code
+
+
 @dataclasses.dataclass
 class ApiClass(ABC):
     _upper_snake_case_keys: bool = dataclasses.field(default=False, repr=False, init=False)

{abacusai-1.3.9 → abacusai-1.4.0}/abacusai/api_class/ai_agents.py

@@ -2,7 +2,7 @@ import dataclasses
 from typing import List, Union
 
 from . import enums
-from .abstract import ApiClass, get_clean_function_source_code
+from .abstract import ApiClass, get_clean_function_source_code_for_agent
 
 
 @dataclasses.dataclass
@@ -184,7 +184,7 @@ class WorkflowGraphNode(ApiClass):
     def __init__(self, name: str, input_mappings: List[WorkflowNodeInputMapping], output_mappings: List[WorkflowNodeOutputMapping], function: callable = None, function_name: str = None, source_code: str = None, input_schema: WorkflowNodeInputSchema = None, output_schema: WorkflowNodeOutputSchema = None):
         if function:
             self.function_name = function.__name__
-            self.source_code = get_clean_function_source_code(function)
+            self.source_code = get_clean_function_source_code_for_agent(function)
         elif function_name and source_code:
             self.function_name = function_name
             self.source_code = source_code

{abacusai-1.3.9 → abacusai-1.4.0}/abacusai/api_class/enums.py

@@ -380,6 +380,7 @@ class ApplicationConnectorType(ApiEnum):
     FRESHSERVICE = 'FRESHSERVICE'
     ZENDESKSUNSHINEMESSAGING = 'ZENDESKSUNSHINEMESSAGING'
     GOOGLEDRIVEUSER = 'GOOGLEDRIVEUSER'
+    ONEDRIVEUSER = 'ONEDRIVEUSER'
 
 
 class StreamingConnectorType(ApiEnum):

{abacusai-1.3.9 → abacusai-1.4.0}/abacusai/api_class/segments.py

@@ -50,6 +50,9 @@ class ResponseSection(ApiClass):
         return {k: v.value if isinstance(v, enums.Enum) else v for k, v in dataclasses.asdict(self).items()}
 
 
+Segment = ResponseSection
+
+
 @dataclasses.dataclass
 class AttachmentsResponseSection(ResponseSection):
     """
@@ -203,7 +206,7 @@ class CollapseResponseSection(ResponseSection):
     content: ResponseSection
 
     def __init__(self, title: str, content: ResponseSection, section_key: str = None):
-        super().__init__(enums.ResponseSectionType.COLLAPSIBLE_COMPONENT, id=section_key)  # Enum typos are hard to fix.
+        super().__init__(enums.ResponseSectionType.COLLAPSIBLE_COMPONENT, id=section_key)
         self.title = title
         self.content = content
 

{abacusai-1.3.9 → abacusai-1.4.0}/abacusai/batch_prediction.py

@@ -84,7 +84,7 @@ class BatchPrediction(AbstractApiClass):
             BatchPredictionArgs, globalPredictionArgs)
         self.batch_prediction_args = client._build_class(getattr(
             api_class, batchPredictionArgsType, BatchPredictionArgs) if batchPredictionArgsType else BatchPredictionArgs, batchPredictionArgs)
-        self.deprecated_keys = {'global_prediction_args', 'explanations'}
+        self.deprecated_keys = {'explanations', 'global_prediction_args'}
 
     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-1.3.9 → abacusai-1.4.0}/abacusai/batch_prediction_version.py

@@ -100,7 +100,7 @@ class BatchPredictionVersion(AbstractApiClass):
             BatchPredictionArgs, globalPredictionArgs)
         self.batch_prediction_args = client._build_class(getattr(
             api_class, batchPredictionArgsType, BatchPredictionArgs) if batchPredictionArgsType else BatchPredictionArgs, batchPredictionArgs)
-        self.deprecated_keys = {'global_prediction_args', 'explanations'}
+        self.deprecated_keys = {'explanations', 'global_prediction_args'}
 
     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-1.3.9 → abacusai-1.4.0}/abacusai/client.py

@@ -41,9 +41,9 @@ from .api_class import (
     PredictionArguments, ProblemType, ProjectFeatureGroupConfig,
     PythonFunctionType, ResponseSection, SamplingConfig,
     StreamingConnectorDatasetConfig, TrainingConfig, VectorStoreConfig,
-    WorkflowGraph, get_clean_function_source_code
+    WorkflowGraph, get_clean_function_source_code_for_agent
 )
-from .api_class.abstract import get_clean_function_source_code, snake_case
+from .api_class.abstract import get_clean_function_source_code, get_clean_function_source_code_for_agent, snake_case
 from .api_class.ai_agents import WorkflowGraph
 from .api_class.blob_input import Blob, BlobInput
 from .api_class.segments import ResponseSection
@@ -149,6 +149,7 @@ from .pipeline_step_version import PipelineStepVersion
 from .pipeline_step_version_logs import PipelineStepVersionLogs
 from .pipeline_version import PipelineVersion
 from .pipeline_version_logs import PipelineVersionLogs
+from .prediction_log_record import PredictionLogRecord
 from .prediction_operator import PredictionOperator
 from .prediction_operator_version import PredictionOperatorVersion
 from .problem_type import ProblemType
@@ -594,7 +595,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.3.9'
+    client_version = '1.4.0'
 
     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
@@ -2084,6 +2085,19 @@ class ReadOnlyClient(BaseApiClient):
             ModelTrainingTypeForDeployment: Model training types for deployment."""
         return self._call_api('getModelTrainingTypesForDeployment', 'GET', query_params={'modelId': model_id, 'modelVersion': model_version, 'algorithm': algorithm}, parse_type=ModelTrainingTypeForDeployment)
 
+    def get_prediction_logs_records(self, deployment_id: str, limit: int = 10, start_after_request_id: str = '', start_after_timestamp: int = None) -> List[PredictionLogRecord]:
+        """Retrieves the prediction request IDs for the most recent predictions made to the deployment.
+
+        Args:
+            deployment_id (str): The unique identifier of a deployment created under the project.
+            limit (int): The number of prediction log entries to retrieve up to the specified limit.
+            start_after_request_id (str): The request ID of the last log entry to retrieve.
+            start_after_timestamp (int): A Unix timestamp in milliseconds specifying the start point for retrieving log entries.
+
+        Returns:
+            list[PredictionLogRecord]: A list of prediction log records."""
+        return self._call_api('getPredictionLogsRecords', 'GET', query_params={'deploymentId': deployment_id, 'limit': limit, 'startAfterRequestId': start_after_request_id, 'startAfterTimestamp': start_after_timestamp}, parse_type=PredictionLogRecord)
+
     def list_deployment_alerts(self, deployment_id: str) -> List[MonitorAlert]:
         """List the monitor alerts associated with the deployment id.
 
@@ -3520,7 +3534,8 @@ class ApiClient(ReadOnlyClient):
             evaluation_feature_group_id (str): The ID of the feature group to use for evaluation.
             workflow_graph (WorkflowGraph): The workflow graph for the agent.
         """
-        function_source_code = get_clean_function_source_code(agent_function)
+        function_source_code = get_clean_function_source_code_for_agent(
+            agent_function)
         agent_function_name = agent_function.__name__
         return self.create_agent(project_id=project_id, function_source_code=function_source_code, agent_function_name=agent_function_name, name=name, memory=memory, package_requirements=package_requirements, description=description, evaluation_feature_group_id=evaluation_feature_group_id, workflow_graph=workflow_graph)
 
@@ -3538,7 +3553,8 @@ class ApiClient(ReadOnlyClient):
             description (str): A description of the agent.
             workflow_graph (WorkflowGraph): The workflow graph for the agent.
         """
-        function_source_code = get_clean_function_source_code(agent_function)
+        function_source_code = get_clean_function_source_code_for_agent(
+            agent_function)
         agent_function_name = agent_function.__name__
         return self.update_agent(model_id=model_id, function_source_code=function_source_code, agent_function_name=agent_function_name, memory=memory, package_requirements=package_requirements, enable_binary_input=enable_binary_input, description=description, workflow_graph=workflow_graph)
 
@@ -3672,7 +3688,13 @@ class ApiClient(ReadOnlyClient):
         Returns:
             dict: Containing email and name of the end user.
         """
-        return get_object_from_context(self, _request_context, 'user_info', dict)
+        user_info = get_object_from_context(
+            self, _request_context, 'user_info', dict)
+        if user_info:
+            return user_info
+        else:
+            raise ValueError(
+                'User information not available. Please use UI interface for this agent to work.')
 
     def clear_agent_context(self):
         """
@@ -7389,17 +7411,18 @@ Creates a new feature group defined as the union of other feature group versions
             streaming_token, feature_group_id=feature_group_id)
         return self._call_api('appendMultipleData', 'POST', query_params={'streamingToken': streaming_token}, body={'featureGroupId': feature_group_id, 'data': data}, server_override=prediction_url)
 
-    def upsert_data(self, feature_group_id: str, data: dict, streaming_token: str = None) -> FeatureGroupRow:
+    def upsert_data(self, feature_group_id: str, data: dict, streaming_token: str = None, blobs: None = None) -> FeatureGroupRow:
         """Update new data into the feature group for a given lookup key record ID if the record ID is found; otherwise, insert new data into the feature group.
 
         Args:
             feature_group_id (str): A unique string identifier of the online feature group to record data to.
             data (dict): The data to record, in JSON format.
             streaming_token (str): Optional streaming token for authenticating requests if upserting to streaming FG.
+            blobs (None): A dictionary of binary data to populate file fields' in data to upsert to the streaming FG.
 
         Returns:
             FeatureGroupRow: The feature group row that was upserted."""
-        return self._proxy_request('upsertData', 'POST', query_params={'streamingToken': streaming_token}, body={'featureGroupId': feature_group_id, 'data': data}, parse_type=FeatureGroupRow, is_sync=True)
+        return self._proxy_request('upsertData', 'POST', query_params={}, data={'featureGroupId': feature_group_id, 'data': json.dumps(data), 'streamingToken': streaming_token}, files=blobs, parse_type=FeatureGroupRow, is_sync=True)
 
     def delete_data(self, feature_group_id: str, primary_key: str):
         """Deletes a row from the feature group given the primary key
@@ -7955,7 +7978,7 @@ Creates a new feature group defined as the union of other feature group versions
             model_id (str): A unique string identifier associated with the Model."""
         return self._call_api('setNaturalLanguageExplanation', 'POST', query_params={}, body={'shortExplanation': short_explanation, 'longExplanation': long_explanation, 'featureGroupId': feature_group_id, 'featureGroupVersion': feature_group_version, 'modelId': model_id})
 
-    def create_chat_session(self, project_id: str, name: str = None) -> ChatSession:
+    def create_chat_session(self, project_id: str = None, name: str = None) -> ChatSession:
         """Creates a chat session with Data Science Co-pilot.
 
         Args:

{abacusai-1.3.9 → abacusai-1.4.0}/abacusai/deployment.py

@@ -285,6 +285,20 @@ class Deployment(AbstractApiClass):
         """
         return self.client.set_default_prediction_arguments(self.deployment_id, prediction_arguments, set_as_override)
 
+    def get_prediction_logs_records(self, limit: int = 10, start_after_request_id: str = '', start_after_timestamp: int = None):
+        """
+        Retrieves the prediction request IDs for the most recent predictions made to the deployment.
+
+        Args:
+            limit (int): The number of prediction log entries to retrieve up to the specified limit.
+            start_after_request_id (str): The request ID of the last log entry to retrieve.
+            start_after_timestamp (int): A Unix timestamp in milliseconds specifying the start point for retrieving log entries.
+
+        Returns:
+            list[PredictionLogRecord]: A list of prediction log records.
+        """
+        return self.client.get_prediction_logs_records(self.deployment_id, limit, start_after_request_id, start_after_timestamp)
+
     def create_alert(self, alert_name: str, condition_config: Union[dict, AlertConditionConfig], action_config: Union[dict, AlertActionConfig]):
         """
         Create a deployment alert for the given conditions.

{abacusai-1.3.9 → abacusai-1.4.0}/abacusai/deployment_conversation.py

@@ -18,11 +18,13 @@ class DeploymentConversation(AbstractApiClass):
             externalApplicationId (str): The external application id associated with the deployment conversation.
             unusedDocumentUploadIds (list[str]): The list of unused document upload ids associated with the deployment conversation.
             humanizeInstructions (dict): Instructions for humanizing the conversation.
-            conversationWarning (str): Extra text associated with the deployment conversation (to show it at the bottom of chatbot)
+            conversationWarning (str): Extra text associated with the deployment conversation (to show it at the bottom of chatbot).
+            conversationType (str): The type of the conversation, which depicts the application it caters to.
+            metadata (dict): Additional backend information about the conversation.
             history (DeploymentConversationEvent): The history of the deployment conversation.
     """
 
-    def __init__(self, client, deploymentConversationId=None, name=None, deploymentId=None, createdAt=None, lastEventCreatedAt=None, externalSessionId=None, regenerateAttempt=None, externalApplicationId=None, unusedDocumentUploadIds=None, humanizeInstructions=None, conversationWarning=None, history={}):
+    def __init__(self, client, deploymentConversationId=None, name=None, deploymentId=None, createdAt=None, lastEventCreatedAt=None, externalSessionId=None, regenerateAttempt=None, externalApplicationId=None, unusedDocumentUploadIds=None, humanizeInstructions=None, conversationWarning=None, conversationType=None, metadata=None, history={}):
         super().__init__(client, deploymentConversationId)
         self.deployment_conversation_id = deploymentConversationId
         self.name = name
@@ -35,13 +37,15 @@ class DeploymentConversation(AbstractApiClass):
         self.unused_document_upload_ids = unusedDocumentUploadIds
         self.humanize_instructions = humanizeInstructions
         self.conversation_warning = conversationWarning
+        self.conversation_type = conversationType
+        self.metadata = metadata
         self.history = client._build_class(
             DeploymentConversationEvent, history)
         self.deprecated_keys = {}
 
     def __repr__(self):
-        repr_dict = {f'deployment_conversation_id': repr(self.deployment_conversation_id), f'name': repr(self.name), f'deployment_id': repr(self.deployment_id), f'created_at': repr(self.created_at), f'last_event_created_at': repr(self.last_event_created_at), f'external_session_id': repr(self.external_session_id), f'regenerate_attempt': repr(
-            self.regenerate_attempt), f'external_application_id': repr(self.external_application_id), f'unused_document_upload_ids': repr(self.unused_document_upload_ids), f'humanize_instructions': repr(self.humanize_instructions), f'conversation_warning': repr(self.conversation_warning), f'history': repr(self.history)}
+        repr_dict = {f'deployment_conversation_id': repr(self.deployment_conversation_id), f'name': repr(self.name), f'deployment_id': repr(self.deployment_id), f'created_at': repr(self.created_at), f'last_event_created_at': repr(self.last_event_created_at), f'external_session_id': repr(self.external_session_id), f'regenerate_attempt': repr(self.regenerate_attempt), f'external_application_id': repr(
+            self.external_application_id), f'unused_document_upload_ids': repr(self.unused_document_upload_ids), f'humanize_instructions': repr(self.humanize_instructions), f'conversation_warning': repr(self.conversation_warning), f'conversation_type': repr(self.conversation_type), f'metadata': repr(self.metadata), f'history': repr(self.history)}
         class_name = "DeploymentConversation"
         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])
@@ -55,7 +59,7 @@ class DeploymentConversation(AbstractApiClass):
             dict: The dict value representation of the class parameters
         """
         resp = {'deployment_conversation_id': self.deployment_conversation_id, 'name': self.name, 'deployment_id': self.deployment_id, 'created_at': self.created_at, 'last_event_created_at': self.last_event_created_at, 'external_session_id': self.external_session_id, 'regenerate_attempt': self.regenerate_attempt,
-                'external_application_id': self.external_application_id, 'unused_document_upload_ids': self.unused_document_upload_ids, 'humanize_instructions': self.humanize_instructions, 'conversation_warning': self.conversation_warning, 'history': self._get_attribute_as_dict(self.history)}
+                'external_application_id': self.external_application_id, 'unused_document_upload_ids': self.unused_document_upload_ids, 'humanize_instructions': self.humanize_instructions, 'conversation_warning': self.conversation_warning, 'conversation_type': self.conversation_type, 'metadata': self.metadata, 'history': self._get_attribute_as_dict(self.history)}
         return {key: value for key, value in resp.items() if value is not None and key not in self.deprecated_keys}
 
     def get(self, external_session_id: str = None, deployment_id: str = None, deployment_token: str = None, filter_intermediate_conversation_events: bool = True, get_unused_document_uploads: bool = False):

{abacusai-1.3.9 → abacusai-1.4.0}/abacusai/feature_group.py

@@ -1028,18 +1028,19 @@ class FeatureGroup(AbstractApiClass):
         """
         return self.client.append_multiple_data(self.feature_group_id, streaming_token, data)
 
-    def upsert_data(self, data: dict, streaming_token: str = None):
+    def upsert_data(self, data: dict, streaming_token: str = None, blobs: None = None):
         """
         Update new data into the feature group for a given lookup key record ID if the record ID is found; otherwise, insert new data into the feature group.
 
         Args:
             data (dict): The data to record, in JSON format.
             streaming_token (str): Optional streaming token for authenticating requests if upserting to streaming FG.
+            blobs (None): A dictionary of binary data to populate file fields' in data to upsert to the streaming FG.
 
         Returns:
             FeatureGroupRow: The feature group row that was upserted.
         """
-        return self.client.upsert_data(self.feature_group_id, data, streaming_token)
+        return self.client.upsert_data(self.feature_group_id, data, streaming_token, blobs)
 
     def delete_data(self, primary_key: str):
         """

abacusai-1.4.0/abacusai/prediction_log_record.py

@@ -0,0 +1,43 @@
+from .return_class import AbstractApiClass
+
+
+class PredictionLogRecord(AbstractApiClass):
+    """
+        A Record for a prediction request log.
+
+        Args:
+            client (ApiClient): An authenticated API Client instance
+            requestId (str): The unique identifier of the prediction request.
+            query (dict): The query used to make the prediction.
+            queryTimeMs (int): The time taken to make the prediction.
+            timestampMs (str): The timestamp of the prediction request.
+            response (dict): The prediction response.
+    """
+
+    def __init__(self, client, requestId=None, query=None, queryTimeMs=None, timestampMs=None, response=None):
+        super().__init__(client, None)
+        self.request_id = requestId
+        self.query = query
+        self.query_time_ms = queryTimeMs
+        self.timestamp_ms = timestampMs
+        self.response = response
+        self.deprecated_keys = {}
+
+    def __repr__(self):
+        repr_dict = {f'request_id': repr(self.request_id), f'query': repr(self.query), f'query_time_ms': repr(
+            self.query_time_ms), f'timestamp_ms': repr(self.timestamp_ms), f'response': repr(self.response)}
+        class_name = "PredictionLogRecord"
+        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 = {'request_id': self.request_id, 'query': self.query, 'query_time_ms': self.query_time_ms,
+                'timestamp_ms': self.timestamp_ms, 'response': self.response}
+        return {key: value for key, value in resp.items() if value is not None and key not in self.deprecated_keys}

abacusai-1.4.0/abacusai/public.pem

@@ -0,0 +1,5 @@
+-----BEGIN RSA PUBLIC KEY-----
+MIGJAoGBAJaz36Q/miuC7s406vJCCGlHc8ic7gBES4eoD8nCAkWOtgNvH0ir/6JA
+csz2yJfOO1x+LlxQf42I2qs3ZnOUiKHEWzopBo9wklb/3Olan0YIIb8ESiiN+4e1
+X7iYbcyL37tLk2mJxO5rNVipIy58tu9mna7idycLEkq7/ypne6knAgMBAAE=
+-----END RSA PUBLIC KEY-----

{abacusai-1.3.9 → abacusai-1.4.0}/abacusai/return_class.py

@@ -17,3 +17,12 @@ class AbstractApiClass:
             return [elem.to_dict() for elem in attribute if elem]
         elif attribute is not None:
             return attribute.to_dict()
+
+    def __getattribute__(self, __name: str):
+        try:
+            return super().__getattribute__(__name)
+        except AttributeError as e:
+            attributes = {f"'{attr}'" for attr in self.__dict__.keys() if attr not in {
+                'client', 'deprecated_keys'}}
+            raise AttributeError(
+                f'{e}. Did you mean one of {", ".join(attributes)}?') from None

{abacusai-1.3.9 → abacusai-1.4.0/abacusai.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: abacusai
-Version: 1.3.9
+Version: 1.4.0
 Summary: Abacus.AI Python Client Library
 Home-page: https://github.com/abacusai/api-python
 Author: Abacus.AI

{abacusai-1.3.9 → abacusai-1.4.0}/abacusai.egg-info/SOURCES.txt

@@ -177,6 +177,7 @@ abacusai/prediction_client.py
 abacusai/prediction_dataset.py
 abacusai/prediction_feature_group.py
 abacusai/prediction_input.py
+abacusai/prediction_log_record.py
 abacusai/prediction_operator.py
 abacusai/prediction_operator_version.py
 abacusai/problem_type.py

{abacusai-1.3.9 → abacusai-1.4.0}/docs/.buildinfo

@@ -1,4 +1,4 @@
 # Sphinx build info version 1
 # This file hashes the configuration used when building these files. When it is not found, a full rebuild will be done.
-config: 28a7ca0cbd7ead98a38271a62ceea39a
+config: 9cd0e17cedd3a5aab3432b70872522db
 tags: 645f666f9bcd5a90fca523b33c5a78b7

{abacusai-1.3.9 → abacusai-1.4.0}/docs/_sources/autoapi/abacusai/api_class/ai_agents/index.rst.txt

@@ -60,7 +60,7 @@ Module Contents
    Bases: :py:obj:`abacusai.api_class.abstract.ApiClass`
 
 
-   A react-jsonschema-form conformant schema for workflow node input.
+   A react-jsonschema-form conformant schema for workflow node input. To initialize a WorkflowNodeInputSchema dependent on another node's output, use from_workflow_node method.
 
    :param json_schema: The json schema for the input conformant to react-jsonschema-form specification. Must define keys like "title", "type" and "properties". Supported elements - Checkbox, Radio Button, Dropdown, Textarea, Number, Date, file upload. Not supported - Nested elements, arrays and other complex types.
    :type json_schema: dict
@@ -76,6 +76,18 @@ Module Contents
       :type:  dict
 
 
+   .. py:attribute:: schema_source
+      :type:  str
+
+
+   .. py:attribute:: schema_prop
+      :type:  str
+
+
+   .. py:attribute:: runtime_schema
+      :type:  bool
+
+
    .. py:method:: to_dict()
 
       Standardizes converting an ApiClass to dictionary.
@@ -89,6 +101,19 @@ Module Contents
 
 
 
+   .. py:method:: from_workflow_node(schema_source, schema_prop)
+      :classmethod:
+
+
+      Creates a WorkflowNodeInputSchema instance which references the schema generated by a WorkflowGraphNode.
+
+      :param schema_source: The name of the source WorkflowGraphNode.
+      :type schema_source: str
+      :param schema_prop: The name of the input schema parameter which source node outputs.
+      :type schema_prop: str
+
+
+
 .. py:class:: WorkflowNodeOutputSchema
 
    Bases: :py:obj:`abacusai.api_class.abstract.ApiClass`
@@ -200,7 +225,7 @@ Module Contents
 
 
 
-.. py:class:: WorkflowGraphNode(name, input_mappings, output_mappings, function = None, function_name = None, source_code = None, input_schema = None, output_schema = None, package_requirements = None)
+.. py:class:: WorkflowGraphNode(name, input_mappings, output_mappings, function = None, function_name = None, source_code = None, input_schema = None, output_schema = None)
 
    Bases: :py:obj:`abacusai.api_class.abstract.ApiClass`
 
@@ -219,8 +244,6 @@ Module Contents
    :type input_schema: WorkflowNodeInputSchema
    :param output_schema: The react json schema for the output to be shown on UI.
    :type output_schema: WorkflowNodeOutputSchema
-   :param package_requirements: List of package requirements for the node.
-   :type package_requirements: list
 
 
    .. py:method:: to_dict()

{abacusai-1.3.9 → abacusai-1.4.0}/docs/_sources/autoapi/abacusai/api_class/dataset_application_connector/index.rst.txt

@@ -34,12 +34,18 @@ Module Contents
 
    :param application_connector_type: The type of application connector
    :type application_connector_type: enums.ApplicationConnectorType
+   :param document_processing_config: The document processing configuration. Only valid if is_documentset is True for the dataset.
+   :type document_processing_config: DatasetDocumentProcessingConfig
 
 
    .. py:attribute:: application_connector_type
       :type:  abacusai.api_class.enums.ApplicationConnectorType
 
 
+   .. py:attribute:: document_processing_config
+      :type:  abacusai.api_class.dataset.DatasetDocumentProcessingConfig
+
+
    .. py:method:: _get_builder()
       :classmethod:
 

{abacusai-1.3.9 → abacusai-1.4.0}/docs/_sources/autoapi/abacusai/api_class/enums/index.rst.txt

@@ -70,8 +70,8 @@ Classes
    abacusai.api_class.enums.ProjectConfigType
    abacusai.api_class.enums.CPUSize
    abacusai.api_class.enums.MemorySize
-   abacusai.api_class.enums.SegmentType
-   abacusai.api_class.enums.CodeSegmentLanguage
+   abacusai.api_class.enums.ResponseSectionType
+   abacusai.api_class.enums.CodeLanguage
 
 
 Module Contents
@@ -1149,6 +1149,16 @@ Module Contents
 
 
 
+   .. py:attribute:: L_SHAPED_AUTO
+      :value: 'L-shaped Split - Automatic Time Based'
+
+
+
+   .. py:attribute:: L_SHAPED_TIMESTAMP
+      :value: 'L-shaped Split - Timestamp Based'
+
+
+
 .. py:class:: ForecastingLossFunction
 
    Bases: :py:obj:`ApiEnum`
@@ -2329,6 +2339,11 @@ Module Contents
 
 
 
+   .. py:attribute:: RUNTIME_SCHEMA
+      :value: 'RUNTIME_SCHEMA'
+
+
+
 .. py:class:: OcrMode
 
    Bases: :py:obj:`ApiEnum`
@@ -2569,7 +2584,7 @@ Module Contents
 
 
 
-.. py:class:: SegmentType
+.. py:class:: ResponseSectionType
 
    Bases: :py:obj:`ApiEnum`
 
@@ -2579,13 +2594,33 @@ Module Contents
    Derive from this class to define new enumerations.
 
 
+   .. py:attribute:: AGENT_FLOW_BUTTON
+      :value: 'agent_flow_button'
+
+
+
    .. py:attribute:: ATTACHMENTS
       :value: 'attachments'
 
 
 
-   .. py:attribute:: AGENT_FLOW_BUTTON
-      :value: 'agent_flow_button'
+   .. py:attribute:: BASE64_IMAGE
+      :value: 'base64_image'
+
+
+
+   .. py:attribute:: CHART
+      :value: 'chart'
+
+
+
+   .. py:attribute:: CODE
+      :value: 'code'
+
+
+
+   .. py:attribute:: COLLAPSIBLE_COMPONENT
+      :value: 'collapsible_component'
 
 
 
@@ -2594,12 +2629,27 @@ Module Contents
 
 
 
+   .. py:attribute:: RUNTIME_SCHEMA
+      :value: 'runtime_schema'
+
+
+
+   .. py:attribute:: LIST
+      :value: 'list'
+
+
+
+   .. py:attribute:: TABLE
+      :value: 'table'
+
+
+
    .. py:attribute:: TEXT
       :value: 'text'
 
 
 
-.. py:class:: CodeSegmentLanguage
+.. py:class:: CodeLanguage
 
    Bases: :py:obj:`ApiEnum`