Version 1.4.42

Author: arvindsun

abacusai/__init__.py

@@ -43,6 +43,7 @@
 from .code_source import CodeSource
 from .code_suggestion_validation_response import CodeSuggestionValidationResponse
 from .code_summary_response import CodeSummaryResponse
+from .codellm_embedding_constants import CodellmEmbeddingConstants
 from .compute_point_info import ComputePointInfo
 from .concatenation_config import ConcatenationConfig
 from .constants_autocomplete_response import ConstantsAutocompleteResponse
@@ -255,4 +256,4 @@
 from .workflow_node_template import WorkflowNodeTemplate
 
 
-__version__ = "1.4.41"
+__version__ = "1.4.42"

abacusai/api_class/enums.py

@@ -515,6 +515,7 @@ class LLMName(ApiEnum):
     QWEN_2_5_32B_BASE = 'QWEN_2_5_32B_BASE'
     QWEN_2_5_72B = 'QWEN_2_5_72B'
     QWQ_32B = 'QWQ_32B'
+    QWEN3_235B_A22B = 'QWEN3_235B_A22B'
     GEMINI_1_5_FLASH = 'GEMINI_1_5_FLASH'
     XAI_GROK = 'XAI_GROK'
     XAI_GROK_3 = 'XAI_GROK_3'

abacusai/client.py

@@ -661,7 +661,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.41'
+    client_version = '1.4.42'
 
     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
@@ -857,6 +857,13 @@ def _proxy_request(self, name: str, method: str = 'POST', query_params: dict = N
         if deployment_id:
             query_params = {**(query_params or {}),
                             'environmentDeploymentId': deployment_id}
+        caller = self._get_agent_caller()
+        request_id = self._get_agent_app_request_id()
+        if caller and request_id:
+            if get_object_from_context(self, _request_context, 'is_agent', bool):
+                query_params = {**(query_params or {}), 'isAgent': True}
+            if get_object_from_context(self, _request_context, 'is_agent_api', bool):
+                query_params = {**(query_params or {}), 'isAgentApi': True}
         endpoint = self.proxy_endpoint
         if endpoint is None:
             raise Exception(
@@ -1554,17 +1561,18 @@ def list_application_connector_objects(self, application_connector_id: str) -> l
             application_connector_id (str): Unique string identifier for the application connector."""
         return self._call_api('listApplicationConnectorObjects', 'GET', query_params={'applicationConnectorId': application_connector_id})
 
-    def get_connector_auth(self, service: Union[ApplicationConnectorType, str] = None, application_connector_id: str = None, scopes: List = None) -> ApplicationConnector:
+    def get_connector_auth(self, service: Union[ApplicationConnectorType, str] = None, application_connector_id: str = None, scopes: List = None, is_database_connector: bool = None) -> ApplicationConnector:
         """Get the authentication details for a given connector. For user level connectors, the service is required. For org level connectors, the application_connector_id is required.
 
         Args:
             service (ApplicationConnectorType): The service name.
             application_connector_id (str): The unique ID associated with the connector.
             scopes (List): The scopes to request for the connector.
+            is_database_connector (bool): Whether the connector is a database connector.
 
         Returns:
             ApplicationConnector: The application connector with the authentication details."""
-        return self._call_api('getConnectorAuth', 'GET', query_params={'service': service, 'applicationConnectorId': application_connector_id, 'scopes': scopes}, parse_type=ApplicationConnector)
+        return self._call_api('getConnectorAuth', 'GET', query_params={'service': service, 'applicationConnectorId': application_connector_id, 'scopes': scopes, 'isDatabaseConnector': is_database_connector}, parse_type=ApplicationConnector)
 
     def list_streaming_connectors(self) -> List[StreamingConnector]:
         """Retrieves a list of all streaming connectors along with their corresponding attributes.
@@ -2687,7 +2695,7 @@ def list_chat_sessions(self, most_recent_per_project: bool = False) -> ChatSessi
             ChatSession: The chat sessions with Data Science Co-pilot"""
         return self._call_api('listChatSessions', 'GET', query_params={'mostRecentPerProject': most_recent_per_project}, parse_type=ChatSession)
 
-    def get_deployment_conversation(self, deployment_conversation_id: str = None, external_session_id: str = None, deployment_id: str = None, filter_intermediate_conversation_events: bool = True, get_unused_document_uploads: bool = False) -> DeploymentConversation:
+    def get_deployment_conversation(self, deployment_conversation_id: str = None, 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) -> DeploymentConversation:
         """Gets a deployment conversation.
 
         Args:
@@ -2696,10 +2704,12 @@ def get_deployment_conversation(self, deployment_conversation_id: str = None, ex
             deployment_id (str): The deployment this conversation belongs to. This is required if not logged in.
             filter_intermediate_conversation_events (bool): If true, intermediate conversation events will be filtered out. Default is true.
             get_unused_document_uploads (bool): If true, unused document uploads will be returned. Default is false.
+            start (int): The start index of the conversation.
+            limit (int): The limit of the conversation.
 
         Returns:
             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}, parse_type=DeploymentConversation, is_sync=True)
+        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}, parse_type=DeploymentConversation, is_sync=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.
@@ -3344,7 +3354,8 @@ def create_python_function_from_function(self,
                                              function: callable,
                                              function_variable_mappings: list = None,
                                              package_requirements: list = None,
-                                             function_type: str = PythonFunctionType.FEATURE_GROUP.value):
+                                             function_type: str = PythonFunctionType.FEATURE_GROUP.value,
+                                             description: str = None):
         """
         Creates a custom Python function
 
@@ -3354,6 +3365,7 @@ def create_python_function_from_function(self,
             function_variable_mappings (List<PythonFunctionArguments>): List of Python function arguments.
             package_requirements (List): List of package requirement strings. For example: ['numpy==1.2.3', 'pandas>=1.4.0'].
             function_type (PythonFunctionType): Type of Python function to create. Default is FEATURE_GROUP, but can also be PLOTLY_FIG.
+            description (str): Description of the Python function.
         """
         function_source = None
         python_function_name = None
@@ -3368,7 +3380,8 @@ def create_python_function_from_function(self,
                                            function_name=python_function_name,
                                            function_variable_mappings=function_variable_mappings,
                                            package_requirements=package_requirements,
-                                           function_type=function_type)
+                                           function_type=function_type,
+                                           description=description)
 
     def create_feature_group_from_python_function(self, function: callable, table_name: str, input_tables: list = None, python_function_name: str = None, python_function_bindings: list = None, cpu_size: str = None, memory: int = None, package_requirements: list = None, included_modules: list = None):
         """
@@ -4148,11 +4161,11 @@ def streaming_evaluate_prompt(self, prompt: str = None, system_message: str = No
         if caller and request_id:
             is_agent = get_object_from_context(
                 self, _request_context, 'is_agent', bool)
-            is_api = get_object_from_context(
+            is_agent_api = get_object_from_context(
                 self, _request_context, 'is_agent_api', bool)
 
-            result = self._stream_llm_call(prompt=prompt, system_message=system_message, llm_name=llm_name, max_tokens=max_tokens, temperature=temperature,
-                                           messages=messages, response_type=response_type, json_response_schema=json_response_schema, section_key=section_key, is_agent=is_agent, is_api=is_api)
+            result = self._stream_llm_call(prompt=prompt, system_message=system_message, llm_name=llm_name, max_tokens=max_tokens, temperature=temperature, messages=messages,
+                                           response_type=response_type, json_response_schema=json_response_schema, section_key=section_key, is_agent=is_agent, is_agent_api=is_agent_api)
         else:
             result = self.evaluate_prompt(prompt, system_message=system_message, llm_name=llm_name, max_tokens=max_tokens,
                                           temperature=temperature, messages=messages, response_type=response_type, json_response_schema=json_response_schema).content

abacusai/code_embeddings.py

@@ -8,15 +8,18 @@ class CodeEmbeddings(AbstractApiClass):
         Args:
             client (ApiClient): An authenticated API Client instance
             embeddings (dict): A dictionary mapping the file name to its embeddings.
+            chunkingScheme (str): The scheme used for chunking the embeddings.
     """
 
-    def __init__(self, client, embeddings=None):
+    def __init__(self, client, embeddings=None, chunkingScheme=None):
         super().__init__(client, None)
         self.embeddings = embeddings
+        self.chunking_scheme = chunkingScheme
         self.deprecated_keys = {}
 
     def __repr__(self):
-        repr_dict = {f'embeddings': repr(self.embeddings)}
+        repr_dict = {f'embeddings': repr(
+            self.embeddings), f'chunking_scheme': repr(self.chunking_scheme)}
         class_name = "CodeEmbeddings"
         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])
@@ -29,5 +32,6 @@ def to_dict(self):
         Returns:
             dict: The dict value representation of the class parameters
         """
-        resp = {'embeddings': self.embeddings}
+        resp = {'embeddings': self.embeddings,
+                'chunking_scheme': self.chunking_scheme}
         return {key: value for key, value in resp.items() if value is not None and key not in self.deprecated_keys}

abacusai/codellm_embedding_constants.py

@@ -0,0 +1,43 @@
+from .return_class import AbstractApiClass
+
+
+class CodellmEmbeddingConstants(AbstractApiClass):
+    """
+        A dictionary of constants to be used in the autocomplete.
+
+        Args:
+            client (ApiClient): An authenticated API Client instance
+            maxSupportedWorkspaceFiles (int): Max supported workspace files
+            maxSupportedWorkspaceChunks (int): Max supported workspace chunks
+            maxConcurrentRequests (int): Max concurrent requests
+            fileExtensionToChunkingScheme (dict): Map between the file extensions and their chunking schema
+            idleTimeoutSeconds (int): The idle timeout without any activity before the workspace is refreshed.
+    """
+
+    def __init__(self, client, maxSupportedWorkspaceFiles=None, maxSupportedWorkspaceChunks=None, maxConcurrentRequests=None, fileExtensionToChunkingScheme=None, idleTimeoutSeconds=None):
+        super().__init__(client, None)
+        self.max_supported_workspace_files = maxSupportedWorkspaceFiles
+        self.max_supported_workspace_chunks = maxSupportedWorkspaceChunks
+        self.max_concurrent_requests = maxConcurrentRequests
+        self.file_extension_to_chunking_scheme = fileExtensionToChunkingScheme
+        self.idle_timeout_seconds = idleTimeoutSeconds
+        self.deprecated_keys = {}
+
+    def __repr__(self):
+        repr_dict = {f'max_supported_workspace_files': repr(self.max_supported_workspace_files), f'max_supported_workspace_chunks': repr(self.max_supported_workspace_chunks), f'max_concurrent_requests': repr(
+            self.max_concurrent_requests), f'file_extension_to_chunking_scheme': repr(self.file_extension_to_chunking_scheme), f'idle_timeout_seconds': repr(self.idle_timeout_seconds)}
+        class_name = "CodellmEmbeddingConstants"
+        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 = {'max_supported_workspace_files': self.max_supported_workspace_files, 'max_supported_workspace_chunks': self.max_supported_workspace_chunks,
+                'max_concurrent_requests': self.max_concurrent_requests, 'file_extension_to_chunking_scheme': self.file_extension_to_chunking_scheme, 'idle_timeout_seconds': self.idle_timeout_seconds}
+        return {key: value for key, value in resp.items() if value is not None and key not in self.deprecated_keys}