abacusai 1.4.65__tar.gz → 1.4.67__tar.gz

{abacusai-1.4.65 → abacusai-1.4.67}/PKG-INFO

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

{abacusai-1.4.65 → abacusai-1.4.67}/abacusai/__init__.py

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

{abacusai-1.4.65 → abacusai-1.4.67}/abacusai/api_class/ai_agents.py

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

{abacusai-1.4.65 → abacusai-1.4.67}/abacusai/api_class/enums.py

@@ -224,6 +224,8 @@ class FileFormat(ApiEnum):
     WEBM = 'webm'
     WMV = 'wmv'
     MSG = 'msg'
+    TS = 'TS'
+    TSX = 'TSX'
 
 
 class ExperimentationMode(ApiEnum):

{abacusai-1.4.65 → abacusai-1.4.67}/abacusai/api_class/model.py

@@ -486,6 +486,7 @@ class ChatLLMTrainingConfig(TrainingConfig):
         data_prompt_column_context (Dict[str, str]): Dict of 'table_name.column_name' and 'column_context' pairs to provide column context for some selected columns in the selected structured data table. This replaces the default auto-generated information about the column data.
         hide_sql_and_code (bool): When running data queries, this will hide the generated SQL and Code in the response.
         disable_data_summarization (bool): After executing a query summarize the reponse and reply back with only the table and query run.
+        disable_data_fetch_for_training (bool): Train using only table and column schema metadata without fetching sample data. This speeds up training but may result in less context for the model.
         data_columns_to_ignore (List[str]): Columns to ignore while encoding information about structured data tables in context for the LLM. A list of strings of format "<table_name>.<column_name>"
         search_score_cutoff (float): Minimum search score to consider a document as a valid search result.
         include_bm25_retrieval (bool): Combine BM25 search score with vector search using reciprocal rank fusion.
@@ -503,7 +504,7 @@ class ChatLLMTrainingConfig(TrainingConfig):
         mask_pii (bool): Mask PII in the prompts and uploaded documents before sending it to the LLM. Only available for Enterprise users and will cause validation errors if set to True for ChatLLM Teams users.
         builtin_tools (List[SystemConnectorTool]): List of builtin system connector tools to use in the ChatLLM. Using builtin tools does not require enabling tool bar (enable_tool_bar flag).
         mcp_servers (List[str]): List of names of MCP servers to use in the ChatLLM. This should not be used with document_retrievers.
-        agentic_loop_mode (bool): Force use of agentic loop that uses a series of tool calls when needed to respond.
+        agentic_loop_mode (bool): Enables use of agentic loop that uses a series of tool calls when needed to respond. If set to False, the agentic loop will not be used. If not set or set to Auto, the agentic loop will be automatically used based on certain conditions like presence of tools in the model.
     """
     document_retrievers: List[str] = dataclasses.field(default=None)
     num_completion_tokens: int = dataclasses.field(default=None)
@@ -525,6 +526,7 @@ class ChatLLMTrainingConfig(TrainingConfig):
     data_prompt_column_context: Dict[str, str] = dataclasses.field(default=None)
     hide_sql_and_code: bool = dataclasses.field(default=None)
     disable_data_summarization: bool = dataclasses.field(default=None)
+    disable_data_fetch_for_training: bool = dataclasses.field(default=None)
     data_columns_to_ignore: List[str] = dataclasses.field(default=None)
     search_score_cutoff: float = dataclasses.field(default=None)
     include_bm25_retrieval: bool = dataclasses.field(default=None)

{abacusai-1.4.65 → abacusai-1.4.67}/abacusai/api_endpoint.py

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

abacusai-1.4.67/abacusai/audit_log.py

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

{abacusai-1.4.65 → abacusai-1.4.67}/abacusai/client.py

@@ -180,6 +180,7 @@ from .upload_part import UploadPart
 from .use_case import UseCase
 from .use_case_requirements import UseCaseRequirements
 from .user import User
+from .user_group_object_permission import UserGroupObjectPermission
 from .web_page_response import WebPageResponse
 from .web_search_response import WebSearchResponse
 from .webhook import Webhook
@@ -218,17 +219,20 @@ async def sse_asynchronous_generator(endpoint: str, headers: dict, body: dict):
     async with aiohttp.request('POST', endpoint, json=body, headers=headers, timeout=aiohttp.ClientTimeout(total=0)) as response:
         async for line in response.content:
             if line:
-                streamed_responses = line.decode('utf-8').split('\n\n')
+                streamed_responses = line.decode('utf-8').split('\n')
                 for resp in streamed_responses:
                     if resp:
                         resp = resp.strip()
                         if resp:
-                            resp = json.loads(resp)
-                            resp = {snake_case(
-                                key): value for key, value in resp.items()}
-                            if 'ping' in resp:
-                                continue
-                            yield resp
+                            try:
+                                resp = json.loads(resp)
+                                resp = {snake_case(
+                                    key): value for key, value in resp.items()}
+                                if 'ping' in resp:
+                                    continue
+                                yield resp
+                            except Exception:
+                                pass
 
 
 def _requests_retry_session(retries=5, backoff_factor=0.1, status_forcelist=(502, 503, 504), session=None, retry_500: bool = False):
@@ -663,7 +667,7 @@ class BaseApiClient:
         client_options (ClientOptions): Optional API client configurations
         skip_version_check (bool): If true, will skip checking the server's current API version on initializing the client
     """
-    client_version = '1.4.65'
+    client_version = '1.4.67'
 
     def __init__(self, api_key: str = None, server: str = None, client_options: ClientOptions = None, skip_version_check: bool = False, include_tb: bool = False):
         self.api_key = api_key
@@ -2777,6 +2781,14 @@ 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, 'filterIntermediateConversationEvents': filter_intermediate_conversation_events, 'getUnusedDocumentUploads': get_unused_document_uploads, 'start': start, 'limit': limit, 'includeAllVersions': include_all_versions}, parse_type=DeploymentConversation, is_sync=True)
 
+    def get_deployment_conversation_file(self, deployment_conversation_id: str, file_path: str) -> io.BytesIO:
+        """Gets a deployment conversation file.
+
+        Args:
+            deployment_conversation_id (str): Unique ID of the conversation.
+            file_path (str): The path of the file to get."""
+        return self._proxy_request('getDeploymentConversationFile', 'GET', query_params={'deploymentConversationId': deployment_conversation_id, 'filePath': file_path}, is_sync=True, streamable_response=True)
+
     def list_deployment_conversations(self, deployment_id: str = None, external_application_id: str = None, conversation_type: Union[DeploymentConversationType, str] = None, fetch_last_llm_info: bool = False, limit: int = None, search: str = None) -> List[DeploymentConversation]:
         """Lists all conversations for the given deployment and current user.
 
@@ -2803,6 +2815,16 @@ class ReadOnlyClient(BaseApiClient):
             DeploymentConversationExport: The deployment conversation html export."""
         return self._proxy_request('exportDeploymentConversation', 'GET', query_params={'deploymentConversationId': deployment_conversation_id, 'externalSessionId': external_session_id}, parse_type=DeploymentConversationExport, is_sync=True)
 
+    def list_user_group_object_permissions(self) -> List[UserGroupObjectPermission]:
+        """List all user groups permissions associated with the objects in the organization.
+
+        If no associated user groups, the object is public among developers and admins.
+
+
+        Returns:
+            list[UserGroupObjectPermission]: List of user group object permissions in the organization."""
+        return self._call_api('listUserGroupObjectPermissions', 'GET', query_params={}, parse_type=UserGroupObjectPermission)
+
     def get_app_user_group(self, user_group_id: str) -> AppUserGroup:
         """Gets an App User Group.
 
@@ -2813,6 +2835,16 @@ class ReadOnlyClient(BaseApiClient):
             AppUserGroup: The App User Group."""
         return self._call_api('getAppUserGroup', 'GET', query_params={'userGroupId': user_group_id}, parse_type=AppUserGroup)
 
+    def get_app_user_group_from_name(self, name: str = None) -> AppUserGroup:
+        """Gets an App User Group by name.
+
+        Args:
+            name (str): The name of the app user group to retrieve.
+
+        Returns:
+            AppUserGroup: The app user group with the specified name."""
+        return self._call_api('getAppUserGroupFromName', 'GET', query_params={'name': name}, parse_type=AppUserGroup)
+
     def describe_external_application(self, external_application_id: str) -> ExternalApplication:
         """Describes an External Application.
 
@@ -5037,6 +5069,19 @@ class ApiClient(ReadOnlyClient):
             Project: This object represents the newly created project."""
         return self._call_api('createProject', 'POST', query_params={}, body={'name': name, 'useCase': use_case}, parse_type=Project)
 
+    def list_projects_dashboard(self, updated_filter: int = None, limit: int = 50, since_project_id: str = None, search: str = None, starred: bool = None, tag: str = None, tags: list = None) -> dict:
+        """Retrieves a list of projects for dashboard display with filtering and pagination support.
+
+        Args:
+            updated_filter (int): Unix timestamp to filter projects updated after this time.
+            limit (int): The maximum number of projects to return (default: 50).
+            since_project_id (str): The ID of the project after which the list starts for pagination.
+            search (str): Search term to filter projects by name, creator email, or use case.
+            starred (bool): Filter by starred status (True for starred, False for non-starred).
+            tag (str): Filter by a single tag (deprecated, use tags instead).
+            tags (list): List of tags to filter projects by."""
+        return self._call_api('listProjectsDashboard', 'POST', query_params={}, body={'updatedFilter': updated_filter, 'limit': limit, 'sinceProjectId': since_project_id, 'search': search, 'starred': starred, 'tag': tag, 'tags': tags})
+
     def rename_project(self, project_id: str, name: str):
         """This method renames a project after it is created.
 
@@ -5136,6 +5181,15 @@ class ApiClient(ReadOnlyClient):
             list[Feature]: A list of objects that describes the resulting feature group's schema after the feature's featureMapping is set."""
         return self._call_api('setFeatureMapping', 'POST', query_params={}, body={'projectId': project_id, 'featureGroupId': feature_group_id, 'featureName': feature_name, 'featureMapping': feature_mapping, 'nestedColumnName': nested_column_name}, parse_type=Feature)
 
+    def add_project_scope_for_user(self, project_id: str, email: str, scope: list):
+        """Add a user to a project.
+
+        Args:
+            project_id (str): The project's id.
+            email (str): The user's email.
+            scope (list): The list of project scopes."""
+        return self._call_api('addProjectScopeForUser', 'POST', query_params={}, body={'projectId': project_id, 'email': email, 'scope': scope})
+
     def add_annotation(self, annotation: dict, feature_group_id: str, feature_name: str, doc_id: str = None, feature_group_row_identifier: str = None, annotation_source: str = 'ui', status: str = None, comments: dict = None, project_id: str = None, save_metadata: bool = False, pages: list = None) -> AnnotationEntry:
         """Add an annotation entry to the database.
 
@@ -7659,6 +7713,14 @@ class ApiClient(ReadOnlyClient):
             deployment_id, deployment_token) if deployment_token else None
         return self._call_api('getConversationResponseWithBinaryData', 'POST', query_params={'deploymentId': deployment_id, 'deploymentToken': deployment_token}, data={'message': json.dumps(message) if (message is not None and not isinstance(message, str)) else message, 'deploymentConversationId': json.dumps(deployment_conversation_id) if (deployment_conversation_id is not None and not isinstance(deployment_conversation_id, str)) else deployment_conversation_id, 'externalSessionId': json.dumps(external_session_id) if (external_session_id is not None and not isinstance(external_session_id, str)) else external_session_id, 'llmName': json.dumps(llm_name) if (llm_name is not None and not isinstance(llm_name, str)) else llm_name, 'numCompletionTokens': json.dumps(num_completion_tokens) if (num_completion_tokens is not None and not isinstance(num_completion_tokens, str)) else num_completion_tokens, 'systemMessage': json.dumps(system_message) if (system_message is not None and not isinstance(system_message, str)) else system_message, 'temperature': json.dumps(temperature) if (temperature is not None and not isinstance(temperature, str)) else temperature, 'filterKeyValues': json.dumps(filter_key_values) if (filter_key_values is not None and not isinstance(filter_key_values, str)) else filter_key_values, 'searchScoreCutoff': json.dumps(search_score_cutoff) if (search_score_cutoff is not None and not isinstance(search_score_cutoff, str)) else search_score_cutoff, 'chatConfig': json.dumps(chat_config) if (chat_config is not None and not isinstance(chat_config, str)) else chat_config}, files=attachments, server_override=prediction_url)
 
+    def get_deep_agent_response(self, message: str, deployment_conversation_id: str = None) -> Dict:
+        """Return a DeepAgent response with generated files if any, based on the input message.
+
+        Args:
+            message (str): The user's message/task for DeepAgent to complete
+            deployment_conversation_id (str): The unique identifier of a deployment conversation to continue. If not specified, a new one will be created."""
+        return self._proxy_request('getDeepAgentResponse', 'POST', query_params={}, body={'message': message, 'deploymentConversationId': deployment_conversation_id}, is_sync=True)
+
     def get_search_results(self, deployment_token: str, deployment_id: str, query_data: dict, num: int = 15) -> Dict:
         """Return the most relevant search results to the search query from the uploaded documents.
 
@@ -8894,6 +8956,35 @@ class ApiClient(ReadOnlyClient):
             deployment_id (str): The deployment this conversation belongs to. This is required if not logged in."""
         return self._proxy_request('renameDeploymentConversation', 'POST', query_params={'deploymentId': deployment_id}, body={'deploymentConversationId': deployment_conversation_id, 'name': name}, is_sync=True)
 
+    def add_user_group_object_permission(self, object_id: str, user_group_id: str, object_type: str, permission: str = 'ALL'):
+        """Add user group object permission for any object type.
+
+        Args:
+            object_id (str): The ID of the object (dataset_id, project_id, external_connection_id, feature_group_id, etc.)
+            user_group_id (str): The ID of the user group
+            object_type (str): The type of object ('dataset', 'project', 'external_connection', 'feature_group')
+            permission (str): The permission to grant (default: 'ALL')"""
+        return self._call_api('addUserGroupObjectPermission', 'POST', query_params={}, body={'objectId': object_id, 'userGroupId': user_group_id, 'objectType': object_type, 'permission': permission})
+
+    def update_user_group_object_permission(self, object_id: str, user_group_id: str, object_type: str, permission: str = 'ALL'):
+        """Update user group object permission for any object type.
+
+        Args:
+            object_id (str): The ID of the object (dataset_id, project_id, external_connection_id, feature_group_id, etc.)
+            user_group_id (str): The ID of the user group
+            object_type (str): The type of object ('dataset', 'project', 'external_connection', 'feature_group')
+            permission (str): The permission to grant (default: 'ALL')"""
+        return self._call_api('updateUserGroupObjectPermission', 'POST', query_params={}, body={'objectId': object_id, 'userGroupId': user_group_id, 'objectType': object_type, 'permission': permission})
+
+    def remove_user_group_object_permission(self, object_id: str, user_group_id: str, object_type: str):
+        """Remove user group object permission for any object type.
+
+        Args:
+            object_id (str): The ID of the object (dataset_id, project_id, external_connection_id, feature_group_id, etc.)
+            user_group_id (str): The ID of the user group
+            object_type (str): The type of object ('dataset', 'project', 'external_connection', 'feature_group')"""
+        return self._call_api('removeUserGroupObjectPermission', 'POST', query_params={}, body={'objectId': object_id, 'userGroupId': user_group_id, 'objectType': object_type})
+
     def create_app_user_group(self, name: str) -> AppUserGroup:
         """Creates a new App User Group. This User Group is used to have permissions to access the external chatbots.
 
@@ -9076,7 +9167,7 @@ class ApiClient(ReadOnlyClient):
 
         Returns:
             LlmResponse: The response from the model, raw text and parsed components."""
-        return self._proxy_request('EvaluatePrompt', 'POST', query_params={}, body={'prompt': prompt, 'systemMessage': system_message, 'llmName': llm_name, 'maxTokens': max_tokens, 'temperature': temperature, 'messages': messages, 'responseType': response_type, 'jsonResponseSchema': json_response_schema, 'stopSequences': stop_sequences, 'topP': top_p}, parse_type=LlmResponse)
+        return self._proxy_request('evaluatePrompt', 'POST', query_params={}, body={'prompt': prompt, 'systemMessage': system_message, 'llmName': llm_name, 'maxTokens': max_tokens, 'temperature': temperature, 'messages': messages, 'responseType': response_type, 'jsonResponseSchema': json_response_schema, 'stopSequences': stop_sequences, 'topP': top_p}, parse_type=LlmResponse, is_sync=True)
 
     def render_feature_groups_for_llm(self, feature_group_ids: List, token_budget: int = None, include_definition: bool = True) -> List[LlmInput]:
         """Encode feature groups as language model inputs.

{abacusai-1.4.65 → abacusai-1.4.67}/abacusai/deployment_conversation.py

@@ -37,12 +37,13 @@ class DeploymentConversation(AbstractApiClass):
             daemonTaskId (str): The daemon task id associated with the deployment conversation.
             parentDeploymentConversationId (str): The parent deployment conversation id associated with the deployment conversation.
             introMessage (str): The intro message for the deployment conversation.
-            previewInfo (dict): App preview info
+            previewInfo (dict): App preview info.
+            latestContext (dict): dict containing a list of important segments for pagination in deepagent.
             history (DeploymentConversationEvent): The history of the deployment conversation.
             hostedArtifacts (HostedArtifact): Artifacts that have been deployed by this conversation.
     """
 
-    def __init__(self, client, deploymentConversationId=None, name=None, deploymentId=None, ownerUserId=None, ownerOrgId=None, createdAt=None, lastEventCreatedAt=None, hasHistory=None, externalSessionId=None, regenerateAttempt=None, externalApplicationId=None, unusedDocumentUploadIds=None, humanizeInstructions=None, conversationWarning=None, conversationType=None, metadata=None, llmDisplayName=None, llmBotIcon=None, searchSuggestions=None, chatllmTaskId=None, conversationStatus=None, computerStatus=None, filesystemStatus=None, totalEvents=None, contestNames=None, daemonTaskId=None, parentDeploymentConversationId=None, introMessage=None, previewInfo=None, history={}, hostedArtifacts={}):
+    def __init__(self, client, deploymentConversationId=None, name=None, deploymentId=None, ownerUserId=None, ownerOrgId=None, createdAt=None, lastEventCreatedAt=None, hasHistory=None, externalSessionId=None, regenerateAttempt=None, externalApplicationId=None, unusedDocumentUploadIds=None, humanizeInstructions=None, conversationWarning=None, conversationType=None, metadata=None, llmDisplayName=None, llmBotIcon=None, searchSuggestions=None, chatllmTaskId=None, conversationStatus=None, computerStatus=None, filesystemStatus=None, totalEvents=None, contestNames=None, daemonTaskId=None, parentDeploymentConversationId=None, introMessage=None, previewInfo=None, latestContext=None, history={}, hostedArtifacts={}):
         super().__init__(client, deploymentConversationId)
         self.deployment_conversation_id = deploymentConversationId
         self.name = name
@@ -73,6 +74,7 @@ class DeploymentConversation(AbstractApiClass):
         self.parent_deployment_conversation_id = parentDeploymentConversationId
         self.intro_message = introMessage
         self.preview_info = previewInfo
+        self.latest_context = latestContext
         self.history = client._build_class(
             DeploymentConversationEvent, history)
         self.hosted_artifacts = client._build_class(
@@ -80,8 +82,8 @@ class DeploymentConversation(AbstractApiClass):
         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'owner_user_id': repr(self.owner_user_id), f'owner_org_id': repr(self.owner_org_id), f'created_at': repr(self.created_at), f'last_event_created_at': repr(self.last_event_created_at), f'has_history': repr(self.has_history), f'external_session_id': repr(self.external_session_id), f'regenerate_attempt': repr(self.regenerate_attempt), f'external_application_id': repr(self.external_application_id), f'unused_document_upload_ids': repr(self.unused_document_upload_ids), f'humanize_instructions': repr(self.humanize_instructions), f'conversation_warning': repr(self.conversation_warning), f'conversation_type': repr(
-            self.conversation_type), f'metadata': repr(self.metadata), f'llm_display_name': repr(self.llm_display_name), f'llm_bot_icon': repr(self.llm_bot_icon), f'search_suggestions': repr(self.search_suggestions), f'chatllm_task_id': repr(self.chatllm_task_id), f'conversation_status': repr(self.conversation_status), f'computer_status': repr(self.computer_status), f'filesystem_status': repr(self.filesystem_status), f'total_events': repr(self.total_events), f'contest_names': repr(self.contest_names), f'daemon_task_id': repr(self.daemon_task_id), f'parent_deployment_conversation_id': repr(self.parent_deployment_conversation_id), f'intro_message': repr(self.intro_message), f'preview_info': repr(self.preview_info), f'history': repr(self.history), f'hosted_artifacts': repr(self.hosted_artifacts)}
+        repr_dict = {f'deployment_conversation_id': repr(self.deployment_conversation_id), f'name': repr(self.name), f'deployment_id': repr(self.deployment_id), f'owner_user_id': repr(self.owner_user_id), f'owner_org_id': repr(self.owner_org_id), f'created_at': repr(self.created_at), f'last_event_created_at': repr(self.last_event_created_at), f'has_history': repr(self.has_history), f'external_session_id': repr(self.external_session_id), f'regenerate_attempt': repr(self.regenerate_attempt), f'external_application_id': repr(self.external_application_id), f'unused_document_upload_ids': repr(self.unused_document_upload_ids), f'humanize_instructions': repr(self.humanize_instructions), f'conversation_warning': repr(self.conversation_warning), f'conversation_type': repr(self.conversation_type), f'metadata': repr(
+            self.metadata), f'llm_display_name': repr(self.llm_display_name), f'llm_bot_icon': repr(self.llm_bot_icon), f'search_suggestions': repr(self.search_suggestions), f'chatllm_task_id': repr(self.chatllm_task_id), f'conversation_status': repr(self.conversation_status), f'computer_status': repr(self.computer_status), f'filesystem_status': repr(self.filesystem_status), f'total_events': repr(self.total_events), f'contest_names': repr(self.contest_names), f'daemon_task_id': repr(self.daemon_task_id), f'parent_deployment_conversation_id': repr(self.parent_deployment_conversation_id), f'intro_message': repr(self.intro_message), f'preview_info': repr(self.preview_info), f'latest_context': repr(self.latest_context), f'history': repr(self.history), f'hosted_artifacts': repr(self.hosted_artifacts)}
         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])
@@ -95,7 +97,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, 'owner_user_id': self.owner_user_id, 'owner_org_id': self.owner_org_id, 'created_at': self.created_at, 'last_event_created_at': self.last_event_created_at, 'has_history': self.has_history, 'external_session_id': self.external_session_id, 'regenerate_attempt': self.regenerate_attempt, 'external_application_id': self.external_application_id, 'unused_document_upload_ids': self.unused_document_upload_ids, 'humanize_instructions': self.humanize_instructions, 'conversation_warning': self.conversation_warning, 'conversation_type': self.conversation_type, 'metadata': self.metadata,
-                'llm_display_name': self.llm_display_name, 'llm_bot_icon': self.llm_bot_icon, 'search_suggestions': self.search_suggestions, 'chatllm_task_id': self.chatllm_task_id, 'conversation_status': self.conversation_status, 'computer_status': self.computer_status, 'filesystem_status': self.filesystem_status, 'total_events': self.total_events, 'contest_names': self.contest_names, 'daemon_task_id': self.daemon_task_id, 'parent_deployment_conversation_id': self.parent_deployment_conversation_id, 'intro_message': self.intro_message, 'preview_info': self.preview_info, 'history': self._get_attribute_as_dict(self.history), 'hosted_artifacts': self._get_attribute_as_dict(self.hosted_artifacts)}
+                'llm_display_name': self.llm_display_name, 'llm_bot_icon': self.llm_bot_icon, 'search_suggestions': self.search_suggestions, 'chatllm_task_id': self.chatllm_task_id, 'conversation_status': self.conversation_status, 'computer_status': self.computer_status, 'filesystem_status': self.filesystem_status, 'total_events': self.total_events, 'contest_names': self.contest_names, 'daemon_task_id': self.daemon_task_id, 'parent_deployment_conversation_id': self.parent_deployment_conversation_id, 'intro_message': self.intro_message, 'preview_info': self.preview_info, 'latest_context': self.latest_context, 'history': self._get_attribute_as_dict(self.history), 'hosted_artifacts': self._get_attribute_as_dict(self.hosted_artifacts)}
         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, filter_intermediate_conversation_events: bool = True, get_unused_document_uploads: bool = False, start: int = None, limit: int = None, include_all_versions: bool = False):
@@ -116,6 +118,15 @@ class DeploymentConversation(AbstractApiClass):
         """
         return self.client.get_deployment_conversation(self.deployment_conversation_id, external_session_id, deployment_id, filter_intermediate_conversation_events, get_unused_document_uploads, start, limit, include_all_versions)
 
+    def get_file(self, file_path: str):
+        """
+        Gets a deployment conversation file.
+
+        Args:
+            file_path (str): The path of the file to get.
+        """
+        return self.client.get_deployment_conversation_file(self.deployment_conversation_id, file_path)
+
     def delete(self, deployment_id: str = None):
         """
         Delete a Deployment Conversation.

abacusai-1.4.67/abacusai/document_retriever_version_logs.py

@@ -0,0 +1,33 @@
+from .return_class import AbstractApiClass
+
+
+class DocumentRetrieverVersionLogs(AbstractApiClass):
+    """
+        Logs from document retriever version.
+
+        Args:
+            client (ApiClient): An authenticated API Client instance
+            logs (list[str]): List of logs from document retriever version.
+    """
+
+    def __init__(self, client, logs=None):
+        super().__init__(client, None)
+        self.logs = logs
+        self.deprecated_keys = {}
+
+    def __repr__(self):
+        repr_dict = {f'logs': repr(self.logs)}
+        class_name = "DocumentRetrieverVersionLogs"
+        repr_str = ',\n  '.join([f'{key}={value}' for key, value in repr_dict.items(
+        ) if getattr(self, key, None) is not None and key not in self.deprecated_keys])
+        return f"{class_name}({repr_str})"
+
+    def to_dict(self):
+        """
+        Get a dict representation of the parameters in this class
+
+        Returns:
+            dict: The dict value representation of the class parameters
+        """
+        resp = {'logs': self.logs}
+        return {key: value for key, value in resp.items() if value is not None and key not in self.deprecated_keys}

{abacusai-1.4.65 → abacusai-1.4.67}/abacusai/hosted_app_container.py

@@ -23,6 +23,7 @@ class HostedAppContainer(AbstractApiClass):
             lifecycle (str): Container lifecycle status (PENDING/DEPLOYING/ACTIVE/FAILED/STOPPED/DELETING)
             status (str): Container status (RUNNING/STOPPED/DEPLOYING/FAILED)
             deployedStatus (str): Deployment status (PENDING/ACTIVE/STOPPED/NOT_DEPLOYED)
+            accessLevel (str): Access Level (PUBLIC/PRIVATE/DEDICATED)
             hostname (str): Hostname of the deployed app
             llmArtifactId (id): The ID of the LLM artifact
             artifactType (str): The type of the artifact
@@ -33,7 +34,7 @@ class HostedAppContainer(AbstractApiClass):
             parentConversationId (id): The ID of the parent conversation
     """
 
-    def __init__(self, client, hostedAppContainerId=None, hostedAppId=None, deploymentConversationId=None, hostedAppVersion=None, name=None, createdAt=None, updatedAt=None, containerImage=None, route=None, appConfig=None, isDev=None, isDeployable=None, isPreviewAvailable=None, lifecycle=None, status=None, deployedStatus=None, hostname=None, llmArtifactId=None, artifactType=None, deployedLlmArtifactId=None, hasDatabase=None, hasStorage=None, webAppProjectId=None, parentConversationId=None):
+    def __init__(self, client, hostedAppContainerId=None, hostedAppId=None, deploymentConversationId=None, hostedAppVersion=None, name=None, createdAt=None, updatedAt=None, containerImage=None, route=None, appConfig=None, isDev=None, isDeployable=None, isPreviewAvailable=None, lifecycle=None, status=None, deployedStatus=None, accessLevel=None, hostname=None, llmArtifactId=None, artifactType=None, deployedLlmArtifactId=None, hasDatabase=None, hasStorage=None, webAppProjectId=None, parentConversationId=None):
         super().__init__(client, hostedAppContainerId)
         self.hosted_app_container_id = hostedAppContainerId
         self.hosted_app_id = hostedAppId
@@ -51,6 +52,7 @@ class HostedAppContainer(AbstractApiClass):
         self.lifecycle = lifecycle
         self.status = status
         self.deployed_status = deployedStatus
+        self.access_level = accessLevel
         self.hostname = hostname
         self.llm_artifact_id = llmArtifactId
         self.artifact_type = artifactType
@@ -63,7 +65,7 @@ class HostedAppContainer(AbstractApiClass):
 
     def __repr__(self):
         repr_dict = {f'hosted_app_container_id': repr(self.hosted_app_container_id), f'hosted_app_id': repr(self.hosted_app_id), f'deployment_conversation_id': repr(self.deployment_conversation_id), f'hosted_app_version': repr(self.hosted_app_version), f'name': repr(self.name), f'created_at': repr(self.created_at), f'updated_at': repr(self.updated_at), f'container_image': repr(self.container_image), f'route': repr(self.route), f'app_config': repr(self.app_config), f'is_dev': repr(self.is_dev), f'is_deployable': repr(self.is_deployable), f'is_preview_available': repr(
-            self.is_preview_available), f'lifecycle': repr(self.lifecycle), f'status': repr(self.status), f'deployed_status': repr(self.deployed_status), f'hostname': repr(self.hostname), f'llm_artifact_id': repr(self.llm_artifact_id), f'artifact_type': repr(self.artifact_type), f'deployed_llm_artifact_id': repr(self.deployed_llm_artifact_id), f'has_database': repr(self.has_database), f'has_storage': repr(self.has_storage), f'web_app_project_id': repr(self.web_app_project_id), f'parent_conversation_id': repr(self.parent_conversation_id)}
+            self.is_preview_available), f'lifecycle': repr(self.lifecycle), f'status': repr(self.status), f'deployed_status': repr(self.deployed_status), f'access_level': repr(self.access_level), f'hostname': repr(self.hostname), f'llm_artifact_id': repr(self.llm_artifact_id), f'artifact_type': repr(self.artifact_type), f'deployed_llm_artifact_id': repr(self.deployed_llm_artifact_id), f'has_database': repr(self.has_database), f'has_storage': repr(self.has_storage), f'web_app_project_id': repr(self.web_app_project_id), f'parent_conversation_id': repr(self.parent_conversation_id)}
         class_name = "HostedAppContainer"
         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])
@@ -76,6 +78,6 @@ class HostedAppContainer(AbstractApiClass):
         Returns:
             dict: The dict value representation of the class parameters
         """
-        resp = {'hosted_app_container_id': self.hosted_app_container_id, 'hosted_app_id': self.hosted_app_id, 'deployment_conversation_id': self.deployment_conversation_id, 'hosted_app_version': self.hosted_app_version, 'name': self.name, 'created_at': self.created_at, 'updated_at': self.updated_at, 'container_image': self.container_image, 'route': self.route, 'app_config': self.app_config, 'is_dev': self.is_dev, 'is_deployable': self.is_deployable,
-                'is_preview_available': self.is_preview_available, 'lifecycle': self.lifecycle, 'status': self.status, 'deployed_status': self.deployed_status, 'hostname': self.hostname, 'llm_artifact_id': self.llm_artifact_id, 'artifact_type': self.artifact_type, 'deployed_llm_artifact_id': self.deployed_llm_artifact_id, 'has_database': self.has_database, 'has_storage': self.has_storage, 'web_app_project_id': self.web_app_project_id, 'parent_conversation_id': self.parent_conversation_id}
+        resp = {'hosted_app_container_id': self.hosted_app_container_id, 'hosted_app_id': self.hosted_app_id, 'deployment_conversation_id': self.deployment_conversation_id, 'hosted_app_version': self.hosted_app_version, 'name': self.name, 'created_at': self.created_at, 'updated_at': self.updated_at, 'container_image': self.container_image, 'route': self.route, 'app_config': self.app_config, 'is_dev': self.is_dev, 'is_deployable': self.is_deployable, 'is_preview_available': self.is_preview_available,
+                'lifecycle': self.lifecycle, 'status': self.status, 'deployed_status': self.deployed_status, 'access_level': self.access_level, 'hostname': self.hostname, 'llm_artifact_id': self.llm_artifact_id, 'artifact_type': self.artifact_type, 'deployed_llm_artifact_id': self.deployed_llm_artifact_id, 'has_database': self.has_database, 'has_storage': self.has_storage, 'web_app_project_id': self.web_app_project_id, 'parent_conversation_id': self.parent_conversation_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.65 → abacusai-1.4.67}/abacusai/hostname_info.py

@@ -10,20 +10,22 @@ class HostnameInfo(AbstractApiClass):
             isRootDomain (bool): Whether the hostname is a root domain
             registrar (str): The registrar of the domain
             suggestedFlow (str): Suggested flow for the domain
+            isCnameSupported (bool): Whether CNAME is supported for the domain
             isAutomaticFlowAvailable (bool): Whether entri is supported for the domain
     """
 
-    def __init__(self, client, isRootDomain=None, registrar=None, suggestedFlow=None, isAutomaticFlowAvailable=None):
+    def __init__(self, client, isRootDomain=None, registrar=None, suggestedFlow=None, isCnameSupported=None, isAutomaticFlowAvailable=None):
         super().__init__(client, None)
         self.is_root_domain = isRootDomain
         self.registrar = registrar
         self.suggested_flow = suggestedFlow
+        self.is_cname_supported = isCnameSupported
         self.is_automatic_flow_available = isAutomaticFlowAvailable
         self.deprecated_keys = {}
 
     def __repr__(self):
         repr_dict = {f'is_root_domain': repr(self.is_root_domain), f'registrar': repr(self.registrar), f'suggested_flow': repr(
-            self.suggested_flow), f'is_automatic_flow_available': repr(self.is_automatic_flow_available)}
+            self.suggested_flow), f'is_cname_supported': repr(self.is_cname_supported), f'is_automatic_flow_available': repr(self.is_automatic_flow_available)}
         class_name = "HostnameInfo"
         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])
@@ -36,6 +38,6 @@ class HostnameInfo(AbstractApiClass):
         Returns:
             dict: The dict value representation of the class parameters
         """
-        resp = {'is_root_domain': self.is_root_domain, 'registrar': self.registrar,
-                'suggested_flow': self.suggested_flow, 'is_automatic_flow_available': self.is_automatic_flow_available}
+        resp = {'is_root_domain': self.is_root_domain, 'registrar': self.registrar, 'suggested_flow': self.suggested_flow,
+                'is_cname_supported': self.is_cname_supported, 'is_automatic_flow_available': self.is_automatic_flow_available}
         return {key: value for key, value in resp.items() if value is not None and key not in self.deprecated_keys}

{abacusai-1.4.65 → abacusai-1.4.67}/abacusai/image_gen_settings.py

@@ -9,18 +9,20 @@ class ImageGenSettings(AbstractApiClass):
         Args:
             client (ApiClient): An authenticated API Client instance
             settings (dict): The settings for each model.
+            warnings (dict): The warnings for each model.
             model (ImageGenModel): Dropdown for models available for image generation.
     """
 
-    def __init__(self, client, settings=None, model={}):
+    def __init__(self, client, settings=None, warnings=None, model={}):
         super().__init__(client, None)
         self.settings = settings
+        self.warnings = warnings
         self.model = client._build_class(ImageGenModel, model)
         self.deprecated_keys = {}
 
     def __repr__(self):
-        repr_dict = {f'settings': repr(
-            self.settings), f'model': repr(self.model)}
+        repr_dict = {f'settings': repr(self.settings), f'warnings': repr(
+            self.warnings), f'model': repr(self.model)}
         class_name = "ImageGenSettings"
         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])
@@ -33,6 +35,6 @@ class ImageGenSettings(AbstractApiClass):
         Returns:
             dict: The dict value representation of the class parameters
         """
-        resp = {'settings': self.settings,
+        resp = {'settings': self.settings, 'warnings': self.warnings,
                 'model': self._get_attribute_as_dict(self.model)}
         return {key: value for key, value in resp.items() if value is not None and key not in self.deprecated_keys}

{abacusai-1.4.65 → abacusai-1.4.67}/abacusai/prediction_client.py

@@ -425,6 +425,14 @@ class PredictionClient(BaseApiClient):
             deployment_id, deployment_token) if deployment_token else None
         return self._call_api('getConversationResponseWithBinaryData', 'POST', query_params={'deploymentId': deployment_id, 'deploymentToken': deployment_token}, data={'message': json.dumps(message) if (message is not None and not isinstance(message, str)) else message, 'deploymentConversationId': json.dumps(deployment_conversation_id) if (deployment_conversation_id is not None and not isinstance(deployment_conversation_id, str)) else deployment_conversation_id, 'externalSessionId': json.dumps(external_session_id) if (external_session_id is not None and not isinstance(external_session_id, str)) else external_session_id, 'llmName': json.dumps(llm_name) if (llm_name is not None and not isinstance(llm_name, str)) else llm_name, 'numCompletionTokens': json.dumps(num_completion_tokens) if (num_completion_tokens is not None and not isinstance(num_completion_tokens, str)) else num_completion_tokens, 'systemMessage': json.dumps(system_message) if (system_message is not None and not isinstance(system_message, str)) else system_message, 'temperature': json.dumps(temperature) if (temperature is not None and not isinstance(temperature, str)) else temperature, 'filterKeyValues': json.dumps(filter_key_values) if (filter_key_values is not None and not isinstance(filter_key_values, str)) else filter_key_values, 'searchScoreCutoff': json.dumps(search_score_cutoff) if (search_score_cutoff is not None and not isinstance(search_score_cutoff, str)) else search_score_cutoff, 'chatConfig': json.dumps(chat_config) if (chat_config is not None and not isinstance(chat_config, str)) else chat_config}, files=attachments, server_override=prediction_url)
 
+    def get_deep_agent_response(self, message: str, deployment_conversation_id: str = None) -> Dict:
+        """Return a DeepAgent response with generated files if any, based on the input message.
+
+        Args:
+            message (str): The user's message/task for DeepAgent to complete
+            deployment_conversation_id (str): The unique identifier of a deployment conversation to continue. If not specified, a new one will be created."""
+        return self._proxy_request('getDeepAgentResponse', 'POST', query_params={}, body={'message': message, 'deploymentConversationId': deployment_conversation_id}, is_sync=True)
+
     def get_search_results(self, deployment_token: str, deployment_id: str, query_data: dict, num: int = 15) -> Dict:
         """Return the most relevant search results to the search query from the uploaded documents.