abacusai 0.65.0__tar.gz → 0.65.1__tar.gz

{abacusai-0.65.0/abacusai.egg-info → abacusai-0.65.1}/PKG-INFO

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

{abacusai-0.65.0 → abacusai-0.65.1}/abacusai/__init__.py

@@ -4,4 +4,4 @@ from .prediction_client import PredictionClient
 from .streaming_client import StreamingClient
 
 
-__version__ = "0.65.0"
+__version__ = "0.65.1"

{abacusai-0.65.0 → abacusai-0.65.1}/abacusai/client.py

@@ -201,7 +201,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 = '0.65.0'
+    client_version = '0.65.1'
 
     def __init__(self, api_key: str = None, server: str = None, client_options: ClientOptions = None, skip_version_check: bool = False):
         self.api_key = api_key
@@ -284,7 +284,7 @@ class BaseApiClient:
 
     def _call_api(
             self, action, method, query_params=None,
-            body=None, files=None, parse_type=None, streamable_response=False, server_override=None):
+            body=None, files=None, parse_type=None, streamable_response=False, server_override=None, timeout=None):
         headers = {'apiKey': self.api_key, 'clientVersion': self.client_version,
                    'User-Agent': f'python-abacusai-{self.client_version}', 'notebookId': self.notebook_id}
         url = (server_override or self.server) + '/api/v0/' + action
@@ -295,8 +295,8 @@ class BaseApiClient:
                 'deploymentId') if query_params else None, query_params.get('deploymentToken') if query_params else None)
             if discovered_url:
                 url = discovered_url + '/api/' + action
-        response = self._request(url, method, query_params=query_params,
-                                 headers=headers, body=body, files=files, stream=streamable_response)
+        response = self._request(url, method, query_params=query_params, headers=headers,
+                                 body=body, files=files, stream=streamable_response, timeout=timeout)
 
         result = None
         success = False
@@ -340,17 +340,17 @@ class BaseApiClient:
         return return_class(self, **{key: val for key, val in values.items() if key in type_inputs})
 
     def _request(self, url, method, query_params=None, headers=None,
-                 body=None, files=None, stream=False):
+                 body=None, files=None, stream=False, timeout=None):
         if method == 'GET':
             cleaned_params = {key: ','.join([str(item) for item in val]) if isinstance(
                 val, list) else val for key, val in query_params.items()} if query_params else query_params
             return _requests_retry_session().get(url, params=cleaned_params, headers=headers, stream=stream)
         elif method == 'POST':
-            return _requests_retry_session().post(url, params=query_params, json=body, headers=headers, files=files, timeout=90)
+            return _requests_retry_session().post(url, params=query_params, json=body, headers=headers, files=files, timeout=timeout or 200)
         elif method == 'PUT':
-            return _requests_retry_session().put(url, params=query_params, data=body, headers=headers, files=files, timeout=90)
+            return _requests_retry_session().put(url, params=query_params, data=body, headers=headers, files=files, timeout=timeout or 200)
         elif method == 'PATCH':
-            return _requests_retry_session().patch(url, params=query_params, json=body, headers=headers, files=files, timeout=90)
+            return _requests_retry_session().patch(url, params=query_params, json=body, headers=headers, files=files, timeout=timeout or 200)
         elif method == 'DELETE':
             return _requests_retry_session().delete(url, params=query_params, data=body, headers=headers)
         else:
@@ -1863,7 +1863,7 @@ class ReadOnlyClient(BaseApiClient):
 
         Returns:
             LlmResponse: The response from the model, raw text and parsed components."""
-        return self._call_api('queryFeatureGroupCodeGenerator', 'GET', query_params={'query': query, 'language': language, 'projectId': project_id, 'llmName': llm_name, 'maxTokens': max_tokens}, parse_type=LlmResponse)
+        return self._call_api('queryFeatureGroupCodeGenerator', 'GET', query_params={'query': query, 'language': language, 'projectId': project_id, 'llmName': llm_name, 'maxTokens': max_tokens}, parse_type=LlmResponse, timeout=300)
 
 
 def get_source_code_info(train_function: callable, predict_function: callable = None, predict_many_function: callable = None, initialize_function: callable = None, common_functions: list = None):
@@ -2152,7 +2152,7 @@ class ApiClient(ReadOnlyClient):
             included_modules (list): A list of user-created modules that will be included, which is equivalent to 'from module import *'
         """
         if function:
-            function_source = inspect.getsource(function)
+            function_source = get_clean_function_source_code(function)
             function_source = include_modules(
                 function_source, included_modules)
             python_function_name = python_function_name or function.__name__
@@ -4929,16 +4929,17 @@ Creates a new feature group defined as the union of other feature group versions
             deployment_id, deployment_token)
         return self._call_api('getChatResponse', 'POST', query_params={'deploymentToken': deployment_token, 'deploymentId': deployment_id}, body={'messages': messages, 'searchResults': search_results, 'chatConfig': chat_config}, server_override=prediction_url)
 
-    def get_search_results(self, deployment_token: str, deployment_id: str, query_data: dict) -> Dict:
+    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.
 
         Args:
             deployment_token (str): A token used to authenticate access to created deployments. This token is only authorized to predict on deployments in this project, so it can be securely embedded in an application or website.
             deployment_id (str): A unique identifier of a deployment created under the project.
-            query_data (dict): A dictionary where the key is "Content" and the value is the text from which entities are to be extracted."""
+            query_data (dict): A dictionary where the key is "Content" and the value is the text from which entities are to be extracted.
+            num (int): Number of search results to return."""
         prediction_url = self._get_prediction_endpoint(
             deployment_id, deployment_token)
-        return self._call_api('getSearchResults', 'POST', query_params={'deploymentToken': deployment_token, 'deploymentId': deployment_id}, body={'queryData': query_data}, server_override=prediction_url)
+        return self._call_api('getSearchResults', 'POST', query_params={'deploymentToken': deployment_token, 'deploymentId': deployment_id}, body={'queryData': query_data, 'num': num}, server_override=prediction_url)
 
     def get_sentiment(self, deployment_token: str, deployment_id: str, document: str) -> Dict:
         """Predicts sentiment on a document
@@ -5848,7 +5849,7 @@ Creates a new feature group defined as the union of other feature group versions
             long_explanation (str): verbose explanation of the artifact with given ID
             feature_group_id (str): A unique string identifier associated with the Feature Group.
             feature_group_version (str): A unique string identifier associated with the Feature Group Version."""
-        return self._call_api('setNaturalLanguageExplanation', 'POST', query_params={}, body={'shortExplanation': short_explanation, 'longExplanation': long_explanation, 'featureGroupId': feature_group_id, 'featureGroupVersion': feature_group_version})
+        return self._call_api('setNaturalLanguageExplanation', 'POST', query_params={}, body={'shortExplanation': short_explanation, 'longExplanation': long_explanation, 'featureGroupId': feature_group_id, 'featureGroupVersion': feature_group_version}, timeout=300)
 
     def create_chat_session(self, project_id: str = None) -> ChatSession:
         """Creates a chat session with Abacus Chat.
@@ -5902,7 +5903,7 @@ Creates a new feature group defined as the union of other feature group versions
 
         Returns:
             LlmResponse: The response from the model, raw text and parsed components."""
-        return self._call_api('evaluatePrompt', 'POST', query_params={}, body={'prompt': prompt, 'systemMessage': system_message, 'llmName': llm_name, 'maxTokens': max_tokens}, parse_type=LlmResponse)
+        return self._call_api('evaluatePrompt', 'POST', query_params={}, body={'prompt': prompt, 'systemMessage': system_message, 'llmName': llm_name, 'maxTokens': max_tokens}, parse_type=LlmResponse, timeout=300)
 
     def render_feature_groups_for_llm(self, feature_group_ids: list, token_budget: int = None, include_definition: bool = True) -> LlmInput:
         """Encode feature groups as language model inputs.

{abacusai-0.65.0 → abacusai-0.65.1}/abacusai/feature_group_export.py

@@ -110,6 +110,15 @@ class FeatureGroupExport(AbstractApiClass):
         """
         return self.client._poll(self, {'PENDING', 'EXPORTING'}, timeout=timeout)
 
+    def wait_for_export(self, timeout=3600):
+        """
+        A waiting call until feature group export is created.
+
+        Args:
+            timeout (int, optional): The waiting time given to the call to finish, if it doesn't finish by the allocated time, the call is said to be timed out.
+        """
+        return self.wait_for_results(timeout=timeout)
+
     def get_status(self):
         """
         Gets the status of the feature group export.

{abacusai-0.65.0 → abacusai-0.65.1}/abacusai/prediction_client.py

@@ -322,16 +322,17 @@ class PredictionClient(BaseApiClient):
             deployment_id, deployment_token)
         return self._call_api('getChatResponse', 'POST', query_params={'deploymentToken': deployment_token, 'deploymentId': deployment_id}, body={'messages': messages, 'searchResults': search_results, 'chatConfig': chat_config}, server_override=prediction_url)
 
-    def get_search_results(self, deployment_token: str, deployment_id: str, query_data: dict) -> Dict:
+    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.
 
         Args:
             deployment_token (str): A token used to authenticate access to created deployments. This token is only authorized to predict on deployments in this project, so it can be securely embedded in an application or website.
             deployment_id (str): A unique identifier of a deployment created under the project.
-            query_data (dict): A dictionary where the key is "Content" and the value is the text from which entities are to be extracted."""
+            query_data (dict): A dictionary where the key is "Content" and the value is the text from which entities are to be extracted.
+            num (int): Number of search results to return."""
         prediction_url = self._get_prediction_endpoint(
             deployment_id, deployment_token)
-        return self._call_api('getSearchResults', 'POST', query_params={'deploymentToken': deployment_token, 'deploymentId': deployment_id}, body={'queryData': query_data}, server_override=prediction_url)
+        return self._call_api('getSearchResults', 'POST', query_params={'deploymentToken': deployment_token, 'deploymentId': deployment_id}, body={'queryData': query_data, 'num': num}, server_override=prediction_url)
 
     def get_sentiment(self, deployment_token: str, deployment_id: str, document: str) -> Dict:
         """Predicts sentiment on a document

abacusai-0.65.1/abacusai/public.pem

@@ -0,0 +1,5 @@
+-----BEGIN RSA PUBLIC KEY-----
+MIGJAoGBAJ1ljtCT1s7OPVDjFmmMuTv9BwGEx5ACdz2fcygW6pl7rUfWoYBrjV5k
+s0f28tObziI0+4YRwiGnt2lezvqVxOWakp5fVUVJAKMGtAkhoqK6gjQfwYiEsFkQ
+JE/sR2mdagI4pia38vCAITWLwQJfwRI9HGBVlXkOen2XvMuRTpNpAgMBAAE=
+-----END RSA PUBLIC KEY-----

{abacusai-0.65.0 → abacusai-0.65.1/abacusai.egg-info}/PKG-INFO

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

{abacusai-0.65.0 → abacusai-0.65.1}/abacusai.egg-info/SOURCES.txt

@@ -232,6 +232,7 @@ docs/_sources/autoapi/abacusai/embedding_feature_drift_distribution/index.rst.tx
 docs/_sources/autoapi/abacusai/execute_feature_group_operation/index.rst.txt
 docs/_sources/autoapi/abacusai/execute_fg_operation/index.rst.txt
 docs/_sources/autoapi/abacusai/feature/index.rst.txt
+docs/_sources/autoapi/abacusai/feature_distribution/index.rst.txt
 docs/_sources/autoapi/abacusai/feature_drift_summary/index.rst.txt
 docs/_sources/autoapi/abacusai/feature_group/index.rst.txt
 docs/_sources/autoapi/abacusai/feature_group_export/index.rst.txt
@@ -326,6 +327,8 @@ docs/_sources/autoapi/abacusai/use_case/index.rst.txt
 docs/_sources/autoapi/abacusai/use_case_requirements/index.rst.txt
 docs/_sources/autoapi/abacusai/user/index.rst.txt
 docs/_sources/autoapi/abacusai/user_exception/index.rst.txt
+docs/_sources/autoapi/abacusai/vector_store/index.rst.txt
+docs/_sources/autoapi/abacusai/vector_store_version/index.rst.txt
 docs/_sources/autoapi/abacusai/webhook/index.rst.txt
 docs/_static/_sphinx_javascript_frameworks_compat.js
 docs/_static/basic.css
@@ -427,6 +430,7 @@ docs/autoapi/abacusai/embedding_feature_drift_distribution/index.html
 docs/autoapi/abacusai/execute_feature_group_operation/index.html
 docs/autoapi/abacusai/execute_fg_operation/index.html
 docs/autoapi/abacusai/feature/index.html
+docs/autoapi/abacusai/feature_distribution/index.html
 docs/autoapi/abacusai/feature_drift_summary/index.html
 docs/autoapi/abacusai/feature_group/index.html
 docs/autoapi/abacusai/feature_group_export/index.html
@@ -521,6 +525,8 @@ docs/autoapi/abacusai/use_case/index.html
 docs/autoapi/abacusai/use_case_requirements/index.html
 docs/autoapi/abacusai/user/index.html
 docs/autoapi/abacusai/user_exception/index.html
+docs/autoapi/abacusai/vector_store/index.html
+docs/autoapi/abacusai/vector_store_version/index.html
 docs/autoapi/abacusai/webhook/index.html
 examples/fullscript.py
 sphinx/build_docs

{abacusai-0.65.0 → abacusai-0.65.1}/docs/_sources/autoapi/abacusai/annotations_status/index.rst.txt

@@ -17,7 +17,7 @@ Classes
 
 
 
-.. py:class:: AnnotationsStatus(client, total=None, done=None, inProgress=None, todo=None)
+.. py:class:: AnnotationsStatus(client, total=None, done=None, inProgress=None, todo=None, latestUpdatedAt=None)
 
    Bases: :py:obj:`abacusai.return_class.AbstractApiClass`
 
@@ -33,6 +33,8 @@ Classes
    :type inProgress: int
    :param todo: The number of documents that need to be annotated
    :type todo: int
+   :param latestUpdatedAt: The latest time an annotation was updated (ISO-8601 format)
+   :type latestUpdatedAt: str
 
    .. py:method:: __repr__()
 

{abacusai-0.65.0 → abacusai-0.65.1}/docs/_sources/autoapi/abacusai/api_class/enums/index.rst.txt

@@ -26,6 +26,7 @@ Classes
    abacusai.api_class.enums.ForecastingFillMethod
    abacusai.api_class.enums.ForecastingQuanitlesExtensionMethod
    abacusai.api_class.enums.ConnectorType
+   abacusai.api_class.enums.PythonFunctionArgumentType
 
 
 
@@ -696,3 +697,77 @@ Classes
       
 
 
+.. py:class:: PythonFunctionArgumentType
+
+   Bases: :py:obj:`ApiEnum`
+
+   Generic enumeration.
+
+   Derive from this class to define new enumerations.
+
+   .. py:attribute:: FEATURE_GROUP
+      :value: 'FEATURE_GROUP'
+
+      
+
+   .. py:attribute:: INTEGER
+      :value: 'INTEGER'
+
+      
+
+   .. py:attribute:: STRING
+      :value: 'STRING'
+
+      
+
+   .. py:attribute:: BOOLEAN
+      :value: 'BOOLEAN'
+
+      
+
+   .. py:attribute:: FLOAT
+      :value: 'FLOAT'
+
+      
+
+   .. py:attribute:: JSON
+      :value: 'JSON'
+
+      
+
+   .. py:attribute:: LIST
+      :value: 'LIST'
+
+      
+
+   .. py:attribute:: DATASET_ID
+      :value: 'DATASET_ID'
+
+      
+
+   .. py:attribute:: MODEL_ID
+      :value: 'MODEL_ID'
+
+      
+
+   .. py:attribute:: FEATURE_GROUP_ID
+      :value: 'FEATURE_GROUP_ID'
+
+      
+
+   .. py:attribute:: MONITOR_ID
+      :value: 'MONITOR_ID'
+
+      
+
+   .. py:attribute:: BATCH_PREDICTION_ID
+      :value: 'BATCH_PREDICTION_ID'
+
+      
+
+   .. py:attribute:: DEPLOYMENT_ID
+      :value: 'DEPLOYMENT_ID'
+
+      
+
+

{abacusai-0.65.0 → abacusai-0.65.1}/docs/_sources/autoapi/abacusai/chat_session/index.rst.txt

@@ -60,15 +60,4 @@ Classes
       :rtype: ChatSession
 
 
-   .. py:method:: send_chat_message(message)
-
-      Updates chat history with the response from a user message
-
-      :param message: Message you want to send to Abacus Chat
-      :type message: str
-
-      :returns: The chat session with Abacus Chat
-      :rtype: ChatSession
-
-
 

{abacusai-0.65.0 → abacusai-0.65.1}/docs/_sources/autoapi/abacusai/client/index.rst.txt

@@ -104,7 +104,7 @@ Attributes
    :type skip_version_check: bool
 
    .. py:attribute:: client_version
-      :value: '0.64.1'
+      :value: '0.65.0'
 
       
 
@@ -429,6 +429,14 @@ Attributes
       :rtype: ExecuteFeatureGroupOperation
 
 
+   .. py:method:: get_execute_feature_group_operation_result_part_count(feature_group_operation_run_id)
+
+      Gets the number of parts in the result of the execution of fg operation
+
+      :param feature_group_operation_run_id: The unique ID associated with the execution.
+      :type feature_group_operation_run_id: str
+
+
    .. py:method:: download_execute_feature_group_operation_result_part_chunk(feature_group_operation_run_id, part, offset = 0, chunk_size = 10485760)
 
       Downloads a chunk of the result of the execution of fg operation
@@ -1285,6 +1293,9 @@ Attributes
       :param nested_feature_name: Optionally, the name of the nested feature that the feature is in.
       :type nested_feature_name: str
 
+      :returns: An object describing the training and prediction output feature distributions.
+      :rtype: FeatureDistribution
+
 
    .. py:method:: get_outliers_for_feature(model_monitor_version, feature_name = None, nested_feature_name = None)
 
@@ -1549,12 +1560,23 @@ Attributes
       :rtype: PipelineVersion
 
 
-   .. py:method:: describe_pipeline_step(pipeline_name, step_name)
+   .. py:method:: describe_pipeline_step(pipeline_step_id)
 
       Deletes a step from a pipeline.
 
-      :param pipeline_name: The name of the pipeline.
-      :type pipeline_name: str
+      :param pipeline_step_id: The ID of the pipeline step.
+      :type pipeline_step_id: str
+
+      :returns: An object describing the pipeline step.
+      :rtype: PipelineStep
+
+
+   .. py:method:: describe_pipeline_step_by_name(pipeline_id, step_name)
+
+      Describes a pipeline step by the step name.
+
+      :param pipeline_id: The ID of the pipeline.
+      :type pipeline_id: str
       :param step_name: The name of the step.
       :type step_name: str
 
@@ -1973,12 +1995,12 @@ Attributes
       :type use_gpu: bool
 
 
-   .. py:method:: create_pipeline_step_from_function(pipeline_name, step_name, function, step_input_mappings, output_variable_mappings = None, step_dependencies = None, package_requirements = None)
+   .. py:method:: create_pipeline_step_from_function(pipeline_id, step_name, function, step_input_mappings, output_variable_mappings = None, step_dependencies = None, package_requirements = None)
 
       Creates a step in a given pipeline from a python function.
 
-      :param pipeline_name: The name of the pipeline to add the step to.
-      :type pipeline_name: str
+      :param pipeline_id: The ID of the pipeline to add the step to.
+      :type pipeline_id: str
       :param step_name: The name of the step.
       :type step_name: str
       :param function: The python function.
@@ -1993,12 +2015,12 @@ Attributes
       :type package_requirements: list
 
 
-   .. py:method:: update_pipeline_step_from_function(pipeline_name, step_name, function, step_input_mappings, output_variable_mappings = None, step_dependencies = None, package_requirements = None)
+   .. py:method:: update_pipeline_step_by_name_from_function(pipeline_id, step_name, function, step_input_mappings, output_variable_mappings = None, step_dependencies = None, package_requirements = None)
 
       Updates a step in a given pipeline from a python function.
 
-      :param pipeline_name: The name of the pipeline the step belongs to.
-      :type pipeline_name: str
+      :param pipeline_id: The ID of the pipeline the step belongs to.
+      :type pipeline_id: str
       :param step_name: The name of the step.
       :type step_name: str
       :param function: The python function.
@@ -2469,7 +2491,7 @@ Attributes
 
       :param name: The project's name.
       :type name: str
-      :param use_case: The use case that the project solves. Refer to our [guide on use cases](https://api.abacus.ai/app/help/useCases) for further details of each use case. The following enums are currently available for you to choose from:  LANGUAGE_DETECTION,  NLP_SENTIMENT,  NLP_QA,  NLP_SEARCH,  NLP_CHAT,  NLP_SENTENCE_BOUNDARY_DETECTION,  NLP_CLASSIFICATION,  NLP_SUMMARIZATION,  NLP_DOCUMENT_VISUALIZATION,  AI_AGENT,  EMBEDDINGS_ONLY,  MODEL_WITH_EMBEDDINGS,  TORCH_MODEL,  TORCH_MODEL_WITH_EMBEDDINGS,  PYTHON_MODEL,  NOTEBOOK_PYTHON_MODEL,  DOCKER_MODEL,  DOCKER_MODEL_WITH_EMBEDDINGS,  CUSTOMER_CHURN,  ENERGY,  FINANCIAL_METRICS,  CUMULATIVE_FORECASTING,  FRAUD_ACCOUNT,  FRAUD_THREAT,  FRAUD_TRANSACTIONS,  OPERATIONS_CLOUD,  CLOUD_SPEND,  TIMESERIES_ANOMALY_DETECTION,  OPERATIONS_MAINTENANCE,  OPERATIONS_INCIDENT,  PERS_PROMOTIONS,  PREDICTING,  FEATURE_STORE,  RETAIL,  SALES_FORECASTING,  SALES_SCORING,  FEED_RECOMMEND,  USER_RANKINGS,  NAMED_ENTITY_RECOGNITION,  USER_ITEM_AFFINITY,  USER_RECOMMENDATIONS,  USER_RELATED,  VISION,  VISION_REGRESSION,  VISION_OBJECT_DETECTION,  FEATURE_DRIFT,  SCHEDULING,  GENERIC_FORECASTING,  PRETRAINED_IMAGE_TEXT_DESCRIPTION,  PRETRAINED_SPEECH_RECOGNITION,  PRETRAINED_STYLE_TRANSFER,  PRETRAINED_TEXT_TO_IMAGE_GENERATION,  THEME_ANALYSIS,  CLUSTERING,  CLUSTERING_TIMESERIES,  PRETRAINED_INSTRUCT_PIX2PIX,  PRETRAINED_TEXT_CLASSIFICATION.
+      :param use_case: The use case that the project solves. Refer to our [guide on use cases](https://api.abacus.ai/app/help/useCases) for further details of each use case. The following enums are currently available for you to choose from:  LANGUAGE_DETECTION,  NLP_SENTIMENT,  NLP_QA,  NLP_SEARCH,  NLP_CHAT,  NLP_SENTENCE_BOUNDARY_DETECTION,  NLP_CLASSIFICATION,  NLP_SUMMARIZATION,  NLP_DOCUMENT_VISUALIZATION,  AI_AGENT,  EMBEDDINGS_ONLY,  MODEL_WITH_EMBEDDINGS,  TORCH_MODEL,  TORCH_MODEL_WITH_EMBEDDINGS,  PYTHON_MODEL,  NOTEBOOK_PYTHON_MODEL,  DOCKER_MODEL,  DOCKER_MODEL_WITH_EMBEDDINGS,  CUSTOMER_CHURN,  ENERGY,  FINANCIAL_METRICS,  CUMULATIVE_FORECASTING,  FRAUD_ACCOUNT,  FRAUD_THREAT,  FRAUD_TRANSACTIONS,  OPERATIONS_CLOUD,  CLOUD_SPEND,  TIMESERIES_ANOMALY_DETECTION,  OPERATIONS_MAINTENANCE,  OPERATIONS_INCIDENT,  PERS_PROMOTIONS,  PREDICTING,  FEATURE_STORE,  RETAIL,  SALES_FORECASTING,  SALES_SCORING,  FEED_RECOMMEND,  USER_RANKINGS,  NAMED_ENTITY_RECOGNITION,  USER_RECOMMENDATIONS,  USER_RELATED,  VISION,  VISION_REGRESSION,  VISION_OBJECT_DETECTION,  FEATURE_DRIFT,  SCHEDULING,  GENERIC_FORECASTING,  PRETRAINED_IMAGE_TEXT_DESCRIPTION,  PRETRAINED_SPEECH_RECOGNITION,  PRETRAINED_STYLE_TRANSFER,  PRETRAINED_TEXT_TO_IMAGE_GENERATION,  THEME_ANALYSIS,  CLUSTERING,  CLUSTERING_TIMESERIES,  PRETRAINED_INSTRUCT_PIX2PIX,  PRETRAINED_TEXT_CLASSIFICATION.
       :type use_case: str
 
       :returns: This object represents the newly created project. For details, refer to.
@@ -2695,7 +2717,7 @@ Attributes
       :rtype: AnnotationEntry
 
 
-   .. py:method:: update_annotation_status(feature_group_id, feature_name, status, doc_id = None, feature_group_row_identifier = None)
+   .. py:method:: update_annotation_status(feature_group_id, feature_name, status, doc_id = None, feature_group_row_identifier = None, save_metadata = False)
 
       Update the status of an annotation entry.
 
@@ -2709,6 +2731,8 @@ Attributes
       :type doc_id: str
       :param feature_group_row_identifier: The key value of the feature group row the annotation is on (cast to string). Usually the feature group's primary / identifier key value. At least one of the doc_id or feature_group_row_identifier must be provided in order to identify the correct annotation.
       :type feature_group_row_identifier: str
+      :param save_metadata: If True, save the metadata for the annotation entry.
+      :type save_metadata: bool
 
       :returns: None
       :rtype: AnnotationEntry
@@ -3368,14 +3392,6 @@ Attributes
       :rtype: ExecuteFeatureGroupOperation
 
 
-   .. py:method:: get_execute_feature_group_operation_result_part_count(feature_group_operation_run_id)
-
-      Gets the number of parts in the result of the execution of fg operation
-
-      :param feature_group_operation_run_id: The unique ID associated with the execution.
-      :type feature_group_operation_run_id: str
-
-
    .. py:method:: update_feature_group(feature_group_id, description = None)
 
       Modify an existing Feature Group.
@@ -5405,7 +5421,7 @@ Attributes
       :type exclude_items: list
 
 
-   .. py:method:: get_chat_response(deployment_token, deployment_id, messages, search_results = None)
+   .. py:method:: get_chat_response(deployment_token, deployment_id, messages, search_results = None, chat_config = None)
 
       Return a chat response which continues the conversation based on the input messages and search results.
 
@@ -5417,6 +5433,8 @@ Attributes
       :type messages: list
       :param search_results: A list of chronologically ordered retrieved search results using the deployment. A retrieved search result is a dict with attributes:     msg_id (int): The corresponding message's index.      result (list): List of NlpSearchPrediction objects.
       :type search_results: list
+      :param chat_config: A dictionary specifiying the query chat config override
+      :type chat_config: dict
 
 
    .. py:method:: get_search_results(deployment_token, deployment_id, query_data)
@@ -6102,7 +6120,18 @@ Attributes
       :rtype: Pipeline
 
 
-   .. py:method:: describe_pipeline(pipeline_name)
+   .. py:method:: describe_pipeline(pipeline_id)
+
+      Describes a given pipeline.
+
+      :param pipeline_id: The ID of the pipeline to describe.
+      :type pipeline_id: str
+
+      :returns: An object describing a Pipeline
+      :rtype: Pipeline
+
+
+   .. py:method:: describe_pipeline_by_name(pipeline_name)
 
       Describes a given pipeline.
 
@@ -6113,12 +6142,12 @@ Attributes
       :rtype: Pipeline
 
 
-   .. py:method:: update_pipeline(pipeline_name, project_id = None, pipeline_variable_mappings = None, cron = None)
+   .. py:method:: update_pipeline(pipeline_id, project_id = None, pipeline_variable_mappings = None, cron = None)
 
       Updates a pipeline for executing multiple steps.
 
-      :param pipeline_name: The name of the pipeline, which should be unique to the organization.
-      :type pipeline_name: str
+      :param pipeline_id: The ID of the pipeline to update.
+      :type pipeline_id: str
       :param project_id: A unique string identifier for the pipeline.
       :type project_id: str
       :param pipeline_variable_mappings: List of Python function arguments for the pipeline.
@@ -6130,31 +6159,31 @@ Attributes
       :rtype: Pipeline
 
 
-   .. py:method:: delete_pipeline(pipeline_name)
+   .. py:method:: delete_pipeline(pipeline_id)
 
       Deletes a pipeline.
 
-      :param pipeline_name: The name of the pipeline to describe.
-      :type pipeline_name: str
+      :param pipeline_id: The ID of the pipeline to delete.
+      :type pipeline_id: str
 
 
-   .. py:method:: list_pipeline_versions(pipeline_name)
+   .. py:method:: list_pipeline_versions(pipeline_id)
 
       Lists the pipeline versions for a specified pipeline
 
-      :param pipeline_name: The name of the pipeline to describe.
-      :type pipeline_name: str
+      :param pipeline_id: The ID of the pipeline to list versions for.
+      :type pipeline_id: str
 
       :returns: A list of pipeline versions.
       :rtype: PipelineVersion
 
 
-   .. py:method:: run_pipeline(pipeline_name, pipeline_variable_mappings = None)
+   .. py:method:: run_pipeline(pipeline_id, pipeline_variable_mappings = None)
 
       Runs a specified pipeline with the arguments provided.
 
-      :param pipeline_name: The name of the pipeline to run.
-      :type pipeline_name: str
+      :param pipeline_id: The ID of the pipeline to run.
+      :type pipeline_id: str
       :param pipeline_variable_mappings: List of Python function arguments for the pipeline.
       :type pipeline_variable_mappings: list
 
@@ -6162,12 +6191,12 @@ Attributes
       :rtype: PipelineVersion
 
 
-   .. py:method:: create_pipeline_step(pipeline_name, step_name, function_name = None, source_code = None, step_input_mappings = None, output_variable_mappings = None, step_dependencies = None, package_requirements = None)
+   .. py:method:: create_pipeline_step(pipeline_id, step_name, function_name = None, source_code = None, step_input_mappings = None, output_variable_mappings = None, step_dependencies = None, package_requirements = None)
 
       Creates a step in a given pipeline.
 
-      :param pipeline_name: The name of the pipeline to run.
-      :type pipeline_name: str
+      :param pipeline_id: The ID of the pipeline to run.
+      :type pipeline_id: str
       :param step_name: The name of the step.
       :type step_name: str
       :param function_name: The name of the Python function.
@@ -6187,22 +6216,53 @@ Attributes
       :rtype: Pipeline
 
 
-   .. py:method:: delete_pipeline_step(pipeline_name, step_name)
+   .. py:method:: delete_pipeline_step(pipeline_step_id)
 
       Deletes a step from a pipeline.
 
-      :param pipeline_name: The name of the pipeline.
-      :type pipeline_name: str
+      :param pipeline_step_id: The ID of the pipeline step.
+      :type pipeline_step_id: str
+
+
+   .. py:method:: delete_pipeline_step_by_name(pipeline_id, step_name)
+
+      Deletes a step from a pipeline.
+
+      :param pipeline_id: The ID of the pipeline.
+      :type pipeline_id: str
       :param step_name: The name of the step.
       :type step_name: str
 
 
-   .. py:method:: update_pipeline_step(pipeline_name, step_name, function_name = None, source_code = None, step_input_mappings = None, output_variable_mappings = None, step_dependencies = None, package_requirements = None)
+   .. py:method:: update_pipeline_step(pipeline_step_id, function_name = None, source_code = None, step_input_mappings = None, output_variable_mappings = None, step_dependencies = None, package_requirements = None)
 
       Creates a step in a given pipeline.
 
-      :param pipeline_name: The name of the pipeline to run.
-      :type pipeline_name: str
+      :param pipeline_step_id: The ID of the pipeline_step to update.
+      :type pipeline_step_id: str
+      :param function_name: The name of the Python function.
+      :type function_name: str
+      :param source_code: Contents of a valid Python source code file. The source code should contain the transform feature group functions. A list of allowed imports and system libraries for each language is specified in the user functions documentation section.
+      :type source_code: str
+      :param step_input_mappings: List of Python function arguments.
+      :type step_input_mappings: list
+      :param output_variable_mappings: List of Python function ouputs.
+      :type output_variable_mappings: list
+      :param step_dependencies: List of step names this step depends on.
+      :type step_dependencies: list
+      :param package_requirements: List of package requirement strings. For example: ['numpy==1.2.3', 'pandas>=1.4.0'].
+      :type package_requirements: list
+
+      :returns: Object describing the pipeline.
+      :rtype: PipelineStep
+
+
+   .. py:method:: update_pipeline_step_by_name(pipeline_id, step_name, function_name = None, source_code = None, step_input_mappings = None, output_variable_mappings = None, step_dependencies = None, package_requirements = None)
+
+      Creates a step in a given pipeline.
+
+      :param pipeline_id: The ID of the pipeline to update the step for.
+      :type pipeline_id: str
       :param step_name: The name of the step.
       :type step_name: str
       :param function_name: The name of the Python function.
@@ -6537,19 +6597,6 @@ Attributes
       :rtype: ChatSession
 
 
-   .. py:method:: send_chat_message(chat_session_id, message)
-
-      Updates chat history with the response from a user message
-
-      :param chat_session_id: Unique ID of the chat session.
-      :type chat_session_id: str
-      :param message: Message you want to send to Abacus Chat
-      :type message: str
-
-      :returns: The chat session with Abacus Chat
-      :rtype: ChatSession
-
-
    .. py:method:: create_agent(project_id, function_source_code, agent_function_name, name = None, memory = None, package_requirements = None, description = None)
 
       Creates a new AI agent.