abacusai 1.4.55__tar.gz → 1.4.56__tar.gz

{abacusai-1.4.55/abacusai.egg-info → abacusai-1.4.56}/PKG-INFO

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

{abacusai-1.4.55 → abacusai-1.4.56}/abacusai/__init__.py

@@ -140,6 +140,7 @@ from .hosted_artifact import HostedArtifact
 from .hosted_database import HostedDatabase
 from .hosted_database_snapshot import HostedDatabaseSnapshot
 from .hosted_model_token import HostedModelToken
+from .hostname_info import HostnameInfo
 from .hume_voice import HumeVoice
 from .image_gen_model import ImageGenModel
 from .image_gen_model_options import ImageGenModelOptions
@@ -151,6 +152,7 @@ from .item_statistics import ItemStatistics
 from .lip_sync_gen_settings import LipSyncGenSettings
 from .llm_app import LlmApp
 from .llm_artifact import LlmArtifact
+from .llm_artifact_conversation import LlmArtifactConversation
 from .llm_code_block import LlmCodeBlock
 from .llm_execution_preview import LlmExecutionPreview
 from .llm_execution_result import LlmExecutionResult
@@ -160,6 +162,8 @@ from .llm_parameters import LlmParameters
 from .llm_response import LlmResponse
 from .mcp_config import McpConfig
 from .mcp_server import McpServer
+from .mcp_server_connection import McpServerConnection
+from .mcp_server_query_result import McpServerQueryResult
 from .memory_options import MemoryOptions
 from .messaging_connector_response import MessagingConnectorResponse
 from .model import Model
@@ -273,4 +277,4 @@ from .workflow_graph_node_details import WorkflowGraphNodeDetails
 from .workflow_node_template import WorkflowNodeTemplate
 
 
-__version__ = "1.4.55"
+__version__ = "1.4.56"

{abacusai-1.4.55 → abacusai-1.4.56}/abacusai/api_class/dataset.py

@@ -64,7 +64,7 @@ class DocumentProcessingConfig(ApiClass):
         if self.document_type is not None:
             if DocumentType.is_ocr_forced(self.document_type):
                 self.highlight_relevant_text = True
-            else:
+            elif self.highlight_relevant_text is None:
                 self.highlight_relevant_text = False
         if self.highlight_relevant_text is not None:
             self.extract_bounding_boxes = self.highlight_relevant_text  # Highlight_relevant text acts as a wrapper over extract_bounding_boxes

{abacusai-1.4.55 → abacusai-1.4.56}/abacusai/api_class/enums.py

@@ -418,6 +418,7 @@ class ApplicationConnectorType(ApiEnum):
     OAUTH = 'OAUTH'
     SALESFORCE = 'SALESFORCE'
     TWITTER = 'TWITTER'
+    MCP = 'MCP'
 
     @classmethod
     def user_connectors(cls):
@@ -430,7 +431,8 @@ class ApplicationConnectorType(ApiEnum):
             cls.ONEDRIVE,
             cls.SALESFORCE,
             cls.TWITTER,
-            cls.SHAREPOINT
+            cls.SHAREPOINT,
+            cls.CONFLUENCE
         ]
 
     @classmethod

{abacusai-1.4.55 → abacusai-1.4.56}/abacusai/api_class/model.py

@@ -501,7 +501,7 @@ class ChatLLMTrainingConfig(TrainingConfig):
         json_response_schema (str): Specifies the JSON schema that the model should adhere to if `response_format` is set to "JSON". This should be a json-formatted string where each field of the expected schema is mapped to a dictionary containing the fields 'type', 'required' and 'description'. For example - '{"sample_field": {"type": "integer", "required": true, "description": "Sample Field"}}'
         mask_pii (bool): Mask PII in the prompts and uploaded documents before sending it to the LLM.
         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_server_configs (str): JSON string of MCP servers configs to use in the ChatLLM model. This should not be used with document_retrievers.
+        mcp_servers (List[str]): List of names of MCP servers to use in the ChatLLM. This should not be used with document_retrievers.
     """
     document_retrievers: List[str] = dataclasses.field(default=None)
     num_completion_tokens: int = dataclasses.field(default=None)
@@ -540,7 +540,7 @@ class ChatLLMTrainingConfig(TrainingConfig):
     json_response_schema: str = dataclasses.field(default=None)
     mask_pii: bool = dataclasses.field(default=None)
     builtin_tools: List[SystemConnectorTool] = dataclasses.field(default=None)
-    mcp_server_configs: str = dataclasses.field(default=None)
+    mcp_servers: List[str] = dataclasses.field(default=None)
 
     def __post_init__(self):
         self.problem_type = enums.ProblemType.CHAT_LLM

{abacusai-1.4.55 → abacusai-1.4.56}/abacusai/client.py

@@ -126,6 +126,7 @@ from .llm_execution_result import LlmExecutionResult
 from .llm_generated_code import LlmGeneratedCode
 from .llm_input import LlmInput
 from .llm_response import LlmResponse
+from .mcp_server_query_result import McpServerQueryResult
 from .model import Model
 from .model_artifacts_export import ModelArtifactsExport
 from .model_metrics import ModelMetrics
@@ -662,7 +663,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.55'
+    client_version = '1.4.56'
 
     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
@@ -1611,6 +1612,16 @@ class ReadOnlyClient(BaseApiClient):
             UnifiedConnector: The application connector with the authentication details."""
         return self._call_api('getUserConnectorAuth', 'GET', query_params={'service': service, 'scopes': scopes}, parse_type=UnifiedConnector)
 
+    def get_user_mcp_connector_auth(self, mcp_server_name: str = None) -> ApplicationConnector:
+        """Get the auth for a MCP connector
+
+        Args:
+            mcp_server_name (str): The name of the MCP server
+
+        Returns:
+            ApplicationConnector: The MCP connector"""
+        return self._call_api('getUserMcpConnectorAuth', 'GET', query_params={'mcpServerName': mcp_server_name}, parse_type=ApplicationConnector)
+
     def list_streaming_connectors(self) -> List[StreamingConnector]:
         """Retrieves a list of all streaming connectors along with their corresponding attributes.
 
@@ -8970,7 +8981,7 @@ Creates a new feature group defined as the union of other feature group versions
             external_application_id (str): The ID of the External Application."""
         return self._call_api('deleteExternalApplication', 'DELETE', query_params={'externalApplicationId': external_application_id})
 
-    def create_agent(self, project_id: str, function_source_code: str = None, agent_function_name: str = None, name: str = None, memory: int = None, package_requirements: list = [], description: str = None, enable_binary_input: bool = False, evaluation_feature_group_id: str = None, agent_input_schema: dict = None, agent_output_schema: dict = None, workflow_graph: Union[dict, WorkflowGraph] = None, agent_interface: Union[AgentInterface, str] = AgentInterface.DEFAULT, included_modules: List = None, org_level_connectors: List = None, user_level_connectors: Dict = None, initialize_function_name: str = None, initialize_function_code: str = None, agent_mcp_config: dict = None) -> Agent:
+    def create_agent(self, project_id: str, function_source_code: str = None, agent_function_name: str = None, name: str = None, memory: int = None, package_requirements: list = [], description: str = None, enable_binary_input: bool = False, evaluation_feature_group_id: str = None, agent_input_schema: dict = None, agent_output_schema: dict = None, workflow_graph: Union[dict, WorkflowGraph] = None, agent_interface: Union[AgentInterface, str] = AgentInterface.DEFAULT, included_modules: List = None, org_level_connectors: List = None, user_level_connectors: Dict = None, initialize_function_name: str = None, initialize_function_code: str = None) -> Agent:
         """Creates a new AI agent using the given agent workflow graph definition.
 
         Args:
@@ -8987,13 +8998,12 @@ Creates a new feature group defined as the union of other feature group versions
             user_level_connectors (Dict): A dictionary mapping ApplicationConnectorType keys to lists of OAuth scopes. Each key represents a specific user level application connector, while the value is a list of scopes that define the permissions granted to the application.
             initialize_function_name (str): The name of the function to be used for initialization.
             initialize_function_code (str): The function code to be used for initialization.
-            agent_mcp_config (dict): The MCP config for the agent.
 
         Returns:
             Agent: The new agent."""
-        return self._call_api('createAgent', 'POST', query_params={}, body={'projectId': project_id, 'functionSourceCode': function_source_code, 'agentFunctionName': agent_function_name, 'name': name, 'memory': memory, 'packageRequirements': package_requirements, 'description': description, 'enableBinaryInput': enable_binary_input, 'evaluationFeatureGroupId': evaluation_feature_group_id, 'agentInputSchema': agent_input_schema, 'agentOutputSchema': agent_output_schema, 'workflowGraph': workflow_graph, 'agentInterface': agent_interface, 'includedModules': included_modules, 'orgLevelConnectors': org_level_connectors, 'userLevelConnectors': user_level_connectors, 'initializeFunctionName': initialize_function_name, 'initializeFunctionCode': initialize_function_code, 'agentMcpConfig': agent_mcp_config}, parse_type=Agent)
+        return self._call_api('createAgent', 'POST', query_params={}, body={'projectId': project_id, 'functionSourceCode': function_source_code, 'agentFunctionName': agent_function_name, 'name': name, 'memory': memory, 'packageRequirements': package_requirements, 'description': description, 'enableBinaryInput': enable_binary_input, 'evaluationFeatureGroupId': evaluation_feature_group_id, 'agentInputSchema': agent_input_schema, 'agentOutputSchema': agent_output_schema, 'workflowGraph': workflow_graph, 'agentInterface': agent_interface, 'includedModules': included_modules, 'orgLevelConnectors': org_level_connectors, 'userLevelConnectors': user_level_connectors, 'initializeFunctionName': initialize_function_name, 'initializeFunctionCode': initialize_function_code}, parse_type=Agent)
 
-    def update_agent(self, model_id: str, function_source_code: str = None, agent_function_name: str = None, memory: int = None, package_requirements: list = None, description: str = None, enable_binary_input: bool = None, agent_input_schema: dict = None, agent_output_schema: dict = None, workflow_graph: Union[dict, WorkflowGraph] = None, agent_interface: Union[AgentInterface, str] = None, included_modules: List = None, org_level_connectors: List = None, user_level_connectors: Dict = None, initialize_function_name: str = None, initialize_function_code: str = None, agent_mcp_config: dict = None) -> Agent:
+    def update_agent(self, model_id: str, function_source_code: str = None, agent_function_name: str = None, memory: int = None, package_requirements: list = None, description: str = None, enable_binary_input: bool = None, agent_input_schema: dict = None, agent_output_schema: dict = None, workflow_graph: Union[dict, WorkflowGraph] = None, agent_interface: Union[AgentInterface, str] = None, included_modules: List = None, org_level_connectors: List = None, user_level_connectors: Dict = None, initialize_function_name: str = None, initialize_function_code: str = None) -> Agent:
         """Updates an existing AI Agent. A new version of the agent will be created and published.
 
         Args:
@@ -9008,11 +9018,10 @@ Creates a new feature group defined as the union of other feature group versions
             user_level_connectors (Dict): A dictionary mapping ApplicationConnectorType keys to lists of OAuth scopes. Each key represents a specific user level application connector, while the value is a list of scopes that define the permissions granted to the application.
             initialize_function_name (str): The name of the function to be used for initialization.
             initialize_function_code (str): The function code to be used for initialization.
-            agent_mcp_config (dict): The MCP config for the agent.
 
         Returns:
             Agent: The updated agent."""
-        return self._call_api('updateAgent', 'POST', query_params={}, body={'modelId': model_id, 'functionSourceCode': function_source_code, 'agentFunctionName': agent_function_name, 'memory': memory, 'packageRequirements': package_requirements, 'description': description, 'enableBinaryInput': enable_binary_input, 'agentInputSchema': agent_input_schema, 'agentOutputSchema': agent_output_schema, 'workflowGraph': workflow_graph, 'agentInterface': agent_interface, 'includedModules': included_modules, 'orgLevelConnectors': org_level_connectors, 'userLevelConnectors': user_level_connectors, 'initializeFunctionName': initialize_function_name, 'initializeFunctionCode': initialize_function_code, 'agentMcpConfig': agent_mcp_config}, parse_type=Agent)
+        return self._call_api('updateAgent', 'POST', query_params={}, body={'modelId': model_id, 'functionSourceCode': function_source_code, 'agentFunctionName': agent_function_name, 'memory': memory, 'packageRequirements': package_requirements, 'description': description, 'enableBinaryInput': enable_binary_input, 'agentInputSchema': agent_input_schema, 'agentOutputSchema': agent_output_schema, 'workflowGraph': workflow_graph, 'agentInterface': agent_interface, 'includedModules': included_modules, 'orgLevelConnectors': org_level_connectors, 'userLevelConnectors': user_level_connectors, 'initializeFunctionName': initialize_function_name, 'initializeFunctionCode': initialize_function_code}, parse_type=Agent)
 
     def generate_agent_code(self, project_id: str, prompt: str, fast_mode: bool = None) -> list:
         """Generates the code for defining an AI Agent
@@ -9257,3 +9266,17 @@ 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={}, data={'docIds': doc_ids, 'query': query, 'documentRetrieverConfig': json.dumps(document_retriever_config.to_dict()) if hasattr(document_retriever_config, 'to_dict') else 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, 'documentProcessingConfig': json.dumps(document_processing_config.to_dict()) if hasattr(document_processing_config, 'to_dict') else json.dumps(document_processing_config)}, files=blobs, parse_type=DocumentRetrieverLookupResult)
+
+    def query_mcp_server(self, task: str, server_name: str) -> McpServerQueryResult:
+        """Query a Remote MCP server. For a given task, it runs the required tools of the MCP
+
+        server with appropriate arguments and returns the output.
+
+
+        Args:
+            task (str): a comprehensive description of the task to perform using the MCP server tools.
+            server_name (str): The name of the MCP server.
+
+        Returns:
+            McpServerQueryResult: The execution logs as well as the final output of the task execution."""
+        return self._proxy_request('QueryMcpServer', 'POST', query_params={}, body={'task': task, 'serverName': server_name}, parse_type=McpServerQueryResult)

{abacusai-1.4.55 → abacusai-1.4.56}/abacusai/deployment_conversation.py

@@ -33,11 +33,13 @@ class DeploymentConversation(AbstractApiClass):
             totalEvents (int): The total number of events in the deployment conversation.
             contestNames (list[str]): Names of contests that this deployment is a part of.
             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.
             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, 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, totalEvents=None, contestNames=None, daemonTaskId=None, history={}, hostedArtifacts={}):
+    def __init__(self, client, deploymentConversationId=None, name=None, deploymentId=None, ownerUserId=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, totalEvents=None, contestNames=None, daemonTaskId=None, parentDeploymentConversationId=None, introMessage=None, history={}, hostedArtifacts={}):
         super().__init__(client, deploymentConversationId)
         self.deployment_conversation_id = deploymentConversationId
         self.name = name
@@ -63,6 +65,8 @@ class DeploymentConversation(AbstractApiClass):
         self.total_events = totalEvents
         self.contest_names = contestNames
         self.daemon_task_id = daemonTaskId
+        self.parent_deployment_conversation_id = parentDeploymentConversationId
+        self.intro_message = introMessage
         self.history = client._build_class(
             DeploymentConversationEvent, history)
         self.hosted_artifacts = client._build_class(
@@ -70,8 +74,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'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'total_events': repr(self.total_events), f'contest_names': repr(self.contest_names), f'daemon_task_id': repr(self.daemon_task_id), 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'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'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'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])
@@ -84,8 +88,8 @@ class DeploymentConversation(AbstractApiClass):
         Returns:
             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, '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, 'total_events': self.total_events, 'contest_names': self.contest_names, 'daemon_task_id': self.daemon_task_id, 'history': self._get_attribute_as_dict(self.history), 'hosted_artifacts': self._get_attribute_as_dict(self.hosted_artifacts)}
+        resp = {'deployment_conversation_id': self.deployment_conversation_id, 'name': self.name, 'deployment_id': self.deployment_id, 'owner_user_id': self.owner_user_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, '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, '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):

{abacusai-1.4.55 → abacusai-1.4.56}/abacusai/deployment_conversation_event.py

@@ -40,9 +40,10 @@ class DeploymentConversationEvent(AbstractApiClass):
             toolUseRequest (dict): The tool use request for the message.
             verificationSummary (str): The summary of the verification process for the message.
             attachedUserFileNames (list): The list of files attached by the user on the message.
+            forkedDeploymentConversationIds (list): The list of deployment conversation ids that were forked from this conversation.
     """
 
-    def __init__(self, client, role=None, text=None, timestamp=None, messageIndex=None, regenerateAttempt=None, modelVersion=None, searchResults=None, isUseful=None, feedback=None, feedbackType=None, docInfos=None, keywordArguments=None, inputParams=None, attachments=None, responseVersion=None, agentWorkflowNodeId=None, nextAgentWorkflowNodeId=None, chatType=None, agentResponse=None, error=None, segments=None, streamedData=None, streamedSectionData=None, highlights=None, llmDisplayName=None, llmBotIcon=None, formResponse=None, routedLlm=None, computePointsUsed=None, computerFiles=None, toolUseRequest=None, verificationSummary=None, attachedUserFileNames=None):
+    def __init__(self, client, role=None, text=None, timestamp=None, messageIndex=None, regenerateAttempt=None, modelVersion=None, searchResults=None, isUseful=None, feedback=None, feedbackType=None, docInfos=None, keywordArguments=None, inputParams=None, attachments=None, responseVersion=None, agentWorkflowNodeId=None, nextAgentWorkflowNodeId=None, chatType=None, agentResponse=None, error=None, segments=None, streamedData=None, streamedSectionData=None, highlights=None, llmDisplayName=None, llmBotIcon=None, formResponse=None, routedLlm=None, computePointsUsed=None, computerFiles=None, toolUseRequest=None, verificationSummary=None, attachedUserFileNames=None, forkedDeploymentConversationIds=None):
         super().__init__(client, None)
         self.role = role
         self.text = text
@@ -77,11 +78,12 @@ class DeploymentConversationEvent(AbstractApiClass):
         self.tool_use_request = toolUseRequest
         self.verification_summary = verificationSummary
         self.attached_user_file_names = attachedUserFileNames
+        self.forked_deployment_conversation_ids = forkedDeploymentConversationIds
         self.deprecated_keys = {}
 
     def __repr__(self):
-        repr_dict = {f'role': repr(self.role), f'text': repr(self.text), f'timestamp': repr(self.timestamp), f'message_index': repr(self.message_index), f'regenerate_attempt': repr(self.regenerate_attempt), f'model_version': repr(self.model_version), f'search_results': repr(self.search_results), f'is_useful': repr(self.is_useful), f'feedback': repr(self.feedback), f'feedback_type': repr(self.feedback_type), f'doc_infos': repr(self.doc_infos), f'keyword_arguments': repr(self.keyword_arguments), f'input_params': repr(self.input_params), f'attachments': repr(self.attachments), f'response_version': repr(self.response_version), f'agent_workflow_node_id': repr(self.agent_workflow_node_id), f'next_agent_workflow_node_id': repr(
-            self.next_agent_workflow_node_id), f'chat_type': repr(self.chat_type), f'agent_response': repr(self.agent_response), f'error': repr(self.error), f'segments': repr(self.segments), f'streamed_data': repr(self.streamed_data), f'streamed_section_data': repr(self.streamed_section_data), f'highlights': repr(self.highlights), f'llm_display_name': repr(self.llm_display_name), f'llm_bot_icon': repr(self.llm_bot_icon), f'form_response': repr(self.form_response), f'routed_llm': repr(self.routed_llm), f'compute_points_used': repr(self.compute_points_used), f'computer_files': repr(self.computer_files), f'tool_use_request': repr(self.tool_use_request), f'verification_summary': repr(self.verification_summary), f'attached_user_file_names': repr(self.attached_user_file_names)}
+        repr_dict = {f'role': repr(self.role), f'text': repr(self.text), f'timestamp': repr(self.timestamp), f'message_index': repr(self.message_index), f'regenerate_attempt': repr(self.regenerate_attempt), f'model_version': repr(self.model_version), f'search_results': repr(self.search_results), f'is_useful': repr(self.is_useful), f'feedback': repr(self.feedback), f'feedback_type': repr(self.feedback_type), f'doc_infos': repr(self.doc_infos), f'keyword_arguments': repr(self.keyword_arguments), f'input_params': repr(self.input_params), f'attachments': repr(self.attachments), f'response_version': repr(self.response_version), f'agent_workflow_node_id': repr(self.agent_workflow_node_id), f'next_agent_workflow_node_id': repr(self.next_agent_workflow_node_id), f'chat_type': repr(
+            self.chat_type), f'agent_response': repr(self.agent_response), f'error': repr(self.error), f'segments': repr(self.segments), f'streamed_data': repr(self.streamed_data), f'streamed_section_data': repr(self.streamed_section_data), f'highlights': repr(self.highlights), f'llm_display_name': repr(self.llm_display_name), f'llm_bot_icon': repr(self.llm_bot_icon), f'form_response': repr(self.form_response), f'routed_llm': repr(self.routed_llm), f'compute_points_used': repr(self.compute_points_used), f'computer_files': repr(self.computer_files), f'tool_use_request': repr(self.tool_use_request), f'verification_summary': repr(self.verification_summary), f'attached_user_file_names': repr(self.attached_user_file_names), f'forked_deployment_conversation_ids': repr(self.forked_deployment_conversation_ids)}
         class_name = "DeploymentConversationEvent"
         repr_str = ',\n  '.join([f'{key}={value}' for key, value in repr_dict.items(
         ) if getattr(self, key, None) is not None and key not in self.deprecated_keys])
@@ -94,6 +96,6 @@ class DeploymentConversationEvent(AbstractApiClass):
         Returns:
             dict: The dict value representation of the class parameters
         """
-        resp = {'role': self.role, 'text': self.text, 'timestamp': self.timestamp, 'message_index': self.message_index, 'regenerate_attempt': self.regenerate_attempt, 'model_version': self.model_version, 'search_results': self.search_results, 'is_useful': self.is_useful, 'feedback': self.feedback, 'feedback_type': self.feedback_type, 'doc_infos': self.doc_infos, 'keyword_arguments': self.keyword_arguments, 'input_params': self.input_params, 'attachments': self.attachments, 'response_version': self.response_version, 'agent_workflow_node_id': self.agent_workflow_node_id, 'next_agent_workflow_node_id': self.next_agent_workflow_node_id,
-                'chat_type': self.chat_type, 'agent_response': self.agent_response, 'error': self.error, 'segments': self.segments, 'streamed_data': self.streamed_data, 'streamed_section_data': self.streamed_section_data, 'highlights': self.highlights, 'llm_display_name': self.llm_display_name, 'llm_bot_icon': self.llm_bot_icon, 'form_response': self.form_response, 'routed_llm': self.routed_llm, 'compute_points_used': self.compute_points_used, 'computer_files': self.computer_files, 'tool_use_request': self.tool_use_request, 'verification_summary': self.verification_summary, 'attached_user_file_names': self.attached_user_file_names}
+        resp = {'role': self.role, 'text': self.text, 'timestamp': self.timestamp, 'message_index': self.message_index, 'regenerate_attempt': self.regenerate_attempt, 'model_version': self.model_version, 'search_results': self.search_results, 'is_useful': self.is_useful, 'feedback': self.feedback, 'feedback_type': self.feedback_type, 'doc_infos': self.doc_infos, 'keyword_arguments': self.keyword_arguments, 'input_params': self.input_params, 'attachments': self.attachments, 'response_version': self.response_version, 'agent_workflow_node_id': self.agent_workflow_node_id, 'next_agent_workflow_node_id': self.next_agent_workflow_node_id, 'chat_type': self.chat_type,
+                'agent_response': self.agent_response, 'error': self.error, 'segments': self.segments, 'streamed_data': self.streamed_data, 'streamed_section_data': self.streamed_section_data, 'highlights': self.highlights, 'llm_display_name': self.llm_display_name, 'llm_bot_icon': self.llm_bot_icon, 'form_response': self.form_response, 'routed_llm': self.routed_llm, 'compute_points_used': self.compute_points_used, 'computer_files': self.computer_files, 'tool_use_request': self.tool_use_request, 'verification_summary': self.verification_summary, 'attached_user_file_names': self.attached_user_file_names, 'forked_deployment_conversation_ids': self.forked_deployment_conversation_ids}
         return {key: value for key, value in resp.items() if value is not None and key not in self.deprecated_keys}

{abacusai-1.4.55 → abacusai-1.4.56}/abacusai/hosted_app_container.py

@@ -28,9 +28,11 @@ class HostedAppContainer(AbstractApiClass):
             artifactType (str): The type of the artifact
             deployedLlmArtifactId (id): The ID of the deployed LLM artifact
             hasDatabase (bool): Whether the app has a database associated to it
+            webAppProjectId (id): The ID of the web app project
+            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):
+    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, webAppProjectId=None, parentConversationId=None):
         super().__init__(client, hostedAppContainerId)
         self.hosted_app_container_id = hostedAppContainerId
         self.hosted_app_id = hostedAppId
@@ -53,11 +55,13 @@ class HostedAppContainer(AbstractApiClass):
         self.artifact_type = artifactType
         self.deployed_llm_artifact_id = deployedLlmArtifactId
         self.has_database = hasDatabase
+        self.web_app_project_id = webAppProjectId
+        self.parent_conversation_id = parentConversationId
         self.deprecated_keys = {}
 
     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)}
+        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'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])
@@ -70,6 +74,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}
+        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, '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.56/abacusai/hostname_info.py

@@ -0,0 +1,33 @@
+from .return_class import AbstractApiClass
+
+
+class HostnameInfo(AbstractApiClass):
+    """
+        Hostname Info
+
+        Args:
+            client (ApiClient): An authenticated API Client instance
+            isRootDomain (bool): Whether the hostname is a root domain
+    """
+
+    def __init__(self, client, isRootDomain=None):
+        super().__init__(client, None)
+        self.is_root_domain = isRootDomain
+        self.deprecated_keys = {}
+
+    def __repr__(self):
+        repr_dict = {f'is_root_domain': repr(self.is_root_domain)}
+        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])
+        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 = {'is_root_domain': self.is_root_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.55 → abacusai-1.4.56}/abacusai/llm_artifact.py

@@ -14,9 +14,11 @@ class LlmArtifact(AbstractApiClass):
             webAppDeploymentId (id): The ID of the associated web app deployment
             deploymentStatus (str): The status of the associated web app deployment
             isLatest (bool): Whether it is the most recent version of the artifact
+            deploymentConversationId (id): The ID of the deployment conversation
+            webAppProjectId (id): The ID of the web app project
     """
 
-    def __init__(self, client, llmArtifactId=None, info=None, description=None, createdAt=None, webAppDeploymentId=None, deploymentStatus=None, isLatest=None):
+    def __init__(self, client, llmArtifactId=None, info=None, description=None, createdAt=None, webAppDeploymentId=None, deploymentStatus=None, isLatest=None, deploymentConversationId=None, webAppProjectId=None):
         super().__init__(client, llmArtifactId)
         self.llm_artifact_id = llmArtifactId
         self.info = info
@@ -25,11 +27,13 @@ class LlmArtifact(AbstractApiClass):
         self.web_app_deployment_id = webAppDeploymentId
         self.deployment_status = deploymentStatus
         self.is_latest = isLatest
+        self.deployment_conversation_id = deploymentConversationId
+        self.web_app_project_id = webAppProjectId
         self.deprecated_keys = {}
 
     def __repr__(self):
-        repr_dict = {f'llm_artifact_id': repr(self.llm_artifact_id), f'info': repr(self.info), f'description': repr(self.description), f'created_at': repr(
-            self.created_at), f'web_app_deployment_id': repr(self.web_app_deployment_id), f'deployment_status': repr(self.deployment_status), f'is_latest': repr(self.is_latest)}
+        repr_dict = {f'llm_artifact_id': repr(self.llm_artifact_id), f'info': repr(self.info), f'description': repr(self.description), f'created_at': repr(self.created_at), f'web_app_deployment_id': repr(
+            self.web_app_deployment_id), f'deployment_status': repr(self.deployment_status), f'is_latest': repr(self.is_latest), f'deployment_conversation_id': repr(self.deployment_conversation_id), f'web_app_project_id': repr(self.web_app_project_id)}
         class_name = "LlmArtifact"
         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])
@@ -42,6 +46,6 @@ class LlmArtifact(AbstractApiClass):
         Returns:
             dict: The dict value representation of the class parameters
         """
-        resp = {'llm_artifact_id': self.llm_artifact_id, 'info': self.info, 'description': self.description, 'created_at': self.created_at,
-                'web_app_deployment_id': self.web_app_deployment_id, 'deployment_status': self.deployment_status, 'is_latest': self.is_latest}
+        resp = {'llm_artifact_id': self.llm_artifact_id, 'info': self.info, 'description': self.description, 'created_at': self.created_at, 'web_app_deployment_id': self.web_app_deployment_id,
+                'deployment_status': self.deployment_status, 'is_latest': self.is_latest, 'deployment_conversation_id': self.deployment_conversation_id, 'web_app_project_id': self.web_app_project_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.56/abacusai/llm_artifact_conversation.py

@@ -0,0 +1,43 @@
+from .return_class import AbstractApiClass
+
+
+class LlmArtifactConversation(AbstractApiClass):
+    """
+        LLM Artifact Conversation
+
+        Args:
+            client (ApiClient): An authenticated API Client instance
+            deploymentConversationId (id): The ID of the deployment conversation
+            llmArtifactId (id): The ID of the LLM artifact
+            deploymentConversationName (str): The name of the conversation
+            externalApplicationId (str): The external application ID
+            createdAt (str): The creation timestamp
+    """
+
+    def __init__(self, client, deploymentConversationId=None, llmArtifactId=None, deploymentConversationName=None, externalApplicationId=None, createdAt=None):
+        super().__init__(client, None)
+        self.deployment_conversation_id = deploymentConversationId
+        self.llm_artifact_id = llmArtifactId
+        self.deployment_conversation_name = deploymentConversationName
+        self.external_application_id = externalApplicationId
+        self.created_at = createdAt
+        self.deprecated_keys = {}
+
+    def __repr__(self):
+        repr_dict = {f'deployment_conversation_id': repr(self.deployment_conversation_id), f'llm_artifact_id': repr(self.llm_artifact_id), f'deployment_conversation_name': repr(
+            self.deployment_conversation_name), f'external_application_id': repr(self.external_application_id), f'created_at': repr(self.created_at)}
+        class_name = "LlmArtifactConversation"
+        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 = {'deployment_conversation_id': self.deployment_conversation_id, 'llm_artifact_id': self.llm_artifact_id,
+                'deployment_conversation_name': self.deployment_conversation_name, 'external_application_id': self.external_application_id, 'created_at': self.created_at}
+        return {key: value for key, value in resp.items() if value is not None and key not in self.deprecated_keys}

abacusai-1.4.56/abacusai/mcp_server_connection.py

@@ -0,0 +1,59 @@
+from .return_class import AbstractApiClass
+
+
+class McpServerConnection(AbstractApiClass):
+    """
+        Model Context Protocol Server Connection
+
+        Args:
+            client (ApiClient): An authenticated API Client instance
+            mcpServerConnectionId (id): the id of the MCP server connection.
+            createdAt (str): the date and time the MCP server connection was created.
+            updatedAt (str): the date and time the MCP server connection was updated.
+            name (str): The name of the MCP server.
+            config (dict): a dictionary containing the config for the MCP server.
+            description (str): description of what the MCP server does.
+            transport (str): the transport type for the MCP server.
+            authType (str): the auth type for the MCP server.
+            externalConnectionId (id): the external connection id for the MCP server.
+            inactive (bool): whether the MCP server is inactive.
+            tools (list): the tools for the MCP server.
+            errorMsg (str): the error message for the MCP server.
+            metadata (dict): the metadata for the MCP server.
+    """
+
+    def __init__(self, client, mcpServerConnectionId=None, createdAt=None, updatedAt=None, name=None, config=None, description=None, transport=None, authType=None, externalConnectionId=None, inactive=None, tools=None, errorMsg=None, metadata=None):
+        super().__init__(client, mcpServerConnectionId)
+        self.mcp_server_connection_id = mcpServerConnectionId
+        self.created_at = createdAt
+        self.updated_at = updatedAt
+        self.name = name
+        self.config = config
+        self.description = description
+        self.transport = transport
+        self.auth_type = authType
+        self.external_connection_id = externalConnectionId
+        self.inactive = inactive
+        self.tools = tools
+        self.error_msg = errorMsg
+        self.metadata = metadata
+        self.deprecated_keys = {}
+
+    def __repr__(self):
+        repr_dict = {f'mcp_server_connection_id': repr(self.mcp_server_connection_id), f'created_at': repr(self.created_at), f'updated_at': repr(self.updated_at), f'name': repr(self.name), f'config': repr(self.config), f'description': repr(self.description), f'transport': repr(
+            self.transport), f'auth_type': repr(self.auth_type), f'external_connection_id': repr(self.external_connection_id), f'inactive': repr(self.inactive), f'tools': repr(self.tools), f'error_msg': repr(self.error_msg), f'metadata': repr(self.metadata)}
+        class_name = "McpServerConnection"
+        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 = {'mcp_server_connection_id': self.mcp_server_connection_id, 'created_at': self.created_at, 'updated_at': self.updated_at, 'name': self.name, 'config': self.config, 'description': self.description,
+                'transport': self.transport, 'auth_type': self.auth_type, 'external_connection_id': self.external_connection_id, 'inactive': self.inactive, 'tools': self.tools, 'error_msg': self.error_msg, 'metadata': self.metadata}
+        return {key: value for key, value in resp.items() if value is not None and key not in self.deprecated_keys}

abacusai-1.4.56/abacusai/mcp_server_query_result.py

@@ -0,0 +1,33 @@
+from .return_class import AbstractApiClass
+
+
+class McpServerQueryResult(AbstractApiClass):
+    """
+        Result of a MCP server query
+
+        Args:
+            client (ApiClient): An authenticated API Client instance
+            output (dict): The execution logs as well as the final output of the MCP server query
+    """
+
+    def __init__(self, client, output=None):
+        super().__init__(client, None)
+        self.output = output
+        self.deprecated_keys = {}
+
+    def __repr__(self):
+        repr_dict = {f'output': repr(self.output)}
+        class_name = "McpServerQueryResult"
+        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 = {'output': self.output}
+        return {key: value for key, value in resp.items() if value is not None and key not in self.deprecated_keys}

{abacusai-1.4.55 → abacusai-1.4.56}/abacusai/model.py

@@ -351,7 +351,7 @@ class Model(AbstractApiClass):
         """
         return self.client.get_model_training_types_for_deployment(self.model_id, model_version, algorithm)
 
-    def update_agent(self, function_source_code: str = None, agent_function_name: str = None, memory: int = None, package_requirements: list = None, description: str = None, enable_binary_input: bool = None, agent_input_schema: dict = None, agent_output_schema: dict = None, workflow_graph: Union[dict, WorkflowGraph] = None, agent_interface: Union[dict, AgentInterface] = None, included_modules: List = None, org_level_connectors: List = None, user_level_connectors: Dict = None, initialize_function_name: str = None, initialize_function_code: str = None, agent_mcp_config: dict = None):
+    def update_agent(self, function_source_code: str = None, agent_function_name: str = None, memory: int = None, package_requirements: list = None, description: str = None, enable_binary_input: bool = None, agent_input_schema: dict = None, agent_output_schema: dict = None, workflow_graph: Union[dict, WorkflowGraph] = None, agent_interface: Union[dict, AgentInterface] = None, included_modules: List = None, org_level_connectors: List = None, user_level_connectors: Dict = None, initialize_function_name: str = None, initialize_function_code: str = None):
         """
         Updates an existing AI Agent. A new version of the agent will be created and published.
 
@@ -366,12 +366,11 @@ class Model(AbstractApiClass):
             user_level_connectors (Dict): A dictionary mapping ApplicationConnectorType keys to lists of OAuth scopes. Each key represents a specific user level application connector, while the value is a list of scopes that define the permissions granted to the application.
             initialize_function_name (str): The name of the function to be used for initialization.
             initialize_function_code (str): The function code to be used for initialization.
-            agent_mcp_config (dict): The MCP config for the agent.
 
         Returns:
             Agent: The updated agent.
         """
-        return self.client.update_agent(self.model_id, function_source_code, agent_function_name, memory, package_requirements, description, enable_binary_input, agent_input_schema, agent_output_schema, workflow_graph, agent_interface, included_modules, org_level_connectors, user_level_connectors, initialize_function_name, initialize_function_code, agent_mcp_config)
+        return self.client.update_agent(self.model_id, function_source_code, agent_function_name, memory, package_requirements, description, enable_binary_input, agent_input_schema, agent_output_schema, workflow_graph, agent_interface, included_modules, org_level_connectors, user_level_connectors, initialize_function_name, initialize_function_code)
 
     def wait_for_training(self, timeout=None):
         """

{abacusai-1.4.55 → abacusai-1.4.56}/abacusai/project.py

@@ -609,7 +609,7 @@ class Project(AbstractApiClass):
         """
         return self.client.create_chat_session(self.project_id, name)
 
-    def create_agent(self, function_source_code: str = None, agent_function_name: str = None, name: str = None, memory: int = None, package_requirements: list = [], description: str = None, enable_binary_input: bool = False, evaluation_feature_group_id: str = None, agent_input_schema: dict = None, agent_output_schema: dict = None, workflow_graph: Union[dict, WorkflowGraph] = None, agent_interface: Union[dict, AgentInterface] = AgentInterface.DEFAULT, included_modules: List = None, org_level_connectors: List = None, user_level_connectors: Dict = None, initialize_function_name: str = None, initialize_function_code: str = None, agent_mcp_config: dict = None):
+    def create_agent(self, function_source_code: str = None, agent_function_name: str = None, name: str = None, memory: int = None, package_requirements: list = [], description: str = None, enable_binary_input: bool = False, evaluation_feature_group_id: str = None, agent_input_schema: dict = None, agent_output_schema: dict = None, workflow_graph: Union[dict, WorkflowGraph] = None, agent_interface: Union[dict, AgentInterface] = AgentInterface.DEFAULT, included_modules: List = None, org_level_connectors: List = None, user_level_connectors: Dict = None, initialize_function_name: str = None, initialize_function_code: str = None):
         """
         Creates a new AI agent using the given agent workflow graph definition.
 
@@ -626,12 +626,11 @@ class Project(AbstractApiClass):
             user_level_connectors (Dict): A dictionary mapping ApplicationConnectorType keys to lists of OAuth scopes. Each key represents a specific user level application connector, while the value is a list of scopes that define the permissions granted to the application.
             initialize_function_name (str): The name of the function to be used for initialization.
             initialize_function_code (str): The function code to be used for initialization.
-            agent_mcp_config (dict): The MCP config for the agent.
 
         Returns:
             Agent: The new agent.
         """
-        return self.client.create_agent(self.project_id, function_source_code, agent_function_name, name, memory, package_requirements, description, enable_binary_input, evaluation_feature_group_id, agent_input_schema, agent_output_schema, workflow_graph, agent_interface, included_modules, org_level_connectors, user_level_connectors, initialize_function_name, initialize_function_code, agent_mcp_config)
+        return self.client.create_agent(self.project_id, function_source_code, agent_function_name, name, memory, package_requirements, description, enable_binary_input, evaluation_feature_group_id, agent_input_schema, agent_output_schema, workflow_graph, agent_interface, included_modules, org_level_connectors, user_level_connectors, initialize_function_name, initialize_function_code)
 
     def generate_agent_code(self, prompt: str, fast_mode: bool = None):
         """

{abacusai-1.4.55 → abacusai-1.4.56/abacusai.egg-info}/PKG-INFO

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