abacusai 1.2.5__tar.gz → 1.2.6__tar.gz

{abacusai-1.2.5/abacusai.egg-info → abacusai-1.2.6}/PKG-INFO

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

{abacusai-1.2.5 → abacusai-1.2.6}/abacusai/api_class/ai_agents.py

@@ -140,8 +140,8 @@ class WorkflowGraphNode(ApiClass):
 
     Args:
         name (str): A unique name for the workflow node.
-        input_mappings (List[WorkflowNodeInputMapping]): List of input mappings for the node. Each arg/kwarg of the node function should have a corresponding input mapping.
-        output_mappings (List[WorkflowNodeOutputMapping]): List of output mappings for the node. Each field in the returned dict/AgentResponse must have a corresponding output mapping.
+        input_mappings (List[WorkflowNodeInputMapping]): List of input mappings for the node.
+        output_mappings (List[WorkflowNodeOutputMapping]): List of output mappings for the node.
         function (callable): The callable node function reference.
         input_schema (WorkflowNodeInputSchema): The react json schema for the input form if applicable.
         output_schema (WorkflowNodeOutputSchema): The react json schema for the output if applicable.
@@ -161,8 +161,8 @@ class WorkflowGraphNode(ApiClass):
         self.name = name
         self.input_mappings = input_mappings
         self.output_mappings = output_mappings
-        self.input_schema = input_schema if input_schema else WorkflowNodeInputSchema({})
-        self.output_schema = output_schema if output_schema else WorkflowNodeOutputSchema({})
+        self.input_schema = input_schema if input_schema else {}
+        self.output_schema = output_schema if output_schema else {}
         self.package_requirements = package_requirements if package_requirements else []
 
     def to_dict(self):
@@ -212,7 +212,7 @@ class WorkflowGraphEdge(ApiClass):
 @dataclasses.dataclass
 class WorkflowGraph(ApiClass):
     """
-    An Agent workflow graph. The edges define the node invokation order. The workflow must follow linear invokation order.
+    An Agent workflow graph.
 
     Args:
         nodes (List[WorkflowGraphNode]): A list of nodes in the workflow graph.

{abacusai-1.2.5 → abacusai-1.2.6}/abacusai/api_class/batch_prediction.py

@@ -54,8 +54,10 @@ class NamedEntityExtractionBatchPredictionArgs(BatchPredictionArgs):
 
     Args:
        for_eval (bool): If True, the test fold which was created during training and used for metrics calculation will be used as input data. These predictions are hence, used for model evaluation.
+       verbose_predictions (bool): Return prediction inputs, predicted annotations and token label probabilities.
     """
     for_eval: bool = dataclasses.field(default=None)
+    verbose_predictions: bool = dataclasses.field(default=None)
 
     def __post_init__(self):
         self.problem_type = enums.ProblemType.NAMED_ENTITY_EXTRACTION

{abacusai-1.2.5 → abacusai-1.2.6}/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 = {'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-1.2.5 → abacusai-1.2.6}/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 = {'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-1.2.5 → abacusai-1.2.6}/abacusai/client.py

@@ -199,17 +199,18 @@ async def sse_asynchronous_generator(endpoint: str, body: dict):
         import aiohttp
     except Exception:
         raise Exception('Please install aiohttp to use this functionality')
-    async with aiohttp.ClientSession() as session:
-        async with session.post(endpoint, json=body) as response:
-            async for lines, _ in response.content.iter_chunks():
-                lines = lines.decode('utf-8').split('\n\n')
-                for line in lines:
-                    if line:
-                        line = line.strip()
-                        line = json.loads(line)
-                        line = {snake_case(key): value for key,
-                                value in line.items()}
-                        yield line
+    async with aiohttp.request('POST', endpoint, json=body) as response:
+        async for line in response.content:
+            if line:
+                streamed_responses = line.decode('utf-8').split('\n\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()}
+                            yield resp
 
 
 def _requests_retry_session(retries=5, backoff_factor=0.1, status_forcelist=(502, 503, 504), session=None, retry_500: bool = False):
@@ -3989,7 +3990,7 @@ class ApiClient(ReadOnlyClient):
             filter_key_values (dict): A dictionary mapping column names to a list of values to restrict the retrieved search results.
             search_score_cutoff (float): Cutoff for the document retriever score. Matching search results below this score will be ignored.
             chat_config (dict): A dictionary specifying the query chat config override.
-            ignore_documents (bool): If True, will ignore any documents and search results, and only use the messages to generate a response. 
+            ignore_documents (bool): If True, will ignore any documents and search results, and only use the messages to generate a response.
             include_search_results (bool): If True, will also return search results, if relevant. """
         body = {
             'deploymentToken': deployment_token,
@@ -8258,10 +8259,7 @@ Creates a new feature group defined as the union of other feature group versions
         return self._call_api('getDocumentSnippet', 'POST', query_params={}, body={'documentRetrieverId': document_retriever_id, 'documentId': document_id, 'startWordIndex': start_word_index, 'endWordIndex': end_word_index}, parse_type=DocumentRetrieverLookupResult)
 
     def restart_document_retriever(self, document_retriever_id: str):
-        """Restart the document retriever if it is stopped. This will start the deployment of the document retriever,
-
-        but will not wait for it to be ready. You need to call wait_until_ready to wait until the deployment is ready.
-
+        """Restart the document retriever if it is stopped.
 
         Args:
             document_retriever_id (str): A unique string identifier associated with the document retriever."""

{abacusai-1.2.5 → abacusai-1.2.6}/abacusai/deployment_conversation_event.py

@@ -27,11 +27,9 @@ class DeploymentConversationEvent(AbstractApiClass):
             agentResponse (dict): Response from the agent. Only for conversation with agents.
             error (str): The error message in case of an error.
             segments (list): The segments of the message.
-            streamedData (str): Aggregated streamed messages from the agent.
-            streamedSectionData (str): Aggregated streamed section outputs from the agent in a list.
     """
 
-    def __init__(self, client, role=None, text=None, timestamp=None, messageIndex=None, regenerateAttempt=None, modelVersion=None, searchResults=None, isUseful=None, feedback=None, feedbackType=None, docInfos=None, keywordArguments=None, inputParams=None, attachments=None, responseVersion=None, agentWorkflowNodeId=None, chatType=None, agentResponse=None, error=None, segments=None, streamedData=None, streamedSectionData=None):
+    def __init__(self, client, role=None, text=None, timestamp=None, messageIndex=None, regenerateAttempt=None, modelVersion=None, searchResults=None, isUseful=None, feedback=None, feedbackType=None, docInfos=None, keywordArguments=None, inputParams=None, attachments=None, responseVersion=None, agentWorkflowNodeId=None, chatType=None, agentResponse=None, error=None, segments=None):
         super().__init__(client, None)
         self.role = role
         self.text = text
@@ -53,13 +51,11 @@ class DeploymentConversationEvent(AbstractApiClass):
         self.agent_response = agentResponse
         self.error = error
         self.segments = segments
-        self.streamed_data = streamedData
-        self.streamed_section_data = streamedSectionData
         self.deprecated_keys = {}
 
     def __repr__(self):
-        repr_dict = {f'role': repr(self.role), f'text': repr(self.text), f'timestamp': repr(self.timestamp), f'message_index': repr(self.message_index), f'regenerate_attempt': repr(self.regenerate_attempt), f'model_version': repr(self.model_version), f'search_results': repr(self.search_results), f'is_useful': repr(self.is_useful), f'feedback': repr(self.feedback), f'feedback_type': repr(self.feedback_type), f'doc_infos': repr(self.doc_infos), f'keyword_arguments': repr(
-            self.keyword_arguments), f'input_params': repr(self.input_params), f'attachments': repr(self.attachments), f'response_version': repr(self.response_version), f'agent_workflow_node_id': repr(self.agent_workflow_node_id), f'chat_type': repr(self.chat_type), f'agent_response': repr(self.agent_response), f'error': repr(self.error), f'segments': repr(self.segments), f'streamed_data': repr(self.streamed_data), f'streamed_section_data': repr(self.streamed_section_data)}
+        repr_dict = {f'role': repr(self.role), f'text': repr(self.text), f'timestamp': repr(self.timestamp), f'message_index': repr(self.message_index), f'regenerate_attempt': repr(self.regenerate_attempt), f'model_version': repr(self.model_version), f'search_results': repr(self.search_results), f'is_useful': repr(self.is_useful), f'feedback': repr(self.feedback), f'feedback_type': repr(self.feedback_type), f'doc_infos': repr(
+            self.doc_infos), f'keyword_arguments': repr(self.keyword_arguments), f'input_params': repr(self.input_params), f'attachments': repr(self.attachments), f'response_version': repr(self.response_version), f'agent_workflow_node_id': repr(self.agent_workflow_node_id), f'chat_type': repr(self.chat_type), f'agent_response': repr(self.agent_response), f'error': repr(self.error), f'segments': repr(self.segments)}
         class_name = "DeploymentConversationEvent"
         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])
@@ -72,6 +68,6 @@ class DeploymentConversationEvent(AbstractApiClass):
         Returns:
             dict: The dict value representation of the class parameters
         """
-        resp = {'role': self.role, 'text': self.text, 'timestamp': self.timestamp, 'message_index': self.message_index, 'regenerate_attempt': self.regenerate_attempt, 'model_version': self.model_version, 'search_results': self.search_results, 'is_useful': self.is_useful, 'feedback': self.feedback, 'feedback_type': self.feedback_type, 'doc_infos': self.doc_infos, 'keyword_arguments': self.keyword_arguments,
-                'input_params': self.input_params, 'attachments': self.attachments, 'response_version': self.response_version, 'agent_workflow_node_id': self.agent_workflow_node_id, 'chat_type': self.chat_type, 'agent_response': self.agent_response, 'error': self.error, 'segments': self.segments, 'streamed_data': self.streamed_data, 'streamed_section_data': self.streamed_section_data}
+        resp = {'role': self.role, 'text': self.text, 'timestamp': self.timestamp, 'message_index': self.message_index, 'regenerate_attempt': self.regenerate_attempt, 'model_version': self.model_version, 'search_results': self.search_results, 'is_useful': self.is_useful, 'feedback': self.feedback, 'feedback_type': self.feedback_type, 'doc_infos': self.doc_infos,
+                'keyword_arguments': self.keyword_arguments, 'input_params': self.input_params, 'attachments': self.attachments, 'response_version': self.response_version, 'agent_workflow_node_id': self.agent_workflow_node_id, 'chat_type': self.chat_type, 'agent_response': self.agent_response, 'error': self.error, 'segments': self.segments}
         return {key: value for key, value in resp.items() if value is not None and key not in self.deprecated_keys}

{abacusai-1.2.5 → abacusai-1.2.6}/abacusai/document_retriever.py

@@ -131,10 +131,7 @@ class DocumentRetriever(AbstractApiClass):
 
     def restart(self):
         """
-        Restart the document retriever if it is stopped. This will start the deployment of the document retriever,
-
-        but will not wait for it to be ready. You need to call wait_until_ready to wait until the deployment is ready.
-
+        Restart the document retriever if it is stopped.
 
         Args:
             document_retriever_id (str): A unique string identifier associated with the document retriever.
@@ -143,7 +140,7 @@ class DocumentRetriever(AbstractApiClass):
 
     def wait_until_ready(self, timeout: int = 3600):
         """
-        A waiting call until document retriever is ready. It restarts the document retriever if it is stopped.
+        A waiting call until document retriever is ready.
 
         Args:
             timeout (int, optional): The waiting time given to the call to finish, if it doesn't finish by the allocated time, the call is said to be timed out. Default value given is 3600 seconds.

{abacusai-1.2.5 → abacusai-1.2.6}/abacusai/document_retriever_version.py

@@ -96,7 +96,7 @@ class DocumentRetrieverVersion(AbstractApiClass):
 
     def wait_until_ready(self, timeout=3600):
         """
-        A waiting call until the document retriever version is ready.  It restarts the document retriever if it is stopped.
+        A waiting call until the document retriever version is ready.
 
         Args:
             timeout (int, optional): The waiting time given to the call to finish, if it doesn't finish by the allocated time, the call is said to be timed out.

{abacusai-1.2.5 → abacusai-1.2.6}/abacusai/schema.py

@@ -12,7 +12,7 @@ class Schema(AbstractApiClass):
             name (str): The unique name of the feature.
             featureMapping (str): The mapping of the feature. The possible values will be based on the project's use-case. See the (Use Case Documentation)[https://api.abacus.ai/app/help/useCases] for more details.
             detectedFeatureMapping (str): Detected feature mapping for this feature
-            featureType (str): The underlying data type of each feature:  CATEGORICAL,  CATEGORICAL_LIST,  NUMERICAL,  TIMESTAMP,  TEXT,  EMAIL,  LABEL_LIST,  ENTITY_LABEL_LIST,  JSON,  OBJECT_REFERENCE,  MULTICATEGORICAL_LIST,  COORDINATE_LIST,  NUMERICAL_LIST,  TIMESTAMP_LIST,  ZIPCODE,  URL,  PAGE_INFOS,  PAGES_DOCUMENT,  TOKENS_DOCUMENT.
+            featureType (str): The underlying data type of each feature:  CATEGORICAL,  CATEGORICAL_LIST,  NUMERICAL,  TIMESTAMP,  TEXT,  EMAIL,  LABEL_LIST,  ENTITY_LABEL_LIST,  JSON,  OBJECT_REFERENCE,  MULTICATEGORICAL_LIST,  COORDINATE_LIST,  NUMERICAL_LIST,  TIMESTAMP_LIST,  ZIPCODE,  URL,  PAGE_INFOS.
             detectedFeatureType (str): The detected feature type for this feature
             dataType (str): The underlying data type of each feature:  INTEGER,  FLOAT,  STRING,  DATE,  DATETIME,  BOOLEAN,  LIST,  STRUCT,  NULL,  BINARY.
             detectedDataType (str): The detected data type for this feature

{abacusai-1.2.5 → abacusai-1.2.6/abacusai.egg-info}/PKG-INFO

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

{abacusai-1.2.5 → abacusai-1.2.6}/setup.py

@@ -1,7 +1,7 @@
 from setuptools import find_packages, setup
 
 
-__version__ = '1.2.5'
+__version__ = '1.2.6'
 
 
 setup(name='abacusai',