abacusai 1.4.0__tar.gz → 1.4.1__tar.gz

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

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

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

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

abacusai-1.4.1/abacusai/agent_conversation.py

@@ -0,0 +1,34 @@
+from .api_class import AgentConversationMessage
+from .return_class import AbstractApiClass
+
+
+class AgentConversation(AbstractApiClass):
+    """
+        List of messages with Agent chat
+
+        Args:
+            client (ApiClient): An authenticated API Client instance
+            messages (AgentConversationMessage): list of messages in the conversation with agent.
+    """
+
+    def __init__(self, client, messages={}):
+        super().__init__(client, None)
+        self.messages = client._build_class(AgentConversationMessage, messages)
+        self.deprecated_keys = {}
+
+    def __repr__(self):
+        repr_dict = {f'messages': repr(self.messages)}
+        class_name = "AgentConversation"
+        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 = {'messages': self._get_attribute_as_dict(self.messages)}
+        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-1.4.1}/abacusai/api_class/ai_agents.py

@@ -73,6 +73,8 @@ class WorkflowNodeInputSchema(ApiClass):
             schema_source (str): The name of the source WorkflowGraphNode.
             schema_prop (str): The name of the input schema parameter which source node outputs.
         """
+        if not schema_source or not schema_prop:
+            raise ValueError('input_schema', 'Valid schema_source and schema_prop must be provided for runtime schema.')
         instance = cls(json_schema={})
         instance.schema_source = schema_source
         instance.schema_prop = schema_prop
@@ -161,9 +163,11 @@ class WorkflowNodeOutputMapping(ApiClass):
 
     @classmethod
     def from_dict(cls, mapping: dict):
+        if 'name' not in mapping:
+            raise ValueError('output_mapping', 'Invalid workflow node output mapping. Must contain keys - name.')
         return cls(
             name=mapping['name'],
-            variable_type=mapping['variable_type']
+            variable_type=mapping.get('variable_type', enums.WorkflowNodeOutputType.STRING)
         )
 
 

abacusai-1.4.1/abacusai/api_class/ai_chat.py

@@ -0,0 +1,40 @@
+import dataclasses
+
+from .abstract import ApiClass
+
+
+@dataclasses.dataclass
+class HotkeyPrompt(ApiClass):
+    """
+    A config class for a Data Science Co-Pilot Hotkey
+
+    Args:
+        prompt (str): The prompt to send to Data Science Co-Pilot
+        title (str): A short, descriptive title for the prompt. If not provided, one will be automatically generated.
+    """
+    prompt: str
+    title: str = dataclasses.field(default=None)
+    disable_problem_type_context: bool = dataclasses.field(default=True)
+    ignore_history: bool = dataclasses.field(default=None)
+
+
+@dataclasses.dataclass
+class AgentConversationMessage(ApiClass):
+    """
+    Message format for agent conversation
+
+    Args:
+        is_user (bool): Whether the message is from the user.
+        text (str): The message's text.
+        document_content (str): Document text in case of any document present.
+    """
+    is_user: bool = dataclasses.field(default=None)
+    text: str = dataclasses.field(default=None)
+    document_content: str = dataclasses.field(default=None)
+
+    def to_dict(self):
+        return {
+            'is_user': self.is_user,
+            'text': self.text,
+            'document_content': self.document_content
+        }

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

@@ -612,3 +612,11 @@ class ResponseSectionType(ApiEnum):
 class CodeLanguage(ApiEnum):
     PYTHON = 'python'
     SQL = 'sql'
+
+
+class DeploymentConversationType(ApiEnum):
+    CHAT_LLM = 'CHATLLM'
+    SIMPLE_AGENT = 'SIMPLE_AGENT'
+    COMPLEX_AGENT = 'COMPLEX_AGENT'
+    WORKFLOW_AGENT = 'WORKFLOW_AGENT'
+    COPILOT = 'COPILOT'

{abacusai-1.4.0 → abacusai-1.4.1}/abacusai/api_class/model.py

@@ -537,18 +537,10 @@ class SentimentDetectionTrainingConfig(TrainingConfig):
     Args:
         sentiment_type (SentimentType): Type of sentiment to detect.
         test_split (int): Percent of dataset to use for test data. We support using a range between 5 ( i.e. 5% ) to 20 ( i.e. 20% ) of your dataset.
-        dropout_rate (float): Dropout rate for neural network.
-        batch_size (BatchSize): Batch size for neural network.
-        compute_metrics (bool): Whether to compute metrics.
     """
     sentiment_type: enums.SentimentType = dataclasses.field(default=None)
     # Data Split Params
     test_split: int = dataclasses.field(default=None)
-    # Neural Network
-    dropout_rate: float = dataclasses.field(default=None)
-    batch_size: enums.BatchSize = dataclasses.field(default=None)
-    # Metrics
-    compute_metrics: bool = dataclasses.field(default=None)
 
     def __post_init__(self):
         self.problem_type = enums.ProblemType.SENTIMENT_DETECTION
@@ -562,15 +554,10 @@ class DocumentClassificationTrainingConfig(TrainingConfig):
     Args:
         zero_shot_hypotheses (List[str]): Zero shot hypotheses. Example text: 'This text is about pricing'.
         test_split (int): Percent of dataset to use for test data. We support using a range between 5 ( i.e. 5% ) to 20 ( i.e. 20% ) of your dataset.
-        dropout_rate (float): Dropout rate for neural network.
-        batch_size (BatchSize): Batch size for neural network.
     """
     zero_shot_hypotheses: List[str] = dataclasses.field(default=None)
     # Data Split Params
     test_split: int = dataclasses.field(default=None)
-    # Neural Network
-    dropout_rate: float = dataclasses.field(default=None)
-    batch_size: enums.BatchSize = dataclasses.field(default=None)
 
     def __post_init__(self):
         self.problem_type = enums.ProblemType.DOCUMENT_CLASSIFICATION

abacusai-1.4.1/abacusai/chatllm_referral_invite.py

@@ -0,0 +1,37 @@
+from .return_class import AbstractApiClass
+
+
+class ChatllmReferralInvite(AbstractApiClass):
+    """
+        The response of the Chatllm Referral Invite for different emails
+
+        Args:
+            client (ApiClient): An authenticated API Client instance
+            userAlreadyExists (list): List of user emails not successfullt invited, because they are already registered users.
+            successfulInvites (list): List of users successfully invited.
+    """
+
+    def __init__(self, client, userAlreadyExists=None, successfulInvites=None):
+        super().__init__(client, None)
+        self.user_already_exists = userAlreadyExists
+        self.successful_invites = successfulInvites
+        self.deprecated_keys = {}
+
+    def __repr__(self):
+        repr_dict = {f'user_already_exists': repr(
+            self.user_already_exists), f'successful_invites': repr(self.successful_invites)}
+        class_name = "ChatllmReferralInvite"
+        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_already_exists': self.user_already_exists,
+                'successful_invites': self.successful_invites}
+        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-1.4.1}/abacusai/client.py

@@ -24,6 +24,7 @@ from requests.packages.urllib3.util.retry import Retry
 
 from .abacus_api import AbacusApi
 from .agent import Agent
+from .agent_conversation import AgentConversation
 from .agent_data_execution_result import AgentDataExecutionResult
 from .agent_version import AgentVersion
 from .algorithm import Algorithm
@@ -35,18 +36,18 @@ from .api_class import (
     AgentInterface, AlertActionConfig, AlertConditionConfig, ApiClass, ApiEnum,
     ApplicationConnectorDatasetConfig, BatchPredictionArgs, Blob, BlobInput,
     CPUSize, DatasetDocumentProcessingConfig, DataType,
-    DocumentProcessingConfig, EvalArtifactType, FeatureGroupExportConfig,
-    ForecastingMonitorConfig, IncrementalDatabaseConnectorConfig, LLMName,
-    MemorySize, MergeConfig, OperatorConfig, ParsingConfig,
-    PredictionArguments, ProblemType, ProjectFeatureGroupConfig,
-    PythonFunctionType, ResponseSection, SamplingConfig,
-    StreamingConnectorDatasetConfig, TrainingConfig, VectorStoreConfig,
-    WorkflowGraph, get_clean_function_source_code_for_agent
+    DeploymentConversationType, DocumentProcessingConfig, EvalArtifactType,
+    FeatureGroupExportConfig, ForecastingMonitorConfig,
+    IncrementalDatabaseConnectorConfig, LLMName, MemorySize, MergeConfig,
+    OperatorConfig, ParsingConfig, PredictionArguments, ProblemType,
+    ProjectFeatureGroupConfig, PythonFunctionType, ResponseSection,
+    SamplingConfig, Segment, StreamingConnectorDatasetConfig, TrainingConfig,
+    VectorStoreConfig, WorkflowGraph, get_clean_function_source_code_for_agent
 )
 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
+from .api_class.segments import ResponseSection, Segment
 from .api_client_utils import (
     INVALID_PANDAS_COLUMN_NAME_CHARACTERS, StreamingHandler, StreamType,
     clean_column_name, get_object_from_context
@@ -58,7 +59,6 @@ from .application_connector import ApplicationConnector
 from .batch_prediction import BatchPrediction
 from .batch_prediction_version import BatchPredictionVersion
 from .batch_prediction_version_logs import BatchPredictionVersionLogs
-from .chat_message import ChatMessage
 from .chat_session import ChatSession
 from .custom_loss_function import CustomLossFunction
 from .custom_metric import CustomMetric
@@ -268,14 +268,14 @@ class AgentResponse:
         self.data_list = []
         self.section_data_list = []
         for arg in args:
-            if isinstance(arg, Blob) or _is_json_serializable(arg) or isinstance(arg, ResponseSection):
+            if isinstance(arg, Blob) or _is_json_serializable(arg) or isinstance(arg, (ResponseSection, Segment)):
                 self.data_list.append(arg)
             else:
                 raise Exception(
                     'AgentResponse can only contain Blob, ResponseSection or json serializable objects')
 
         for key, value in kwargs.items():
-            if isinstance(value, Blob) or _is_json_serializable(value) or isinstance(value, ResponseSection):
+            if isinstance(value, Blob) or _is_json_serializable(value) or isinstance(value, (ResponseSection, Segment)):
                 self.section_data_list.append({key: value})
             else:
                 raise Exception(
@@ -595,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.4.0'
+    client_version = '1.4.1'
 
     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
@@ -2570,16 +2570,17 @@ class ReadOnlyClient(BaseApiClient):
             DeploymentConversation: The deployment conversation."""
         return self._proxy_request('getDeploymentConversation', 'GET', query_params={'deploymentConversationId': deployment_conversation_id, 'externalSessionId': external_session_id, 'deploymentId': deployment_id, 'deploymentToken': deployment_token, 'filterIntermediateConversationEvents': filter_intermediate_conversation_events, 'getUnusedDocumentUploads': get_unused_document_uploads}, parse_type=DeploymentConversation, is_sync=True)
 
-    def list_deployment_conversations(self, deployment_id: str, external_application_id: str = None) -> List[DeploymentConversation]:
+    def list_deployment_conversations(self, deployment_id: str = None, external_application_id: str = None, conversation_type: Union[DeploymentConversationType, str] = None) -> List[DeploymentConversation]:
         """Lists all conversations for the given deployment and current user.
 
         Args:
             deployment_id (str): The deployment to get conversations for.
             external_application_id (str): The external application id associated with the deployment conversation. If specified, only conversations created on that application will be listed.
+            conversation_type (DeploymentConversationType): The type of the conversation indicating its origin.
 
         Returns:
             list[DeploymentConversation]: The deployment conversations."""
-        return self._proxy_request('listDeploymentConversations', 'GET', query_params={'deploymentId': deployment_id, 'externalApplicationId': external_application_id}, parse_type=DeploymentConversation, is_sync=True)
+        return self._proxy_request('listDeploymentConversations', 'GET', query_params={'deploymentId': deployment_id, 'externalApplicationId': external_application_id, 'conversationType': conversation_type}, parse_type=DeploymentConversation, is_sync=True)
 
     def export_deployment_conversation(self, deployment_conversation_id: str = None, external_session_id: str = None) -> DeploymentConversationExport:
         """Export a Deployment Conversation.
@@ -4176,6 +4177,8 @@ class ApiClient(ReadOnlyClient):
         else:
             warnings.warn(
                 'Using local cache as no deployment id set, expected for non-deployment environment.')
+            if key not in self._cache:
+                raise Generic404Error(f'Cache key {key} does not exist', 404)
             return self._cache.get(key)
 
     def delete_scoped_cache_key(self, key: str):
@@ -4196,6 +4199,20 @@ class ApiClient(ReadOnlyClient):
                 'Using local cache as no deployment id set, expected for non-deployment environment.')
             self._cache.pop(key, None)
 
+    def set_agent_response_document_sources(self, response_document_sources: List[DocumentRetrieverLookupResult]):
+        """
+        Sets the document sources to be shown with the response on the conversation UI.
+
+        Args:
+            response_document_sources (List): List of document retriever results to be displayed in order.
+        Returns:
+            None
+        """
+        if hasattr(_request_context, 'response_document_sources'):
+            raise Exception('Document sources cannot be set more than once')
+        _request_context.agent_response_document_sources = [{'score': float(response_document_source.score), 'answer': response_document_source.document, 'source': response_document_source.document_source, 'image_ids': [
+            response_document_source.image_id]} for response_document_source in response_document_sources]
+
     def add_user_to_organization(self, email: str):
         """Invite a user to your organization. This method will send the specified email address an invitation link to join your organization.
 
@@ -5290,7 +5307,7 @@ Creates a new feature group defined as the union of other feature group versions
 
         Returns:
             UploadPart: The object 'UploadPart' which encapsulates the hash and the etag for the part that got uploaded."""
-        return self._call_api('uploadPart', 'POST', query_params={}, data={'uploadId': json.dumps(upload_id) if (upload_id is not None and not isinstance(upload_id, str)) else upload_id, 'partNumber': json.dumps(part_number) if (part_number is not None and not isinstance(part_number, str)) else part_number}, parse_type=UploadPart, files={'partData': part_data}, retry_500=True)
+        return self._proxy_request('uploadPart', 'POST', query_params={}, data={'uploadId': upload_id, 'partNumber': part_number}, files={'partData': part_data}, parse_type=UploadPart, is_sync=True)
 
     def mark_upload_complete(self, upload_id: str) -> Upload:
         """Marks an upload process as complete.
@@ -8025,7 +8042,7 @@ Creates a new feature group defined as the union of other feature group versions
             list[AbacusApi]: A list of suggested Abacus APIs"""
         return self._call_api('suggestAbacusApis', 'POST', query_params={}, body={'query': query, 'verbosity': verbosity, 'limit': limit, 'includeScores': include_scores}, parse_type=AbacusApi)
 
-    def create_deployment_conversation(self, deployment_id: str, name: str, deployment_token: str = None, external_application_id: str = None) -> DeploymentConversation:
+    def create_deployment_conversation(self, deployment_id: str = None, name: str = None, deployment_token: str = None, external_application_id: str = None) -> DeploymentConversation:
         """Creates a deployment conversation.
 
         Args:
@@ -8310,21 +8327,17 @@ Creates a new feature group defined as the union of other feature group versions
             WebSearchResponse: Results of running the search queries."""
         return self._proxy_request('SearchWebForLlm', 'POST', query_params={}, body={'queries': queries, 'searchProviders': search_providers, 'maxResults': max_results, 'safe': safe, 'fetchContent': fetch_content, 'maxPageTokens': max_page_tokens, 'convertToMarkdown': convert_to_markdown}, parse_type=WebSearchResponse)
 
-    def construct_agent_conversation_messages_for_llm(self, current_message: str = None, current_doc_ids: List = None, include_history: bool = True, include_document_contents: bool = True, deployment_conversation_id: str = None, external_session_id: str = None, max_document_words: int = None) -> ChatMessage:
+    def construct_agent_conversation_messages_for_llm(self, deployment_conversation_id: str = None, external_session_id: str = None, include_document_contents: bool = True) -> AgentConversation:
         """Returns conversation history in a format for LLM calls.
 
         Args:
-            current_message (str): Current turn message from user.
-            current_doc_ids (List): Document IDs associated with the current turn message from user.
-            include_history (bool): If true, include the conversation history in the generated messages.
-            include_document_contents (bool): If true, include contents from uploaded documents in the generated messages.
             deployment_conversation_id (str): Unique ID of the conversation. One of deployment_conversation_id or external_session_id must be provided.
             external_session_id (str): External session ID of the conversation.
-            max_document_words (int): Maximum number of words to include in the generated message from each uploaded document.
+            include_document_contents (bool): If true, include contents from uploaded documents in the generated messages.
 
         Returns:
-            ChatMessage: A list of ChatMessage that represents the conversation."""
-        return self._proxy_request('constructAgentConversationMessagesForLLM', 'POST', query_params={}, body={'currentMessage': current_message, 'currentDocIds': current_doc_ids, 'includeHistory': include_history, 'includeDocumentContents': include_document_contents, 'deploymentConversationId': deployment_conversation_id, 'externalSessionId': external_session_id, 'maxDocumentWords': max_document_words}, parse_type=ChatMessage, is_sync=True)
+            AgentConversation: Contains a list of AgentConversationMessage that represents the conversation."""
+        return self._proxy_request('constructAgentConversationMessagesForLLM', 'POST', query_params={}, body={'deploymentConversationId': deployment_conversation_id, 'externalSessionId': external_session_id, 'includeDocumentContents': include_document_contents}, parse_type=AgentConversation, is_sync=True)
 
     def get_llm_app_response(self, llm_app_name: str, prompt: str) -> LlmResponse:
         """Queries the specified LLM App to generate a response to the prompt. LLM Apps are LLMs tailored to achieve a specific task like code generation for a specific service's API.

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

@@ -2,7 +2,7 @@ from typing import Union
 
 from .api_class import (
     AlertActionConfig, AlertConditionConfig, BatchPredictionArgs,
-    FeatureGroupExportConfig, PredictionArguments
+    DeploymentConversationType, FeatureGroupExportConfig, PredictionArguments
 )
 from .feature_group_export_config import FeatureGroupExportConfig
 from .refresh_schedule import RefreshSchedule
@@ -480,7 +480,7 @@ class Deployment(AbstractApiClass):
         """
         return self.client.get_feature_group_row_process_logs_by_key(self.deployment_id, primary_key_value)
 
-    def create_conversation(self, name: str, deployment_token: str = None, external_application_id: str = None):
+    def create_conversation(self, name: str = None, deployment_token: str = None, external_application_id: str = None):
         """
         Creates a deployment conversation.
 
@@ -494,17 +494,18 @@ class Deployment(AbstractApiClass):
         """
         return self.client.create_deployment_conversation(self.deployment_id, name, deployment_token, external_application_id)
 
-    def list_conversations(self, external_application_id: str = None):
+    def list_conversations(self, external_application_id: str = None, conversation_type: Union[dict, DeploymentConversationType] = None):
         """
         Lists all conversations for the given deployment and current user.
 
         Args:
             external_application_id (str): The external application id associated with the deployment conversation. If specified, only conversations created on that application will be listed.
+            conversation_type (DeploymentConversationType): The type of the conversation indicating its origin.
 
         Returns:
             list[DeploymentConversation]: The deployment conversations.
         """
-        return self.client.list_deployment_conversations(self.deployment_id, external_application_id)
+        return self.client.list_deployment_conversations(self.deployment_id, external_application_id, conversation_type)
 
     def create_external_application(self, name: str = None, description: str = None, logo: str = None, theme: dict = None):
         """

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

@@ -137,3 +137,16 @@ class DeploymentConversation(AbstractApiClass):
             DeploymentConversationExport: The deployment conversation html export.
         """
         return self.client.export_deployment_conversation(self.deployment_conversation_id, external_session_id)
+
+    def construct_agent_conversation_messages_for_llm(self, external_session_id: str = None, include_document_contents: bool = True):
+        """
+        Returns conversation history in a format for LLM calls.
+
+        Args:
+            external_session_id (str): External session ID of the conversation.
+            include_document_contents (bool): If true, include contents from uploaded documents in the generated messages.
+
+        Returns:
+            AgentConversation: Contains a list of AgentConversationMessage that represents the conversation.
+        """
+        return self.client.construct_agent_conversation_messages_for_llm(self.deployment_conversation_id, external_session_id, include_document_contents)

{abacusai-1.4.0 → abacusai-1.4.1}/abacusai/document_retriever_lookup_result.py

@@ -12,20 +12,24 @@ class DocumentRetrieverLookupResult(AbstractApiClass):
             properties (dict): Properties of the retrieved documents.
             pages (list): Pages of the retrieved text from the original document.
             boundingBoxes (list): Bounding boxes of the retrieved text from the original document.
+            documentSource (str): Document source name.
+            imageId (str): Image ID of the document chunk.
     """
 
-    def __init__(self, client, document=None, score=None, properties=None, pages=None, boundingBoxes=None):
+    def __init__(self, client, document=None, score=None, properties=None, pages=None, boundingBoxes=None, documentSource=None, imageId=None):
         super().__init__(client, None)
         self.document = document
         self.score = score
         self.properties = properties
         self.pages = pages
         self.bounding_boxes = boundingBoxes
+        self.document_source = documentSource
+        self.image_id = imageId
         self.deprecated_keys = {}
 
     def __repr__(self):
-        repr_dict = {f'document': repr(self.document), f'score': repr(self.score), f'properties': repr(
-            self.properties), f'pages': repr(self.pages), f'bounding_boxes': repr(self.bounding_boxes)}
+        repr_dict = {f'document': repr(self.document), f'score': repr(self.score), f'properties': repr(self.properties), f'pages': repr(
+            self.pages), f'bounding_boxes': repr(self.bounding_boxes), f'document_source': repr(self.document_source), f'image_id': repr(self.image_id)}
         class_name = "DocumentRetrieverLookupResult"
         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])
@@ -38,6 +42,6 @@ class DocumentRetrieverLookupResult(AbstractApiClass):
         Returns:
             dict: The dict value representation of the class parameters
         """
-        resp = {'document': self.document, 'score': self.score, 'properties': self.properties,
-                'pages': self.pages, 'bounding_boxes': self.bounding_boxes}
+        resp = {'document': self.document, 'score': self.score, 'properties': self.properties, 'pages': self.pages,
+                'bounding_boxes': self.bounding_boxes, 'document_source': self.document_source, 'image_id': self.image_id}
         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-1.4.1}/abacusai/external_invite.py

@@ -7,7 +7,7 @@ class ExternalInvite(AbstractApiClass):
 
         Args:
             client (ApiClient): An authenticated API Client instance
-            userAlreadyInOrg (list): List of user emails not successfullt invited, because they are already in the organization.
+            userAlreadyInOrg (list): List of user emails not successfully invited, because they are already in the organization.
             userAlreadyInAppGroup (list): List of user emails not successfully invited, because they are already in the application group.
             userExistsAsInternal (list): List of user emails not successfully invited, because they are already internal users.
             successfulInvites (list): List of users successfully invited.

abacusai-1.4.1/abacusai/public.pem

@@ -0,0 +1,5 @@
+-----BEGIN RSA PUBLIC KEY-----
+MIGJAoGBAI/44p31ibP1ngQ5G1bMCoSHgPHhE64aiZw10udBmmYVG2hCLpeFMPFp
+MkEUCybxV6N0hjw8auViK8OqJZ1D5AHLV9g+aKOp+lRXAKLGuHGcCwn6aU41YLJ/
+Z4qur3X14nGvpJpJ3apSjQ7JAHz3fN9L1cvP0vzw6rMyNxZz9CrlAgMBAAE=
+-----END RSA PUBLIC KEY-----

{abacusai-1.4.0 → abacusai-1.4.1}/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,  PAGE_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,  PAGE_LABEL_LIST,  JSON,  OBJECT_REFERENCE,  MULTICATEGORICAL_LIST,  COORDINATE_LIST,  NUMERICAL_LIST,  TIMESTAMP_LIST,  ZIPCODE,  URL,  PAGE_INFOS,  PAGES_DOCUMENT,  TOKENS_DOCUMENT,  MESSAGE_LIST.
             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.4.0 → abacusai-1.4.1/abacusai.egg-info}/PKG-INFO

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

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

@@ -8,6 +8,7 @@ setup.py
 abacusai/__init__.py
 abacusai/abacus_api.py
 abacusai/agent.py
+abacusai/agent_conversation.py
 abacusai/agent_data_document_info.py
 abacusai/agent_data_execution_result.py
 abacusai/agent_version.py
@@ -29,6 +30,7 @@ abacusai/batch_prediction_version_logs.py
 abacusai/categorical_range_violation.py
 abacusai/chat_message.py
 abacusai/chat_session.py
+abacusai/chatllm_referral_invite.py
 abacusai/client.py
 abacusai/code_source.py
 abacusai/concatenation_config.py

{abacusai-1.4.0 → abacusai-1.4.1}/setup.py

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

abacusai-1.4.0/abacusai/api_class/ai_chat.py

@@ -1,18 +0,0 @@
-import dataclasses
-
-from .abstract import ApiClass
-
-
-@dataclasses.dataclass
-class HotkeyPrompt(ApiClass):
-    """
-    A config class for a Data Science Co-Pilot Hotkey
-
-    Args:
-        prompt (str): The prompt to send to Data Science Co-Pilot
-        title (str): A short, descriptive title for the prompt. If not provided, one will be automatically generated.
-    """
-    prompt: str
-    title: str = dataclasses.field(default=None)
-    disable_problem_type_context: bool = dataclasses.field(default=True)
-    ignore_history: bool = dataclasses.field(default=None)

abacusai-1.4.0/abacusai/public.pem

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