abacusai 1.3.5__tar.gz → 1.3.7__tar.gz

{abacusai-1.3.5/abacusai.egg-info → abacusai-1.3.7}/PKG-INFO

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

{abacusai-1.3.5 → abacusai-1.3.7}/abacusai/__init__.py

@@ -4,4 +4,4 @@ from .prediction_client import PredictionClient
 from .streaming_client import StreamingClient
 
 
-__version__ = "1.3.5"
+__version__ = "1.3.7"

{abacusai-1.3.5 → abacusai-1.3.7}/abacusai/api_class/ai_agents.py

@@ -238,6 +238,13 @@ class WorkflowGraph(ApiClass):
 
     @classmethod
     def from_dict(cls, graph: dict):
+        if graph.get('primary_start_node') is None:
+            non_primary_nodes = set()
+            for edges in graph.get('edges', []):
+                non_primary_nodes.add(edges['target'])
+            primary_nodes = set([node['name'] for node in graph.get('nodes', [])]) - non_primary_nodes
+            graph['primary_start_node'] = primary_nodes.pop() if primary_nodes else None
+
         return cls(
             nodes=[WorkflowGraphNode.from_dict(node) for node in graph.get('nodes', [])],
             edges=[WorkflowGraphEdge.from_dict(edge) for edge in graph.get('edges', [])],

{abacusai-1.3.5 → abacusai-1.3.7}/abacusai/api_class/batch_prediction.py

@@ -201,6 +201,16 @@ class TrainablePlugAndPlayBatchPredictionArgs(BatchPredictionArgs):
         self.problem_type = enums.ProblemType.CUSTOM_ALGORITHM
 
 
+@dataclasses.dataclass
+class AIAgentBatchPredictionArgs(BatchPredictionArgs):
+    """
+    Batch Prediction Config for the AIAgents problem type
+    """
+
+    def __post_init__(self):
+        self.problem_type = enums.ProblemType.AI_AGENT
+
+
 @dataclasses.dataclass
 class _BatchPredictionArgsFactory(_ApiClassFactory):
     config_abstract_class = BatchPredictionArgs
@@ -215,4 +225,5 @@ class _BatchPredictionArgsFactory(_ApiClassFactory):
         enums.ProblemType.THEME_ANALYSIS: ThemeAnalysisBatchPredictionArgs,
         enums.ProblemType.CHAT_LLM: ChatLLMBatchPredictionArgs,
         enums.ProblemType.CUSTOM_ALGORITHM: TrainablePlugAndPlayBatchPredictionArgs,
+        enums.ProblemType.AI_AGENT: AIAgentBatchPredictionArgs,
     }

{abacusai-1.3.5 → abacusai-1.3.7}/abacusai/api_class/enums.py

@@ -377,6 +377,7 @@ class ApplicationConnectorType(ApiEnum):
     MICROSOFTAUTH = 'MICROSOFTAUTH'
     FRESHSERVICE = 'FRESHSERVICE'
     ZENDESKSUNSHINEMESSAGING = 'ZENDESKSUNSHINEMESSAGING'
+    GOOGLEDRIVEUSER = 'GOOGLEDRIVEUSER'
 
 
 class StreamingConnectorType(ApiEnum):
@@ -438,6 +439,7 @@ class LLMName(ApiEnum):
     CLAUDE_V3_OPUS = 'CLAUDE_V3_OPUS'
     CLAUDE_V3_SONNET = 'CLAUDE_V3_SONNET'
     CLAUDE_V3_HAIKU = 'CLAUDE_V3_HAIKU'
+    CLAUDE_V3_5_SONNET = 'CLAUDE_V3_5_SONNET'
     ABACUS_GIRAFFE = 'ABACUS_GIRAFFE'
     ABACUS_GIRAFFE_LARGE = 'ABACUS_GIRAFFE_LARGE'
     LLAMA2_CHAT = 'LLAMA2_CHAT'
@@ -534,6 +536,7 @@ class OcrMode(ApiEnum):
     COMPREHENSIVE_V2 = 'COMPREHENSIVE_V2'
     COMPREHENSIVE_TABLE_MD = 'COMPREHENSIVE_TABLE_MD'
     TESSERACT_FAST = 'TESSERACT_FAST'
+    LLM = 'LLM'
 
 
 class StdDevThresholdType(ApiEnum):

{abacusai-1.3.5 → abacusai-1.3.7}/abacusai/api_class/model.py

@@ -161,6 +161,7 @@ class RegressionTrainingConfig(TrainingConfig):
         batch_size (BatchSize): Batch size.
         dropout_rate (int): Dropout percentage rate.
         pretrained_model_name (str): Enable algorithms which process text using pretrained multilingual NLP models.
+        pretrained_llm_name (str): Enable algorithms which process text using pretrained large language models.
         is_multilingual (bool): Enable algorithms which process text using pretrained multilingual NLP models.
         loss_function (RegressionLossFunction): Loss function to be used as objective for model training.
         loss_parameters (str): Loss function params in format <key>=<value>;<key>=<value>;.....
@@ -218,6 +219,7 @@ class RegressionTrainingConfig(TrainingConfig):
     batch_size: enums.BatchSize = dataclasses.field(default=None)
     dropout_rate: int = dataclasses.field(default=None)
     pretrained_model_name: str = dataclasses.field(default=None)
+    pretrained_llm_name: str = dataclasses.field(default=None)
     is_multilingual: bool = dataclasses.field(default=None)
     do_masked_language_model_pretraining: bool = dataclasses.field(default=None)
     max_tokens_in_sentence: int = dataclasses.field(default=None)
@@ -451,7 +453,7 @@ class ChatLLMTrainingConfig(TrainingConfig):
     Training config for the CHAT_LLM problem type
 
     Args:
-        document_retrievers (List[str]): List of names of document retrievers to use as vector stores of information for RAG responses.
+        document_retrievers (List[str]): List of names or IDs of document retrievers to use as vector stores of information for RAG responses.
         num_completion_tokens (int): Default for maximum number of tokens for chat answers. Reducing this will get faster responses which are more succinct.
         temperature (float): The generative LLM temperature.
         retrieval_columns (list): Include the metadata column values in the retrieved search results.
@@ -470,6 +472,7 @@ class ChatLLMTrainingConfig(TrainingConfig):
         hide_generated_sql (bool): When running data queries, hides the generated SQL in the response and will just return the table.
         disable_data_summarization (bool): After executing a query summarize the reponse and reply back with only the table and query run.
         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.
         database_connector_id (str): Database connector ID to use for the ChatLLM.
         database_connector_tables (List[str]): List of tables to use from the database connector for the ChatLLM.
         enable_code_execution (bool): Enable python code execution in the ChatLLM. This equips the LLM with a python kernel in which all its code is executed.
@@ -493,6 +496,7 @@ class ChatLLMTrainingConfig(TrainingConfig):
     hide_generated_sql: bool = dataclasses.field(default=None)
     disable_data_summarization: bool = dataclasses.field(default=None)
     search_score_cutoff: float = dataclasses.field(default=None)
+    include_bm25_retrieval: bool = dataclasses.field(default=None)
     database_connector_id: str = dataclasses.field(default=None)
     database_connector_tables: List[str] = dataclasses.field(default=None)
     enable_code_execution: bool = dataclasses.field(default=None)

{abacusai-1.3.5 → abacusai-1.3.7}/abacusai/client.py

@@ -76,6 +76,7 @@ from .deployment import Deployment
 from .deployment_auth_token import DeploymentAuthToken
 from .deployment_conversation import DeploymentConversation
 from .deployment_conversation_export import DeploymentConversationExport
+from .deployment_statistics import DeploymentStatistics
 from .document_data import DocumentData
 from .document_retriever import DocumentRetriever
 from .document_retriever_lookup_result import DocumentRetrieverLookupResult
@@ -593,7 +594,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.5'
+    client_version = '1.3.7'
 
     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
@@ -776,7 +777,7 @@ class BaseApiClient:
                 error_message, response.status_code, error_type, request_id)
         return result
 
-    def _proxy_request(self, name: str, method: str = 'POST', query_params: dict = None, body: dict = None, files=None, parse_type=None, is_sync: bool = False, streamable_response: bool = False):
+    def _proxy_request(self, name: str, method: str = 'POST', query_params: dict = None, body: dict = None, data: dict = None, files=None, parse_type=None, is_sync: bool = False, streamable_response: bool = False):
         headers = {'APIKEY': self.api_key}
         deployment_id = os.getenv('ABACUS_DEPLOYMENT_ID')
         if deployment_id:
@@ -796,12 +797,15 @@ class BaseApiClient:
         body = deepcopy(body)
         self._clean_api_objects(query_params)
         self._clean_api_objects(body)
+        if data:
+            data = deepcopy(data)
+            self._clean_api_objects(data)
 
         try:
             if is_sync:
                 sync_api_endpoint = f'{endpoint}/api/{name}'
                 response = self._request(url=sync_api_endpoint, method=method, query_params=query_params,
-                                         headers=headers, body=body, files=files, stream=streamable_response)
+                                         headers=headers, body=body, data=data, files=files, stream=streamable_response)
                 status_code = response.status_code
                 if streamable_response and status_code == 200:
                     return response.raw
@@ -814,7 +818,7 @@ class BaseApiClient:
                 create_request_endpoint = f'{endpoint}/api/create{name}Request'
                 status_request_endpoint = f'{endpoint}/api/get{name}Status'
                 create_request = self._request(url=create_request_endpoint, method='PUT' if files else 'POST',
-                                               query_params=query_params, headers=headers, body=body, files=files)
+                                               query_params=query_params, headers=headers, body=body or data, files=files)
                 status_code = create_request.status_code
                 if status_code == 200:
                     request_id = create_request.json()['request_id']
@@ -2225,6 +2229,18 @@ class ReadOnlyClient(BaseApiClient):
             BatchPredictionVersionLogs: The logs for the specified batch prediction version."""
         return self._call_api('getBatchPredictionVersionLogs', 'GET', query_params={'batchPredictionVersion': batch_prediction_version}, parse_type=BatchPredictionVersionLogs)
 
+    def get_deployment_statistics_over_time(self, deployment_id: str, start_date: str, end_date: str) -> DeploymentStatistics:
+        """Return basic access statistics for the given window
+
+        Args:
+            deployment_id (str): Unique string identifier of the deployment created under the project.
+            start_date (str): Timeline start date in ISO format.
+            end_date (str): Timeline end date in ISO format. The date range must be 7 days or less.
+
+        Returns:
+            DeploymentStatistics: Object describing Time series data of the number of requests and latency over the specified time period."""
+        return self._call_api('getDeploymentStatisticsOverTime', 'GET', query_params={'deploymentId': deployment_id, 'startDate': start_date, 'endDate': end_date}, parse_type=DeploymentStatistics)
+
     def get_data(self, feature_group_id: str, primary_key: str = None, num_rows: int = None) -> List[FeatureGroupRow]:
         """Gets the feature group rows for online updatable feature groups.
 
@@ -4383,7 +4399,7 @@ class ApiClient(ReadOnlyClient):
 
         Returns:
             AnnotationEntry: The annotation entry that was added."""
-        return self._call_api('addAnnotation', 'POST', query_params={}, body={'annotation': annotation, 'featureGroupId': feature_group_id, 'featureName': feature_name, 'docId': doc_id, 'featureGroupRowIdentifier': feature_group_row_identifier, 'annotationSource': annotation_source, 'status': status, 'comments': comments, 'projectId': project_id, 'saveMetadata': save_metadata, 'pages': pages}, parse_type=AnnotationEntry)
+        return self._proxy_request('addAnnotation', 'POST', query_params={}, body={'annotation': annotation, 'featureGroupId': feature_group_id, 'featureName': feature_name, 'docId': doc_id, 'featureGroupRowIdentifier': feature_group_row_identifier, 'annotationSource': annotation_source, 'status': status, 'comments': comments, 'projectId': project_id, 'saveMetadata': save_metadata, 'pages': pages}, parse_type=AnnotationEntry, is_sync=True)
 
     def describe_annotation(self, feature_group_id: str, feature_name: str = None, doc_id: str = None, feature_group_row_identifier: str = None) -> AnnotationEntry:
         """Get the latest annotation entry for a given feature group, feature, and document.
@@ -5156,6 +5172,13 @@ Creates a new feature group defined as the union of other feature group versions
             feature_group_id (str): Unique string identifier for the feature group to be removed."""
         return self._call_api('deleteFeatureGroup', 'DELETE', query_params={'featureGroupId': feature_group_id})
 
+    def delete_feature_group_version(self, feature_group_version: str):
+        """Deletes a Feature Group Version.
+
+        Args:
+            feature_group_version (str): String identifier for the feature group version to be removed."""
+        return self._call_api('deleteFeatureGroupVersion', 'DELETE', query_params={'featureGroupVersion': feature_group_version})
+
     def create_feature_group_version(self, feature_group_id: str, variable_bindings: dict = None) -> FeatureGroupVersion:
         """Creates a snapshot for a specified feature group. Triggers materialization of the feature group. The new version of the feature group is created after it has materialized.
 
@@ -5556,6 +5579,13 @@ Creates a new feature group defined as the union of other feature group versions
             dataset_id (str): Unique string identifier of the dataset to delete."""
         return self._call_api('deleteDataset', 'DELETE', query_params={'datasetId': dataset_id})
 
+    def delete_dataset_version(self, dataset_version: str):
+        """Deletes the specified dataset version from the organization.
+
+        Args:
+            dataset_version (str): String identifier of the dataset version to delete."""
+        return self._call_api('deleteDatasetVersion', 'DELETE', query_params={'datasetVersion': dataset_version})
+
     def get_docstore_page_data(self, doc_id: str, page: int, document_processing_config: Union[dict, DocumentProcessingConfig] = None, document_processing_version: str = None) -> PageData:
         """Returns the extracted page data for a document page.
 
@@ -5580,7 +5610,7 @@ Creates a new feature group defined as the union of other feature group versions
 
         Returns:
             DocumentData: The extracted document data."""
-        return self._call_api('getDocstoreDocumentData', 'POST', query_params={}, body={'docId': doc_id, 'documentProcessingConfig': document_processing_config, 'documentProcessingVersion': document_processing_version, 'returnExtractedPageText': return_extracted_page_text}, parse_type=DocumentData)
+        return self._proxy_request('getDocstoreDocumentData', 'POST', query_params={}, body={'docId': doc_id, 'documentProcessingConfig': document_processing_config, 'documentProcessingVersion': document_processing_version, 'returnExtractedPageText': return_extracted_page_text}, parse_type=DocumentData, is_sync=True)
 
     def extract_document_data(self, document: io.TextIOBase = None, doc_id: str = None, document_processing_config: Union[dict, DocumentProcessingConfig] = None, start_page: int = None, end_page: int = None, return_extracted_page_text: bool = False) -> DocumentData:
         """Extracts data from a document.
@@ -5595,7 +5625,7 @@ Creates a new feature group defined as the union of other feature group versions
 
         Returns:
             DocumentData: The extracted document data."""
-        return self._proxy_request('ExtractDocumentData', 'POST', query_params={}, body={'docId': doc_id, 'documentProcessingConfig': json.dumps(document_processing_config), 'startPage': start_page, 'endPage': end_page, 'returnExtractedPageText': return_extracted_page_text}, files={'document': document}, parse_type=DocumentData)
+        return self._proxy_request('ExtractDocumentData', 'POST', query_params={}, data={'docId': doc_id, 'documentProcessingConfig': json.dumps(document_processing_config), 'startPage': start_page, 'endPage': end_page, 'returnExtractedPageText': return_extracted_page_text}, files={'document': document}, parse_type=DocumentData)
 
     def get_training_config_options(self, project_id: str, feature_group_ids: List = None, for_retrain: bool = False, current_training_config: Union[dict, TrainingConfig] = None) -> List[TrainingConfigOptions]:
         """Retrieves the full initial description of the model training configuration options available for the specified project. The configuration options available are determined by the use case associated with the specified project. Refer to the [Use Case Documentation]({USE_CASES_URL}) for more information on use cases and use case-specific configuration options.
@@ -7103,7 +7133,7 @@ Creates a new feature group defined as the union of other feature group versions
             deployment_id, deployment_token) if deployment_token else None
         return self._call_api('executeConversationAgent', 'POST', query_params={'deploymentToken': deployment_token, 'deploymentId': deployment_id}, body={'arguments': arguments, 'keywordArguments': keyword_arguments, 'deploymentConversationId': deployment_conversation_id, 'externalSessionId': external_session_id, 'regenerate': regenerate, 'docInfos': doc_infos, 'agentWorkflowNodeId': agent_workflow_node_id}, server_override=prediction_url)
 
-    def lookup_matches(self, deployment_token: str, deployment_id: str, data: str = None, filters: dict = None, num: int = None, result_columns: list = None, max_words: int = None, num_retrieval_margin_words: int = None, max_words_per_chunk: int = None, score_multiplier_column: str = None, min_score: float = None, required_phrases: list = None, filter_clause: str = None, crowding_limits: Dict = None) -> List[DocumentRetrieverLookupResult]:
+    def lookup_matches(self, deployment_token: str, deployment_id: str, data: str = None, filters: dict = None, num: int = None, result_columns: list = None, max_words: int = None, num_retrieval_margin_words: int = None, max_words_per_chunk: int = None, score_multiplier_column: str = None, min_score: float = None, required_phrases: list = None, filter_clause: str = None, crowding_limits: dict = None) -> List[DocumentRetrieverLookupResult]:
         """Lookup document retrievers and return the matching documents from the document retriever deployed with given query.
 
         Original documents are splitted into chunks and stored in the document retriever. This lookup function will return the relevant chunks
@@ -7125,7 +7155,7 @@ Creates a new feature group defined as the union of other feature group versions
             min_score (float): If provided, will filter out the results with score less than the value specified.
             required_phrases (list): If provided, each result will contain at least one of the phrases in the given list. The matching is whitespace and case insensitive.
             filter_clause (str): If provided, filter the results of the query using this sql where clause.
-            crowding_limits (Dict): A dictionary mapping metadata columns to the maximum number of results per unique value of the column. This is used to ensure diversity of metadata attribute values in the results. If a particular attribute value has already reached its maximum count, further results with that same attribute value will be excluded from the final result set.
+            crowding_limits (dict): A dictionary mapping metadata columns to the maximum number of results per unique value of the column. This is used to ensure diversity of metadata attribute values in the results. If a particular attribute value has already reached its maximum count, further results with that same attribute value will be excluded from the final result set. An entry in the map can also be a map specifying the limit per attribute value rather than a single limit for all values. This allows a per value limit for attributes. If an attribute value is not present in the map its limit defaults to zero.
 
         Returns:
             list[DocumentRetrieverLookupResult]: The relevant documentation results found from the document retriever."""
@@ -8361,4 +8391,4 @@ Creates a new feature group defined as the union of other feature group versions
 
         Returns:
             list[DocumentRetrieverLookupResult]: The snippets found from the documents."""
-        return self._proxy_request('GetRelevantSnippets', 'POST', query_params={}, body={'docIds': doc_ids, 'query': query, 'documentRetrieverConfig': json.dumps(document_retriever_config), 'honorSentenceBoundary': honor_sentence_boundary, 'numRetrievalMarginWords': num_retrieval_margin_words, 'maxWordsPerSnippet': max_words_per_snippet, 'maxSnippetsPerDocument': max_snippets_per_document, 'startWordIndex': start_word_index, 'endWordIndex': end_word_index, 'includingBoundingBoxes': including_bounding_boxes, 'text': text}, files=blobs, parse_type=DocumentRetrieverLookupResult)
+        return self._proxy_request('GetRelevantSnippets', 'POST', query_params={}, data={'docIds': doc_ids, 'query': query, 'documentRetrieverConfig': json.dumps(document_retriever_config), 'honorSentenceBoundary': honor_sentence_boundary, 'numRetrievalMarginWords': num_retrieval_margin_words, 'maxWordsPerSnippet': max_words_per_snippet, 'maxSnippetsPerDocument': max_snippets_per_document, 'startWordIndex': start_word_index, 'endWordIndex': end_word_index, 'includingBoundingBoxes': including_bounding_boxes, 'text': text}, files=blobs, parse_type=DocumentRetrieverLookupResult)

abacusai-1.3.7/abacusai/custom_chat_instructions.py

@@ -0,0 +1,43 @@
+from .return_class import AbstractApiClass
+
+
+class CustomChatInstructions(AbstractApiClass):
+    """
+        Custom Chat Instructions
+
+        Args:
+            client (ApiClient): An authenticated API Client instance
+            userInformationInstructions (str): The behavior instructions for the chat.
+            responseInstructions (str): The response instructions for the chat.
+            enableCodeExecution (bool): Whether or not code execution is enabled.
+            enableImageGeneration (bool): Whether or not image generation is enabled.
+            enableWebSearch (bool): Whether or not web search is enabled.
+    """
+
+    def __init__(self, client, userInformationInstructions=None, responseInstructions=None, enableCodeExecution=None, enableImageGeneration=None, enableWebSearch=None):
+        super().__init__(client, None)
+        self.user_information_instructions = userInformationInstructions
+        self.response_instructions = responseInstructions
+        self.enable_code_execution = enableCodeExecution
+        self.enable_image_generation = enableImageGeneration
+        self.enable_web_search = enableWebSearch
+        self.deprecated_keys = {}
+
+    def __repr__(self):
+        repr_dict = {f'user_information_instructions': repr(self.user_information_instructions), f'response_instructions': repr(self.response_instructions), f'enable_code_execution': repr(
+            self.enable_code_execution), f'enable_image_generation': repr(self.enable_image_generation), f'enable_web_search': repr(self.enable_web_search)}
+        class_name = "CustomChatInstructions"
+        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 = {'user_information_instructions': self.user_information_instructions, 'response_instructions': self.response_instructions,
+                'enable_code_execution': self.enable_code_execution, 'enable_image_generation': self.enable_image_generation, 'enable_web_search': self.enable_web_search}
+        return {key: value for key, value in resp.items() if value is not None and key not in self.deprecated_keys}

{abacusai-1.3.5 → abacusai-1.3.7}/abacusai/dataset_version.py

@@ -98,6 +98,15 @@ class DatasetVersion(AbstractApiClass):
         """
         return self.client.describe_dataset_version(self.dataset_version)
 
+    def delete(self):
+        """
+        Deletes the specified dataset version from the organization.
+
+        Args:
+            dataset_version (str): String identifier of the dataset version to delete.
+        """
+        return self.client.delete_dataset_version(self.dataset_version)
+
     def get_logs(self):
         """
         Retrieves the dataset import logs.

{abacusai-1.3.5 → abacusai-1.3.7}/abacusai/deployment.py

@@ -392,6 +392,19 @@ class Deployment(AbstractApiClass):
         """
         return self.client.create_batch_prediction(self.deployment_id, table_name, name, global_prediction_args, batch_prediction_args, explanations, output_format, output_location, database_connector_id, database_output_config, refresh_schedule, csv_input_prefix, csv_prediction_prefix, csv_explanations_prefix, output_includes_metadata, result_input_columns, input_feature_groups)
 
+    def get_statistics_over_time(self, start_date: str, end_date: str):
+        """
+        Return basic access statistics for the given window
+
+        Args:
+            start_date (str): Timeline start date in ISO format.
+            end_date (str): Timeline end date in ISO format. The date range must be 7 days or less.
+
+        Returns:
+            DeploymentStatistics: Object describing Time series data of the number of requests and latency over the specified time period.
+        """
+        return self.client.get_deployment_statistics_over_time(self.deployment_id, start_date, end_date)
+
     def describe_feature_group_row_process_by_key(self, primary_key_value: str):
         """
         Gets the feature group row process.

{abacusai-1.3.5 → abacusai-1.3.7}/abacusai/deployment_conversation.py

@@ -17,11 +17,11 @@ class DeploymentConversation(AbstractApiClass):
             regenerateAttempt (int): The sequence number of regeneration. Not regenerated if 0.
             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.
-            conversationWarning (str): Extra text associated with the deployment conversation (to show it at the bottom of chatbot)
+            humanizeInstructions (dict): Instructions for humanizing 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, 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, history={}):
         super().__init__(client, deploymentConversationId)
         self.deployment_conversation_id = deploymentConversationId
         self.name = name
@@ -32,14 +32,14 @@ class DeploymentConversation(AbstractApiClass):
         self.regenerate_attempt = regenerateAttempt
         self.external_application_id = externalApplicationId
         self.unused_document_upload_ids = unusedDocumentUploadIds
-        self.conversation_warning = conversationWarning
+        self.humanize_instructions = humanizeInstructions
         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'conversation_warning': repr(self.conversation_warning), f'history': repr(self.history)}
+            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'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])
@@ -53,7 +53,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, 'conversation_warning': self.conversation_warning, 'history': self._get_attribute_as_dict(self.history)}
+                '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, '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.7/abacusai/deployment_statistics.py

@@ -0,0 +1,41 @@
+from .return_class import AbstractApiClass
+
+
+class DeploymentStatistics(AbstractApiClass):
+    """
+        A set of statistics for a realtime deployment.
+
+        Args:
+            client (ApiClient): An authenticated API Client instance
+            requestSeries (list): A list of the number of requests per second.
+            latencySeries (list): A list of the latency in milliseconds for each request.
+            dateLabels (list): A list of date labels for each point in the series.
+            httpStatusSeries (list): A list of the HTTP status codes for each request.
+    """
+
+    def __init__(self, client, requestSeries=None, latencySeries=None, dateLabels=None, httpStatusSeries=None):
+        super().__init__(client, None)
+        self.request_series = requestSeries
+        self.latency_series = latencySeries
+        self.date_labels = dateLabels
+        self.http_status_series = httpStatusSeries
+        self.deprecated_keys = {}
+
+    def __repr__(self):
+        repr_dict = {f'request_series': repr(self.request_series), f'latency_series': repr(
+            self.latency_series), f'date_labels': repr(self.date_labels), f'http_status_series': repr(self.http_status_series)}
+        class_name = "DeploymentStatistics"
+        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_series': self.request_series, 'latency_series': self.latency_series,
+                'date_labels': self.date_labels, 'http_status_series': self.http_status_series}
+        return {key: value for key, value in resp.items() if value is not None and key not in self.deprecated_keys}

{abacusai-1.3.5 → abacusai-1.3.7}/abacusai/document_retriever.py

@@ -209,7 +209,7 @@ class DocumentRetriever(AbstractApiClass):
         """
         return self.describe().latest_document_retriever_version.deployment_status
 
-    def get_matching_documents(self, query: str, filters: dict = None, limit: int = None, result_columns: list = None, max_words: int = None, num_retrieval_margin_words: int = None, max_words_per_chunk: int = None, score_multiplier_column: str = None, min_score: float = None, required_phrases: list = None, filter_clause: str = None, crowding_limits: Dict[str, int] = None):
+    def get_matching_documents(self, query: str, filters: dict = None, limit: int = None, result_columns: list = None, max_words: int = None, num_retrieval_margin_words: int = None, max_words_per_chunk: int = None, score_multiplier_column: str = None, min_score: float = None, required_phrases: list = None, filter_clause: str = None, crowding_limits: Dict[str, Union[int, Dict[str, int]]] = None):
         """
         Lookup document retrievers and return the matching documents from the document retriever deployed with given query.
 

{abacusai-1.3.5 → abacusai-1.3.7}/abacusai/external_application.py

@@ -20,9 +20,10 @@ class ExternalApplication(AbstractApiClass):
             deploymentConversationRetentionHours (int): The retention policy for the external application.
             managedUserService (str): The external service that is managing the user accounts.
             predictionOverrides (dict): The prediction overrides for the external application.
+            isSystemCreated (bool): Whether the external application is system created.
     """
 
-    def __init__(self, client, name=None, externalApplicationId=None, deploymentId=None, description=None, logo=None, theme=None, userGroupIds=None, useCase=None, isAgent=None, status=None, deploymentConversationRetentionHours=None, managedUserService=None, predictionOverrides=None):
+    def __init__(self, client, name=None, externalApplicationId=None, deploymentId=None, description=None, logo=None, theme=None, userGroupIds=None, useCase=None, isAgent=None, status=None, deploymentConversationRetentionHours=None, managedUserService=None, predictionOverrides=None, isSystemCreated=None):
         super().__init__(client, externalApplicationId)
         self.name = name
         self.external_application_id = externalApplicationId
@@ -37,11 +38,12 @@ class ExternalApplication(AbstractApiClass):
         self.deployment_conversation_retention_hours = deploymentConversationRetentionHours
         self.managed_user_service = managedUserService
         self.prediction_overrides = predictionOverrides
+        self.is_system_created = isSystemCreated
         self.deprecated_keys = {}
 
     def __repr__(self):
-        repr_dict = {f'name': repr(self.name), f'external_application_id': repr(self.external_application_id), f'deployment_id': repr(self.deployment_id), f'description': repr(self.description), f'logo': repr(self.logo), f'theme': repr(self.theme), f'user_group_ids': repr(self.user_group_ids), f'use_case': repr(
-            self.use_case), f'is_agent': repr(self.is_agent), f'status': repr(self.status), f'deployment_conversation_retention_hours': repr(self.deployment_conversation_retention_hours), f'managed_user_service': repr(self.managed_user_service), f'prediction_overrides': repr(self.prediction_overrides)}
+        repr_dict = {f'name': repr(self.name), f'external_application_id': repr(self.external_application_id), f'deployment_id': repr(self.deployment_id), f'description': repr(self.description), f'logo': repr(self.logo), f'theme': repr(self.theme), f'user_group_ids': repr(self.user_group_ids), f'use_case': repr(self.use_case), f'is_agent': repr(
+            self.is_agent), f'status': repr(self.status), f'deployment_conversation_retention_hours': repr(self.deployment_conversation_retention_hours), f'managed_user_service': repr(self.managed_user_service), f'prediction_overrides': repr(self.prediction_overrides), f'is_system_created': repr(self.is_system_created)}
         class_name = "ExternalApplication"
         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])
@@ -54,8 +56,8 @@ class ExternalApplication(AbstractApiClass):
         Returns:
             dict: The dict value representation of the class parameters
         """
-        resp = {'name': self.name, 'external_application_id': self.external_application_id, 'deployment_id': self.deployment_id, 'description': self.description, 'logo': self.logo, 'theme': self.theme, 'user_group_ids': self.user_group_ids, 'use_case': self.use_case,
-                'is_agent': self.is_agent, 'status': self.status, 'deployment_conversation_retention_hours': self.deployment_conversation_retention_hours, 'managed_user_service': self.managed_user_service, 'prediction_overrides': self.prediction_overrides}
+        resp = {'name': self.name, 'external_application_id': self.external_application_id, 'deployment_id': self.deployment_id, 'description': self.description, 'logo': self.logo, 'theme': self.theme, 'user_group_ids': self.user_group_ids, 'use_case': self.use_case, 'is_agent': self.is_agent,
+                'status': self.status, 'deployment_conversation_retention_hours': self.deployment_conversation_retention_hours, 'managed_user_service': self.managed_user_service, 'prediction_overrides': self.prediction_overrides, 'is_system_created': self.is_system_created}
         return {key: value for key, value in resp.items() if value is not None and key not in self.deprecated_keys}
 
     def update(self, name: str = None, description: str = None, theme: dict = None, deployment_id: str = None, deployment_conversation_retention_hours: int = None, reset_retention_policy: bool = False):

{abacusai-1.3.5 → abacusai-1.3.7}/abacusai/feature_group_version.py

@@ -144,6 +144,15 @@ class FeatureGroupVersion(AbstractApiClass):
         """
         return self.client.export_feature_group_version_to_console(self.feature_group_version, export_file_format)
 
+    def delete(self):
+        """
+        Deletes a Feature Group Version.
+
+        Args:
+            feature_group_version (str): String identifier for the feature group version to be removed.
+        """
+        return self.client.delete_feature_group_version(self.feature_group_version)
+
     def get_materialization_logs(self, stdout: bool = False, stderr: bool = False):
         """
         Returns logs for a materialized feature group version.

{abacusai-1.3.5 → abacusai-1.3.7}/abacusai/prediction_client.py

@@ -687,7 +687,7 @@ class PredictionClient(BaseApiClient):
             deployment_id, deployment_token) if deployment_token else None
         return self._call_api('executeConversationAgent', 'POST', query_params={'deploymentToken': deployment_token, 'deploymentId': deployment_id}, body={'arguments': arguments, 'keywordArguments': keyword_arguments, 'deploymentConversationId': deployment_conversation_id, 'externalSessionId': external_session_id, 'regenerate': regenerate, 'docInfos': doc_infos, 'agentWorkflowNodeId': agent_workflow_node_id}, server_override=prediction_url)
 
-    def lookup_matches(self, deployment_token: str, deployment_id: str, data: str = None, filters: dict = None, num: int = None, result_columns: list = None, max_words: int = None, num_retrieval_margin_words: int = None, max_words_per_chunk: int = None, score_multiplier_column: str = None, min_score: float = None, required_phrases: list = None, filter_clause: str = None, crowding_limits: Dict = None) -> List[DocumentRetrieverLookupResult]:
+    def lookup_matches(self, deployment_token: str, deployment_id: str, data: str = None, filters: dict = None, num: int = None, result_columns: list = None, max_words: int = None, num_retrieval_margin_words: int = None, max_words_per_chunk: int = None, score_multiplier_column: str = None, min_score: float = None, required_phrases: list = None, filter_clause: str = None, crowding_limits: dict = None) -> List[DocumentRetrieverLookupResult]:
         """Lookup document retrievers and return the matching documents from the document retriever deployed with given query.
 
         Original documents are splitted into chunks and stored in the document retriever. This lookup function will return the relevant chunks
@@ -709,7 +709,7 @@ class PredictionClient(BaseApiClient):
             min_score (float): If provided, will filter out the results with score less than the value specified.
             required_phrases (list): If provided, each result will contain at least one of the phrases in the given list. The matching is whitespace and case insensitive.
             filter_clause (str): If provided, filter the results of the query using this sql where clause.
-            crowding_limits (Dict): A dictionary mapping metadata columns to the maximum number of results per unique value of the column. This is used to ensure diversity of metadata attribute values in the results. If a particular attribute value has already reached its maximum count, further results with that same attribute value will be excluded from the final result set.
+            crowding_limits (dict): A dictionary mapping metadata columns to the maximum number of results per unique value of the column. This is used to ensure diversity of metadata attribute values in the results. If a particular attribute value has already reached its maximum count, further results with that same attribute value will be excluded from the final result set. An entry in the map can also be a map specifying the limit per attribute value rather than a single limit for all values. This allows a per value limit for attributes. If an attribute value is not present in the map its limit defaults to zero.
 
         Returns:
             list[DocumentRetrieverLookupResult]: The relevant documentation results found from the document retriever."""

abacusai-1.3.7/abacusai/public.pem

@@ -0,0 +1,5 @@
+-----BEGIN RSA PUBLIC KEY-----
+MIGJAoGBAMlmFbFTRFXBQVITEYhPdGFjr1WWWE7pdEEiB+kY5V/ibtfEqjJRit0j
+lxoXxf08wYQnaU9kbRReZYCENOdlg2I+zwKjqJQkCClyjnzwnf2kHK21yNfvIyxz
+4FqxZc7OdyNZTw/vuXkJUVbakENBH1RdkWQdhTqLfw2bQUMJ18g7AgMBAAE=
+-----END RSA PUBLIC KEY-----

{abacusai-1.3.5 → abacusai-1.3.7/abacusai.egg-info}/PKG-INFO

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

{abacusai-1.3.5 → abacusai-1.3.7}/abacusai.egg-info/SOURCES.txt

@@ -34,6 +34,7 @@ abacusai/code_source.py
 abacusai/concatenation_config.py
 abacusai/cpu_gpu_memory_specs.py
 abacusai/cryptography.py
+abacusai/custom_chat_instructions.py
 abacusai/custom_loss_function.py
 abacusai/custom_metric.py
 abacusai/custom_metric_version.py
@@ -56,6 +57,7 @@ abacusai/deployment_auth_token.py
 abacusai/deployment_conversation.py
 abacusai/deployment_conversation_event.py
 abacusai/deployment_conversation_export.py
+abacusai/deployment_statistics.py
 abacusai/document_data.py
 abacusai/document_retriever.py
 abacusai/document_retriever_config.py

{abacusai-1.3.5 → abacusai-1.3.7}/setup.py

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

abacusai-1.3.5/abacusai/public.pem

@@ -1,5 +0,0 @@
------BEGIN RSA PUBLIC KEY-----
-MIGJAoGBALCV6hsp3UMyBsTMYibY3DaGNt7I26Y9G0uK3iXapWQzCAEFdROVUUZq
-12RUAxZsPfPUwyA2kgDYlew/ZQCWm/V1SNcclSjSocRggThRJOHl8tbvO7+djtv2
-U0nF9jJblLJnYmsHd/i7akU0AV1saoUbAvc8Vig4N5gnb3RE0vmDAgMBAAE=
------END RSA PUBLIC KEY-----