diff --git a/abacusai/__init__.py b/abacusai/__init__.py
index e7ef2933..d37a434b 100644
--- a/abacusai/__init__.py
+++ b/abacusai/__init__.py
@@ -17,6 +17,7 @@
from .api_endpoint import ApiEndpoint
from .api_key import ApiKey
from .app_user_group import AppUserGroup
+from .app_user_group_sign_in_token import AppUserGroupSignInToken
from .application_connector import ApplicationConnector
from .batch_prediction import BatchPrediction
from .batch_prediction_version import BatchPredictionVersion
@@ -224,4 +225,4 @@
from .workflow_node_template import WorkflowNodeTemplate
-__version__ = "1.4.20"
+__version__ = "1.4.21"
diff --git a/abacusai/api_class/enums.py b/abacusai/api_class/enums.py
index 23b4fb16..beb9e97d 100644
--- a/abacusai/api_class/enums.py
+++ b/abacusai/api_class/enums.py
@@ -408,7 +408,6 @@ class ApplicationConnectorType(ApiEnum):
GOOGLECALENDAR = 'GOOGLECALENDAR'
GOOGLESHEETS = 'GOOGLESHEETS'
GOOGLEDOCS = 'GOOGLEDOCS'
- ONEDRIVEUSER = 'ONEDRIVEUSER'
TEAMSSCRAPER = 'TEAMSSCRAPER'
GITHUBUSER = 'GITHUBUSER'
OKTASAML = 'OKTASAML'
@@ -484,6 +483,7 @@ class LLMName(ApiEnum):
ABACUS_DRACARYS = 'ABACUS_DRACARYS'
QWEN_2_5_32B = 'QWEN_2_5_32B'
GEMINI_1_5_FLASH = 'GEMINI_1_5_FLASH'
+ XAI_GROK = 'XAI_GROK'
class MonitorAlertType(ApiEnum):
diff --git a/abacusai/api_class/project.py b/abacusai/api_class/project.py
index c51be978..3a63e212 100644
--- a/abacusai/api_class/project.py
+++ b/abacusai/api_class/project.py
@@ -49,7 +49,7 @@ class ConstraintConfig(ApiClass):
Args:
constant (float): The constant value for the constraint.
operator (str): The operator for the constraint. Could be 'EQ', 'LE', 'GE'
- enforcement (str): The enforcement for the constraint. Could be 'HARD' or 'SOFT'. Default is 'HARD'
+ enforcement (str): The enforcement for the constraint. Could be 'HARD' or 'SOFT' or 'SKIP'. Default is 'HARD'
code (str): The code for the constraint.
penalty (float): The penalty for violating the constraint.
"""
diff --git a/abacusai/api_client_utils.py b/abacusai/api_client_utils.py
index 427c2810..418aebbe 100644
--- a/abacusai/api_client_utils.py
+++ b/abacusai/api_client_utils.py
@@ -438,8 +438,8 @@ def get_pandas_pages_df(cls, df, feature_group_version: str, doc_id_column: str,
chunk_size = 10 * 1024 * 1024
pages_df_with_config = None
- df_with_config = df[df[document_column].apply(
- lambda x: isinstance(x, dict) and cls.DOCUMENT_PROCESSING_CONFIG in x)]
+ df_with_config = df[df[document_column].apply(lambda x: isinstance(
+ x, dict) and x.get(cls.DOCUMENT_PROCESSING_CONFIG) is not None)]
df = df[~df[doc_id_column].isin(df_with_config[doc_id_column])]
if len(df_with_config) > 0:
diff --git a/abacusai/app_user_group_sign_in_token.py b/abacusai/app_user_group_sign_in_token.py
new file mode 100644
index 00000000..c873a8c5
--- /dev/null
+++ b/abacusai/app_user_group_sign_in_token.py
@@ -0,0 +1,33 @@
+from .return_class import AbstractApiClass
+
+
+class AppUserGroupSignInToken(AbstractApiClass):
+ """
+ User Group Sign In Token
+
+ Args:
+ client (ApiClient): An authenticated API Client instance
+ token (str): The token to sign in the user
+ """
+
+ def __init__(self, client, token=None):
+ super().__init__(client, None)
+ self.token = token
+ self.deprecated_keys = {}
+
+ def __repr__(self):
+ repr_dict = {f'token': repr(self.token)}
+ class_name = "AppUserGroupSignInToken"
+ 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 = {'token': self.token}
+ return {key: value for key, value in resp.items() if value is not None and key not in self.deprecated_keys}
diff --git a/abacusai/batch_prediction.py b/abacusai/batch_prediction.py
index 34609d8d..e5928ffb 100644
--- a/abacusai/batch_prediction.py
+++ b/abacusai/batch_prediction.py
@@ -84,7 +84,7 @@ def __init__(self, client, batchPredictionId=None, createdAt=None, name=None, de
BatchPredictionArgs, globalPredictionArgs)
self.batch_prediction_args = client._build_class(getattr(
api_class, batchPredictionArgsType, BatchPredictionArgs) if batchPredictionArgsType else BatchPredictionArgs, batchPredictionArgs)
- self.deprecated_keys = {'global_prediction_args', 'explanations'}
+ self.deprecated_keys = {'explanations', 'global_prediction_args'}
def __repr__(self):
repr_dict = {f'batch_prediction_id': repr(self.batch_prediction_id), f'created_at': repr(self.created_at), f'name': repr(self.name), f'deployment_id': repr(self.deployment_id), f'file_connector_output_location': repr(self.file_connector_output_location), f'database_connector_id': repr(self.database_connector_id), f'database_output_configuration': repr(self.database_output_configuration), f'file_output_format': repr(self.file_output_format), f'connector_type': repr(self.connector_type), f'legacy_input_location': repr(self.legacy_input_location), f'output_feature_group_id': repr(self.output_feature_group_id), f'feature_group_table_name': repr(self.feature_group_table_name), f'output_feature_group_table_name': repr(self.output_feature_group_table_name), f'summary_feature_group_table_name': repr(self.summary_feature_group_table_name), f'csv_input_prefix': repr(
diff --git a/abacusai/batch_prediction_version.py b/abacusai/batch_prediction_version.py
index 5e53ea27..f89c47a1 100644
--- a/abacusai/batch_prediction_version.py
+++ b/abacusai/batch_prediction_version.py
@@ -100,7 +100,7 @@ def __init__(self, client, batchPredictionVersion=None, batchPredictionId=None,
BatchPredictionArgs, globalPredictionArgs)
self.batch_prediction_args = client._build_class(getattr(
api_class, batchPredictionArgsType, BatchPredictionArgs) if batchPredictionArgsType else BatchPredictionArgs, batchPredictionArgs)
- self.deprecated_keys = {'global_prediction_args', 'explanations'}
+ self.deprecated_keys = {'explanations', 'global_prediction_args'}
def __repr__(self):
repr_dict = {f'batch_prediction_version': repr(self.batch_prediction_version), f'batch_prediction_id': repr(self.batch_prediction_id), f'status': repr(self.status), f'drift_monitor_status': repr(self.drift_monitor_status), f'deployment_id': repr(self.deployment_id), f'model_id': repr(self.model_id), f'model_version': repr(self.model_version), f'predictions_started_at': repr(self.predictions_started_at), f'predictions_completed_at': repr(self.predictions_completed_at), f'database_output_error': repr(self.database_output_error), f'total_predictions': repr(self.total_predictions), f'failed_predictions': repr(self.failed_predictions), f'database_connector_id': repr(self.database_connector_id), f'database_output_configuration': repr(self.database_output_configuration), f'file_connector_output_location': repr(self.file_connector_output_location), f'file_output_format': repr(self.file_output_format), f'connector_type': repr(self.connector_type), f'legacy_input_location': repr(self.legacy_input_location), f'error': repr(self.error), f'drift_monitor_error': repr(self.drift_monitor_error), f'monitor_warnings': repr(self.monitor_warnings), f'csv_input_prefix': repr(
diff --git a/abacusai/client.py b/abacusai/client.py
index 7da9dd7e..2264d3c4 100644
--- a/abacusai/client.py
+++ b/abacusai/client.py
@@ -58,6 +58,7 @@
from .api_endpoint import ApiEndpoint
from .api_key import ApiKey
from .app_user_group import AppUserGroup
+from .app_user_group_sign_in_token import AppUserGroupSignInToken
from .application_connector import ApplicationConnector
from .batch_prediction import BatchPrediction
from .batch_prediction_version import BatchPredictionVersion
@@ -640,7 +641,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.20'
+ client_version = '1.4.21'
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
@@ -832,7 +833,7 @@ def _call_api(
def _proxy_request(self, name: str, method: str = 'POST', query_params: dict = None, body: dict = None, data: dict = None, files=None, parse_type=None, is_sync: bool = False, streamable_response: bool = False):
headers = {'APIKEY': self.api_key}
- deployment_id = os.getenv('ABACUS_DEPLOYMENT_ID')
+ deployment_id = os.getenv('ABACUS_EXEC_SERVICE_DEPLOYMENT_ID')
if deployment_id:
query_params = {**(query_params or {}),
'environmentDeploymentId': deployment_id}
@@ -2585,6 +2586,18 @@ def list_organization_secrets(self) -> List[OrganizationSecret]:
list[OrganizationSecret]: list of secrets belonging to the organization."""
return self._call_api('listOrganizationSecrets', 'GET', query_params={}, parse_type=OrganizationSecret)
+ def get_app_user_group_sign_in_token(self, user_group_id: str, email: str, name: str) -> AppUserGroupSignInToken:
+ """Get a token for a user group user to sign in.
+
+ Args:
+ user_group_id (str): The ID of the user group.
+ email (str): The email of the user.
+ name (str): The name of the user.
+
+ Returns:
+ AppUserGroupSignInToken: The token to sign in the user"""
+ return self._call_api('getAppUserGroupSignInToken', 'GET', query_params={'userGroupId': user_group_id, 'email': email, 'name': name}, parse_type=AppUserGroupSignInToken)
+
def query_feature_group_code_generator(self, query: str, language: str, project_id: str = None) -> LlmResponse:
"""Send a query to the feature group code generator tool to generate code for the query.
@@ -4420,7 +4433,8 @@ def set_scoped_cache_value(self, key: str, value: str, expiration_time: int = 21
Raises:
InvalidParameterError: If key, value or expiration_time is invalid.
"""
- scope = self.cache_scope or os.getenv('ABACUS_DEPLOYMENT_ID')
+ scope = self.cache_scope or os.getenv(
+ 'ABACUS_EXEC_SERVICE_DEPLOYMENT_ID')
if scope:
return self._proxy_request('_setScopedCacheValue', 'POST', body={'key': key, 'value': value, 'scope': scope, 'expirationTime': expiration_time}, is_sync=True)
else:
@@ -4440,7 +4454,8 @@ def get_scoped_cache_value(self, key: str):
Raises:
Generic404Error: if the key doesn't exist.
"""
- scope = self.cache_scope or os.getenv('ABACUS_DEPLOYMENT_ID')
+ scope = self.cache_scope or os.getenv(
+ 'ABACUS_EXEC_SERVICE_DEPLOYMENT_ID')
if scope:
return self._proxy_request('_getScopedCacheValue', 'GET', query_params={'key': key, 'scope': scope}, is_sync=True)
else:
@@ -4460,7 +4475,8 @@ def delete_scoped_cache_key(self, key: str):
Returns:
None
"""
- scope = self.cache_scope or os.getenv('ABACUS_DEPLOYMENT_ID')
+ scope = self.cache_scope or os.getenv(
+ 'ABACUS_EXEC_SERVICE_DEPLOYMENT_ID')
if scope:
return self._proxy_request('_deleteScopedCacheKey', 'POST', query_params={'key': key, 'scope': scope}, is_sync=True)
else:
diff --git a/abacusai/code_edit_response.py b/abacusai/code_edit_response.py
index d1aea4b3..25a57139 100644
--- a/abacusai/code_edit_response.py
+++ b/abacusai/code_edit_response.py
@@ -8,18 +8,15 @@ class CodeEditResponse(AbstractApiClass):
Args:
client (ApiClient): An authenticated API Client instance
codeChanges (list): The code changes to be applied.
- deploymentConversationId (str): The unique identifier of the deployment conversation.
"""
- def __init__(self, client, codeChanges=None, deploymentConversationId=None):
+ def __init__(self, client, codeChanges=None):
super().__init__(client, None)
self.code_changes = codeChanges
- self.deployment_conversation_id = deploymentConversationId
self.deprecated_keys = {}
def __repr__(self):
- repr_dict = {f'code_changes': repr(
- self.code_changes), f'deployment_conversation_id': repr(self.deployment_conversation_id)}
+ repr_dict = {f'code_changes': repr(self.code_changes)}
class_name = "CodeEditResponse"
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])
@@ -32,6 +29,5 @@ def to_dict(self):
Returns:
dict: The dict value representation of the class parameters
"""
- resp = {'code_changes': self.code_changes,
- 'deployment_conversation_id': self.deployment_conversation_id}
+ resp = {'code_changes': self.code_changes}
return {key: value for key, value in resp.items() if value is not None and key not in self.deprecated_keys}
diff --git a/abacusai/feature_group.py b/abacusai/feature_group.py
index db138b00..c88424e7 100644
--- a/abacusai/feature_group.py
+++ b/abacusai/feature_group.py
@@ -1199,13 +1199,13 @@ def load_as_pandas(self):
latest_version = self.materialize().latest_feature_group_version
return latest_version.load_as_pandas()
- def load_as_pandas_documents(self, doc_id_column: str, document_column: str):
+ def load_as_pandas_documents(self, doc_id_column: str = 'doc_id', document_column: str = 'page_infos'):
"""
Loads a feature group with documents data into a pandas dataframe.
Args:
- doc_id_feature (str): The name of the feature / column containing the document ID.
- document_feature (str): The name of the feature / column which either contains the document data itself or page infos with path to remotely stored documents. This column will be replaced with the extracted document data.
+ doc_id_column (str): The name of the feature / column containing the document ID.
+ document_column (str): The name of the feature / column which either contains the document data itself or page infos with path to remotely stored documents. This column will be replaced with the extracted document data.
Returns:
DataFrame: A pandas dataframe containing the extracted document data.
diff --git a/abacusai/feature_group_version.py b/abacusai/feature_group_version.py
index 8418085f..bb29a3f4 100644
--- a/abacusai/feature_group_version.py
+++ b/abacusai/feature_group_version.py
@@ -277,13 +277,13 @@ def load_as_pandas(self, max_workers=10):
'_getFeatureGroupVersionParts', 'GET', query_params={'featureGroupVersion': self.id}, retry_500=True)
return load_as_pandas_from_avro_files(file_parts, self._download_avro_file, max_workers=max_workers)
- def load_as_pandas_documents(self, doc_id_column: str, document_column: str, max_workers=10):
+ def load_as_pandas_documents(self, doc_id_column: str = 'doc_id', document_column: str = 'page_infos', max_workers=10):
"""
Loads a feature group with documents data into a pandas dataframe.
Args:
- doc_id_feature (str): The name of the feature / column containing the document ID.
- document_feature (str): The name of the feature / column which either contains the document data itself or page infos with path to remotely stored documents. This column will be replaced with the extracted document data.
+ doc_id_column (str): The name of the feature / column containing the document ID.
+ document_column (str): The name of the feature / column which either contains the document data itself or page infos with path to remotely stored documents. This column will be replaced with the extracted document data.
max_workers (int): The number of threads.
Returns:
diff --git a/abacusai/video_gen_settings.py b/abacusai/video_gen_settings.py
index 9dfe4287..2275fd8a 100644
--- a/abacusai/video_gen_settings.py
+++ b/abacusai/video_gen_settings.py
@@ -7,35 +7,19 @@ class VideoGenSettings(AbstractApiClass):
Args:
client (ApiClient): An authenticated API Client instance
- prompt (dict): The prompt for the video.
- negativePrompt (dict): The negative prompt for the video.
- cfgScale (dict): The flexibility scale for video generation.
- mode (dict): The video generation mode (standard or professional).
- aspectRatio (dict): The aspect ratio of the video in seconds.
- duration (dict): The duration of the video.
- loop (dict): Whether the video should loop.
- startFrame (dict): The start frame of the video.
- endFrame (dict): The end frame of the video.
- rewritePrompt (dict): Whether to rewrite the prompt.
+ model (dict): The model settings.
+ settings (dict): The settings for each model.
"""
- def __init__(self, client, prompt=None, negativePrompt=None, cfgScale=None, mode=None, aspectRatio=None, duration=None, loop=None, startFrame=None, endFrame=None, rewritePrompt=None):
+ def __init__(self, client, model=None, settings=None):
super().__init__(client, None)
- self.prompt = prompt
- self.negative_prompt = negativePrompt
- self.cfg_scale = cfgScale
- self.mode = mode
- self.aspect_ratio = aspectRatio
- self.duration = duration
- self.loop = loop
- self.start_frame = startFrame
- self.end_frame = endFrame
- self.rewrite_prompt = rewritePrompt
+ self.model = model
+ self.settings = settings
self.deprecated_keys = {}
def __repr__(self):
- repr_dict = {f'prompt': repr(self.prompt), f'negative_prompt': repr(self.negative_prompt), f'cfg_scale': repr(self.cfg_scale), f'mode': repr(self.mode), f'aspect_ratio': repr(
- self.aspect_ratio), f'duration': repr(self.duration), f'loop': repr(self.loop), f'start_frame': repr(self.start_frame), f'end_frame': repr(self.end_frame), f'rewrite_prompt': repr(self.rewrite_prompt)}
+ repr_dict = {f'model': repr(self.model),
+ f'settings': repr(self.settings)}
class_name = "VideoGenSettings"
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])
@@ -48,6 +32,5 @@ def to_dict(self):
Returns:
dict: The dict value representation of the class parameters
"""
- resp = {'prompt': self.prompt, 'negative_prompt': self.negative_prompt, 'cfg_scale': self.cfg_scale, 'mode': self.mode, 'aspect_ratio': self.aspect_ratio,
- 'duration': self.duration, 'loop': self.loop, 'start_frame': self.start_frame, 'end_frame': self.end_frame, 'rewrite_prompt': self.rewrite_prompt}
+ resp = {'model': self.model, 'settings': self.settings}
return {key: value for key, value in resp.items() if value is not None and key not in self.deprecated_keys}
diff --git a/docs/_sources/autoapi/abacusai/api_class/enums/index.rst.txt b/docs/_sources/autoapi/abacusai/api_class/enums/index.rst.txt
index ee8eb1ea..d559b2da 100644
--- a/docs/_sources/autoapi/abacusai/api_class/enums/index.rst.txt
+++ b/docs/_sources/autoapi/abacusai/api_class/enums/index.rst.txt
@@ -1769,11 +1769,6 @@ Module Contents
- .. py:attribute:: ONEDRIVEUSER
- :value: 'ONEDRIVEUSER'
-
-
-
.. py:attribute:: TEAMSSCRAPER
:value: 'TEAMSSCRAPER'
@@ -2119,6 +2114,11 @@ Module Contents
+ .. py:attribute:: XAI_GROK
+ :value: 'XAI_GROK'
+
+
+
.. py:class:: MonitorAlertType
Bases: :py:obj:`ApiEnum`
diff --git a/docs/_sources/autoapi/abacusai/api_class/index.rst.txt b/docs/_sources/autoapi/abacusai/api_class/index.rst.txt
index 4f4f8d0a..d9170b36 100644
--- a/docs/_sources/autoapi/abacusai/api_class/index.rst.txt
+++ b/docs/_sources/autoapi/abacusai/api_class/index.rst.txt
@@ -4743,11 +4743,6 @@ Package Contents
- .. py:attribute:: ONEDRIVEUSER
- :value: 'ONEDRIVEUSER'
-
-
-
.. py:attribute:: TEAMSSCRAPER
:value: 'TEAMSSCRAPER'
@@ -5093,6 +5088,11 @@ Package Contents
+ .. py:attribute:: XAI_GROK
+ :value: 'XAI_GROK'
+
+
+
.. py:class:: MonitorAlertType
Bases: :py:obj:`ApiEnum`
@@ -9365,7 +9365,7 @@ Package Contents
:type constant: float
:param operator: The operator for the constraint. Could be 'EQ', 'LE', 'GE'
:type operator: str
- :param enforcement: The enforcement for the constraint. Could be 'HARD' or 'SOFT'. Default is 'HARD'
+ :param enforcement: The enforcement for the constraint. Could be 'HARD' or 'SOFT' or 'SKIP'. Default is 'HARD'
:type enforcement: str
:param code: The code for the constraint.
:type code: str
diff --git a/docs/_sources/autoapi/abacusai/api_class/project/index.rst.txt b/docs/_sources/autoapi/abacusai/api_class/project/index.rst.txt
index aac5af11..250b1424 100644
--- a/docs/_sources/autoapi/abacusai/api_class/project/index.rst.txt
+++ b/docs/_sources/autoapi/abacusai/api_class/project/index.rst.txt
@@ -91,7 +91,7 @@ Module Contents
:type constant: float
:param operator: The operator for the constraint. Could be 'EQ', 'LE', 'GE'
:type operator: str
- :param enforcement: The enforcement for the constraint. Could be 'HARD' or 'SOFT'. Default is 'HARD'
+ :param enforcement: The enforcement for the constraint. Could be 'HARD' or 'SOFT' or 'SKIP'. Default is 'HARD'
:type enforcement: str
:param code: The code for the constraint.
:type code: str
diff --git a/docs/_sources/autoapi/abacusai/app_user_group_sign_in_token/index.rst.txt b/docs/_sources/autoapi/abacusai/app_user_group_sign_in_token/index.rst.txt
new file mode 100644
index 00000000..3b257460
--- /dev/null
+++ b/docs/_sources/autoapi/abacusai/app_user_group_sign_in_token/index.rst.txt
@@ -0,0 +1,48 @@
+abacusai.app_user_group_sign_in_token
+=====================================
+
+.. py:module:: abacusai.app_user_group_sign_in_token
+
+
+Classes
+-------
+
+.. autoapisummary::
+
+ abacusai.app_user_group_sign_in_token.AppUserGroupSignInToken
+
+
+Module Contents
+---------------
+
+.. py:class:: AppUserGroupSignInToken(client, token=None)
+
+ Bases: :py:obj:`abacusai.return_class.AbstractApiClass`
+
+
+ User Group Sign In Token
+
+ :param client: An authenticated API Client instance
+ :type client: ApiClient
+ :param token: The token to sign in the user
+ :type token: str
+
+
+ .. py:attribute:: token
+
+
+ .. py:attribute:: deprecated_keys
+
+
+ .. py:method:: __repr__()
+
+
+ .. py:method:: to_dict()
+
+ Get a dict representation of the parameters in this class
+
+ :returns: The dict value representation of the class parameters
+ :rtype: dict
+
+
+
diff --git a/docs/_sources/autoapi/abacusai/client/index.rst.txt b/docs/_sources/autoapi/abacusai/client/index.rst.txt
index acb7909b..a12b8a58 100644
--- a/docs/_sources/autoapi/abacusai/client/index.rst.txt
+++ b/docs/_sources/autoapi/abacusai/client/index.rst.txt
@@ -564,7 +564,7 @@ Module Contents
.. py:attribute:: client_version
- :value: '1.4.20'
+ :value: '1.4.21'
@@ -2698,6 +2698,22 @@ Module Contents
+ .. py:method:: get_app_user_group_sign_in_token(user_group_id, email, name)
+
+ Get a token for a user group user to sign in.
+
+ :param user_group_id: The ID of the user group.
+ :type user_group_id: str
+ :param email: The email of the user.
+ :type email: str
+ :param name: The name of the user.
+ :type name: str
+
+ :returns: The token to sign in the user
+ :rtype: AppUserGroupSignInToken
+
+
+
.. py:method:: query_feature_group_code_generator(query, language, project_id = None)
Send a query to the feature group code generator tool to generate code for the query.
diff --git a/docs/_sources/autoapi/abacusai/code_edit_response/index.rst.txt b/docs/_sources/autoapi/abacusai/code_edit_response/index.rst.txt
index 798f1590..ac11cd5c 100644
--- a/docs/_sources/autoapi/abacusai/code_edit_response/index.rst.txt
+++ b/docs/_sources/autoapi/abacusai/code_edit_response/index.rst.txt
@@ -15,7 +15,7 @@ Classes
Module Contents
---------------
-.. py:class:: CodeEditResponse(client, codeChanges=None, deploymentConversationId=None)
+.. py:class:: CodeEditResponse(client, codeChanges=None)
Bases: :py:obj:`abacusai.return_class.AbstractApiClass`
@@ -26,16 +26,11 @@ Module Contents
:type client: ApiClient
:param codeChanges: The code changes to be applied.
:type codeChanges: list
- :param deploymentConversationId: The unique identifier of the deployment conversation.
- :type deploymentConversationId: str
.. py:attribute:: code_changes
- .. py:attribute:: deployment_conversation_id
-
-
.. py:attribute:: deprecated_keys
diff --git a/docs/_sources/autoapi/abacusai/feature_group/index.rst.txt b/docs/_sources/autoapi/abacusai/feature_group/index.rst.txt
index 46963cb5..a5238dee 100644
--- a/docs/_sources/autoapi/abacusai/feature_group/index.rst.txt
+++ b/docs/_sources/autoapi/abacusai/feature_group/index.rst.txt
@@ -1413,14 +1413,14 @@ Module Contents
- .. py:method:: load_as_pandas_documents(doc_id_column, document_column)
+ .. py:method:: load_as_pandas_documents(doc_id_column = 'doc_id', document_column = 'page_infos')
Loads a feature group with documents data into a pandas dataframe.
- :param doc_id_feature: The name of the feature / column containing the document ID.
- :type doc_id_feature: str
- :param document_feature: The name of the feature / column which either contains the document data itself or page infos with path to remotely stored documents. This column will be replaced with the extracted document data.
- :type document_feature: str
+ :param doc_id_column: The name of the feature / column containing the document ID.
+ :type doc_id_column: str
+ :param document_column: The name of the feature / column which either contains the document data itself or page infos with path to remotely stored documents. This column will be replaced with the extracted document data.
+ :type document_column: str
:returns: A pandas dataframe containing the extracted document data.
:rtype: DataFrame
diff --git a/docs/_sources/autoapi/abacusai/feature_group_version/index.rst.txt b/docs/_sources/autoapi/abacusai/feature_group_version/index.rst.txt
index a55563ea..50b91849 100644
--- a/docs/_sources/autoapi/abacusai/feature_group_version/index.rst.txt
+++ b/docs/_sources/autoapi/abacusai/feature_group_version/index.rst.txt
@@ -332,14 +332,14 @@ Module Contents
- .. py:method:: load_as_pandas_documents(doc_id_column, document_column, max_workers=10)
+ .. py:method:: load_as_pandas_documents(doc_id_column = 'doc_id', document_column = 'page_infos', max_workers=10)
Loads a feature group with documents data into a pandas dataframe.
- :param doc_id_feature: The name of the feature / column containing the document ID.
- :type doc_id_feature: str
- :param document_feature: The name of the feature / column which either contains the document data itself or page infos with path to remotely stored documents. This column will be replaced with the extracted document data.
- :type document_feature: str
+ :param doc_id_column: The name of the feature / column containing the document ID.
+ :type doc_id_column: str
+ :param document_column: The name of the feature / column which either contains the document data itself or page infos with path to remotely stored documents. This column will be replaced with the extracted document data.
+ :type document_column: str
:param max_workers: The number of threads.
:type max_workers: int
diff --git a/docs/_sources/autoapi/abacusai/index.rst.txt b/docs/_sources/autoapi/abacusai/index.rst.txt
index 001f942d..5d7ee185 100644
--- a/docs/_sources/autoapi/abacusai/index.rst.txt
+++ b/docs/_sources/autoapi/abacusai/index.rst.txt
@@ -30,6 +30,7 @@ Submodules
/autoapi/abacusai/api_endpoint/index
/autoapi/abacusai/api_key/index
/autoapi/abacusai/app_user_group/index
+ /autoapi/abacusai/app_user_group_sign_in_token/index
/autoapi/abacusai/application_connector/index
/autoapi/abacusai/batch_prediction/index
/autoapi/abacusai/batch_prediction_version/index
@@ -492,6 +493,7 @@ Classes
abacusai.ApiEndpoint
abacusai.ApiKey
abacusai.AppUserGroup
+ abacusai.AppUserGroupSignInToken
abacusai.ApplicationConnector
abacusai.BatchPrediction
abacusai.BatchPredictionVersion
@@ -5688,11 +5690,6 @@ Package Contents
- .. py:attribute:: ONEDRIVEUSER
- :value: 'ONEDRIVEUSER'
-
-
-
.. py:attribute:: TEAMSSCRAPER
:value: 'TEAMSSCRAPER'
@@ -5993,6 +5990,11 @@ Package Contents
+ .. py:attribute:: XAI_GROK
+ :value: 'XAI_GROK'
+
+
+
.. py:class:: MonitorAlertType
Bases: :py:obj:`ApiEnum`
@@ -9629,7 +9631,7 @@ Package Contents
:type constant: float
:param operator: The operator for the constraint. Could be 'EQ', 'LE', 'GE'
:type operator: str
- :param enforcement: The enforcement for the constraint. Could be 'HARD' or 'SOFT'. Default is 'HARD'
+ :param enforcement: The enforcement for the constraint. Could be 'HARD' or 'SOFT' or 'SKIP'. Default is 'HARD'
:type enforcement: str
:param code: The code for the constraint.
:type code: str
@@ -10348,6 +10350,37 @@ Package Contents
+.. py:class:: AppUserGroupSignInToken(client, token=None)
+
+ Bases: :py:obj:`abacusai.return_class.AbstractApiClass`
+
+
+ User Group Sign In Token
+
+ :param client: An authenticated API Client instance
+ :type client: ApiClient
+ :param token: The token to sign in the user
+ :type token: str
+
+
+ .. py:attribute:: token
+
+
+ .. py:attribute:: deprecated_keys
+
+
+ .. py:method:: __repr__()
+
+
+ .. py:method:: to_dict()
+
+ Get a dict representation of the parameters in this class
+
+ :returns: The dict value representation of the class parameters
+ :rtype: dict
+
+
+
.. py:class:: ApplicationConnector(client, applicationConnectorId=None, service=None, name=None, createdAt=None, status=None, auth=None)
Bases: :py:obj:`abacusai.return_class.AbstractApiClass`
@@ -20872,6 +20905,22 @@ Package Contents
+ .. py:method:: get_app_user_group_sign_in_token(user_group_id, email, name)
+
+ Get a token for a user group user to sign in.
+
+ :param user_group_id: The ID of the user group.
+ :type user_group_id: str
+ :param email: The email of the user.
+ :type email: str
+ :param name: The name of the user.
+ :type name: str
+
+ :returns: The token to sign in the user
+ :rtype: AppUserGroupSignInToken
+
+
+
.. py:method:: query_feature_group_code_generator(query, language, project_id = None)
Send a query to the feature group code generator tool to generate code for the query.
@@ -21220,7 +21269,7 @@ Package Contents
-.. py:class:: CodeEditResponse(client, codeChanges=None, deploymentConversationId=None)
+.. py:class:: CodeEditResponse(client, codeChanges=None)
Bases: :py:obj:`abacusai.return_class.AbstractApiClass`
@@ -21231,16 +21280,11 @@ Package Contents
:type client: ApiClient
:param codeChanges: The code changes to be applied.
:type codeChanges: list
- :param deploymentConversationId: The unique identifier of the deployment conversation.
- :type deploymentConversationId: str
.. py:attribute:: code_changes
- .. py:attribute:: deployment_conversation_id
-
-
.. py:attribute:: deprecated_keys
@@ -27607,14 +27651,14 @@ Package Contents
- .. py:method:: load_as_pandas_documents(doc_id_column, document_column)
+ .. py:method:: load_as_pandas_documents(doc_id_column = 'doc_id', document_column = 'page_infos')
Loads a feature group with documents data into a pandas dataframe.
- :param doc_id_feature: The name of the feature / column containing the document ID.
- :type doc_id_feature: str
- :param document_feature: The name of the feature / column which either contains the document data itself or page infos with path to remotely stored documents. This column will be replaced with the extracted document data.
- :type document_feature: str
+ :param doc_id_column: The name of the feature / column containing the document ID.
+ :type doc_id_column: str
+ :param document_column: The name of the feature / column which either contains the document data itself or page infos with path to remotely stored documents. This column will be replaced with the extracted document data.
+ :type document_column: str
:returns: A pandas dataframe containing the extracted document data.
:rtype: DataFrame
@@ -28831,14 +28875,14 @@ Package Contents
- .. py:method:: load_as_pandas_documents(doc_id_column, document_column, max_workers=10)
+ .. py:method:: load_as_pandas_documents(doc_id_column = 'doc_id', document_column = 'page_infos', max_workers=10)
Loads a feature group with documents data into a pandas dataframe.
- :param doc_id_feature: The name of the feature / column containing the document ID.
- :type doc_id_feature: str
- :param document_feature: The name of the feature / column which either contains the document data itself or page infos with path to remotely stored documents. This column will be replaced with the extracted document data.
- :type document_feature: str
+ :param doc_id_column: The name of the feature / column containing the document ID.
+ :type doc_id_column: str
+ :param document_column: The name of the feature / column which either contains the document data itself or page infos with path to remotely stored documents. This column will be replaced with the extracted document data.
+ :type document_column: str
:param max_workers: The number of threads.
:type max_workers: int
@@ -39943,7 +39987,7 @@ Package Contents
-.. py:class:: VideoGenSettings(client, prompt=None, negativePrompt=None, cfgScale=None, mode=None, aspectRatio=None, duration=None, loop=None, startFrame=None, endFrame=None, rewritePrompt=None)
+.. py:class:: VideoGenSettings(client, model=None, settings=None)
Bases: :py:obj:`abacusai.return_class.AbstractApiClass`
@@ -39952,56 +39996,16 @@ Package Contents
:param client: An authenticated API Client instance
:type client: ApiClient
- :param prompt: The prompt for the video.
- :type prompt: dict
- :param negativePrompt: The negative prompt for the video.
- :type negativePrompt: dict
- :param cfgScale: The flexibility scale for video generation.
- :type cfgScale: dict
- :param mode: The video generation mode (standard or professional).
- :type mode: dict
- :param aspectRatio: The aspect ratio of the video in seconds.
- :type aspectRatio: dict
- :param duration: The duration of the video.
- :type duration: dict
- :param loop: Whether the video should loop.
- :type loop: dict
- :param startFrame: The start frame of the video.
- :type startFrame: dict
- :param endFrame: The end frame of the video.
- :type endFrame: dict
- :param rewritePrompt: Whether to rewrite the prompt.
- :type rewritePrompt: dict
-
-
- .. py:attribute:: prompt
-
-
- .. py:attribute:: negative_prompt
-
-
- .. py:attribute:: cfg_scale
-
-
- .. py:attribute:: mode
-
-
- .. py:attribute:: aspect_ratio
-
-
- .. py:attribute:: duration
-
-
- .. py:attribute:: loop
-
-
- .. py:attribute:: start_frame
+ :param model: The model settings.
+ :type model: dict
+ :param settings: The settings for each model.
+ :type settings: dict
- .. py:attribute:: end_frame
+ .. py:attribute:: model
- .. py:attribute:: rewrite_prompt
+ .. py:attribute:: settings
.. py:attribute:: deprecated_keys
@@ -40343,6 +40347,6 @@ Package Contents
.. py:data:: __version__
- :value: '1.4.20'
+ :value: '1.4.21'
diff --git a/docs/_sources/autoapi/abacusai/video_gen_settings/index.rst.txt b/docs/_sources/autoapi/abacusai/video_gen_settings/index.rst.txt
index 562bc70a..97e09c0e 100644
--- a/docs/_sources/autoapi/abacusai/video_gen_settings/index.rst.txt
+++ b/docs/_sources/autoapi/abacusai/video_gen_settings/index.rst.txt
@@ -15,7 +15,7 @@ Classes
Module Contents
---------------
-.. py:class:: VideoGenSettings(client, prompt=None, negativePrompt=None, cfgScale=None, mode=None, aspectRatio=None, duration=None, loop=None, startFrame=None, endFrame=None, rewritePrompt=None)
+.. py:class:: VideoGenSettings(client, model=None, settings=None)
Bases: :py:obj:`abacusai.return_class.AbstractApiClass`
@@ -24,56 +24,16 @@ Module Contents
:param client: An authenticated API Client instance
:type client: ApiClient
- :param prompt: The prompt for the video.
- :type prompt: dict
- :param negativePrompt: The negative prompt for the video.
- :type negativePrompt: dict
- :param cfgScale: The flexibility scale for video generation.
- :type cfgScale: dict
- :param mode: The video generation mode (standard or professional).
- :type mode: dict
- :param aspectRatio: The aspect ratio of the video in seconds.
- :type aspectRatio: dict
- :param duration: The duration of the video.
- :type duration: dict
- :param loop: Whether the video should loop.
- :type loop: dict
- :param startFrame: The start frame of the video.
- :type startFrame: dict
- :param endFrame: The end frame of the video.
- :type endFrame: dict
- :param rewritePrompt: Whether to rewrite the prompt.
- :type rewritePrompt: dict
+ :param model: The model settings.
+ :type model: dict
+ :param settings: The settings for each model.
+ :type settings: dict
- .. py:attribute:: prompt
+ .. py:attribute:: model
- .. py:attribute:: negative_prompt
-
-
- .. py:attribute:: cfg_scale
-
-
- .. py:attribute:: mode
-
-
- .. py:attribute:: aspect_ratio
-
-
- .. py:attribute:: duration
-
-
- .. py:attribute:: loop
-
-
- .. py:attribute:: start_frame
-
-
- .. py:attribute:: end_frame
-
-
- .. py:attribute:: rewrite_prompt
+ .. py:attribute:: settings
.. py:attribute:: deprecated_keys
diff --git a/docs/autoapi/abacusai/abacus_api/index.html b/docs/autoapi/abacusai/abacus_api/index.html
index 49204eed..c9d810dd 100644
--- a/docs/autoapi/abacusai/abacus_api/index.html
+++ b/docs/autoapi/abacusai/abacus_api/index.html
@@ -70,6 +70,7 @@
abacusai.api_endpoint
abacusai.api_key
abacusai.app_user_group
+abacusai.app_user_group_sign_in_token
abacusai.application_connector
abacusai.batch_prediction
abacusai.batch_prediction_version
diff --git a/docs/autoapi/abacusai/address/index.html b/docs/autoapi/abacusai/address/index.html
index 66c40e4f..3146c369 100644
--- a/docs/autoapi/abacusai/address/index.html
+++ b/docs/autoapi/abacusai/address/index.html
@@ -70,6 +70,7 @@
abacusai.api_endpoint
abacusai.api_key
abacusai.app_user_group
+abacusai.app_user_group_sign_in_token
abacusai.application_connector
abacusai.batch_prediction
abacusai.batch_prediction_version
diff --git a/docs/autoapi/abacusai/agent/index.html b/docs/autoapi/abacusai/agent/index.html
index af2ac04f..98d0939c 100644
--- a/docs/autoapi/abacusai/agent/index.html
+++ b/docs/autoapi/abacusai/agent/index.html
@@ -70,6 +70,7 @@
abacusai.api_endpoint
abacusai.api_key
abacusai.app_user_group
+abacusai.app_user_group_sign_in_token
abacusai.application_connector
abacusai.batch_prediction
abacusai.batch_prediction_version
diff --git a/docs/autoapi/abacusai/agent_chat_message/index.html b/docs/autoapi/abacusai/agent_chat_message/index.html
index 9a650867..7f53e832 100644
--- a/docs/autoapi/abacusai/agent_chat_message/index.html
+++ b/docs/autoapi/abacusai/agent_chat_message/index.html
@@ -70,6 +70,7 @@
abacusai.api_endpoint
abacusai.api_key
abacusai.app_user_group
+abacusai.app_user_group_sign_in_token
abacusai.application_connector
abacusai.batch_prediction
abacusai.batch_prediction_version
diff --git a/docs/autoapi/abacusai/agent_conversation/index.html b/docs/autoapi/abacusai/agent_conversation/index.html
index 32e176bf..6ef1e417 100644
--- a/docs/autoapi/abacusai/agent_conversation/index.html
+++ b/docs/autoapi/abacusai/agent_conversation/index.html
@@ -70,6 +70,7 @@
abacusai.api_endpoint
abacusai.api_key
abacusai.app_user_group
+abacusai.app_user_group_sign_in_token
abacusai.application_connector
abacusai.batch_prediction
abacusai.batch_prediction_version
diff --git a/docs/autoapi/abacusai/agent_data_document_info/index.html b/docs/autoapi/abacusai/agent_data_document_info/index.html
index 1dfb9c0c..0dcd43fe 100644
--- a/docs/autoapi/abacusai/agent_data_document_info/index.html
+++ b/docs/autoapi/abacusai/agent_data_document_info/index.html
@@ -70,6 +70,7 @@
abacusai.api_endpoint
abacusai.api_key
abacusai.app_user_group
+abacusai.app_user_group_sign_in_token
abacusai.application_connector
abacusai.batch_prediction
abacusai.batch_prediction_version
diff --git a/docs/autoapi/abacusai/agent_data_execution_result/index.html b/docs/autoapi/abacusai/agent_data_execution_result/index.html
index 1172083c..505d81fe 100644
--- a/docs/autoapi/abacusai/agent_data_execution_result/index.html
+++ b/docs/autoapi/abacusai/agent_data_execution_result/index.html
@@ -70,6 +70,7 @@
abacusai.api_endpoint
abacusai.api_key
abacusai.app_user_group
+abacusai.app_user_group_sign_in_token
abacusai.application_connector
abacusai.batch_prediction
abacusai.batch_prediction_version
diff --git a/docs/autoapi/abacusai/agent_version/index.html b/docs/autoapi/abacusai/agent_version/index.html
index 00914dbb..5361a7fa 100644
--- a/docs/autoapi/abacusai/agent_version/index.html
+++ b/docs/autoapi/abacusai/agent_version/index.html
@@ -70,6 +70,7 @@
abacusai.api_endpoint
abacusai.api_key
abacusai.app_user_group
+abacusai.app_user_group_sign_in_token
abacusai.application_connector
abacusai.batch_prediction
abacusai.batch_prediction_version
diff --git a/docs/autoapi/abacusai/ai_building_task/index.html b/docs/autoapi/abacusai/ai_building_task/index.html
index 8d30db88..33076f90 100644
--- a/docs/autoapi/abacusai/ai_building_task/index.html
+++ b/docs/autoapi/abacusai/ai_building_task/index.html
@@ -70,6 +70,7 @@
abacusai.api_endpoint
abacusai.api_key
abacusai.app_user_group
+abacusai.app_user_group_sign_in_token
abacusai.application_connector
abacusai.batch_prediction
abacusai.batch_prediction_version
diff --git a/docs/autoapi/abacusai/algorithm/index.html b/docs/autoapi/abacusai/algorithm/index.html
index 327386b0..b13534a2 100644
--- a/docs/autoapi/abacusai/algorithm/index.html
+++ b/docs/autoapi/abacusai/algorithm/index.html
@@ -70,6 +70,7 @@
abacusai.api_endpoint
abacusai.api_key
abacusai.app_user_group
+abacusai.app_user_group_sign_in_token
abacusai.application_connector
abacusai.batch_prediction
abacusai.batch_prediction_version
diff --git a/docs/autoapi/abacusai/annotation/index.html b/docs/autoapi/abacusai/annotation/index.html
index 0ae13c4c..f29318c1 100644
--- a/docs/autoapi/abacusai/annotation/index.html
+++ b/docs/autoapi/abacusai/annotation/index.html
@@ -70,6 +70,7 @@
abacusai.api_endpoint
abacusai.api_key
abacusai.app_user_group
+abacusai.app_user_group_sign_in_token
abacusai.application_connector
abacusai.batch_prediction
abacusai.batch_prediction_version
diff --git a/docs/autoapi/abacusai/annotation_config/index.html b/docs/autoapi/abacusai/annotation_config/index.html
index 9b499d9f..d525ceb4 100644
--- a/docs/autoapi/abacusai/annotation_config/index.html
+++ b/docs/autoapi/abacusai/annotation_config/index.html
@@ -70,6 +70,7 @@
abacusai.api_endpoint
abacusai.api_key
abacusai.app_user_group
+abacusai.app_user_group_sign_in_token
abacusai.application_connector
abacusai.batch_prediction
abacusai.batch_prediction_version
diff --git a/docs/autoapi/abacusai/annotation_document/index.html b/docs/autoapi/abacusai/annotation_document/index.html
index 3bdaca26..a8adb58b 100644
--- a/docs/autoapi/abacusai/annotation_document/index.html
+++ b/docs/autoapi/abacusai/annotation_document/index.html
@@ -70,6 +70,7 @@
abacusai.api_endpoint
abacusai.api_key
abacusai.app_user_group
+abacusai.app_user_group_sign_in_token
abacusai.application_connector
abacusai.batch_prediction
abacusai.batch_prediction_version
diff --git a/docs/autoapi/abacusai/annotation_entry/index.html b/docs/autoapi/abacusai/annotation_entry/index.html
index 2f875d0d..8ad51673 100644
--- a/docs/autoapi/abacusai/annotation_entry/index.html
+++ b/docs/autoapi/abacusai/annotation_entry/index.html
@@ -70,6 +70,7 @@
abacusai.api_endpoint
abacusai.api_key
abacusai.app_user_group
+abacusai.app_user_group_sign_in_token
abacusai.application_connector
abacusai.batch_prediction
abacusai.batch_prediction_version
diff --git a/docs/autoapi/abacusai/annotations_status/index.html b/docs/autoapi/abacusai/annotations_status/index.html
index 48cdda82..b3365653 100644
--- a/docs/autoapi/abacusai/annotations_status/index.html
+++ b/docs/autoapi/abacusai/annotations_status/index.html
@@ -70,6 +70,7 @@
abacusai.api_endpoint
abacusai.api_key
abacusai.app_user_group
+abacusai.app_user_group_sign_in_token
abacusai.application_connector
abacusai.batch_prediction
abacusai.batch_prediction_version
diff --git a/docs/autoapi/abacusai/api_class/abstract/index.html b/docs/autoapi/abacusai/api_class/abstract/index.html
index 7b52b7b6..3174c040 100644
--- a/docs/autoapi/abacusai/api_class/abstract/index.html
+++ b/docs/autoapi/abacusai/api_class/abstract/index.html
@@ -73,6 +73,7 @@
abacusai.api_endpoint
abacusai.api_key
abacusai.app_user_group
+abacusai.app_user_group_sign_in_token
abacusai.application_connector
abacusai.batch_prediction
abacusai.batch_prediction_version
diff --git a/docs/autoapi/abacusai/api_class/ai_agents/index.html b/docs/autoapi/abacusai/api_class/ai_agents/index.html
index fb20d7d7..3b2e5150 100644
--- a/docs/autoapi/abacusai/api_class/ai_agents/index.html
+++ b/docs/autoapi/abacusai/api_class/ai_agents/index.html
@@ -73,6 +73,7 @@
abacusai.api_endpoint
abacusai.api_key
abacusai.app_user_group
+abacusai.app_user_group_sign_in_token
abacusai.application_connector
abacusai.batch_prediction
abacusai.batch_prediction_version
diff --git a/docs/autoapi/abacusai/api_class/ai_chat/index.html b/docs/autoapi/abacusai/api_class/ai_chat/index.html
index 9f59c921..9ea882a3 100644
--- a/docs/autoapi/abacusai/api_class/ai_chat/index.html
+++ b/docs/autoapi/abacusai/api_class/ai_chat/index.html
@@ -73,6 +73,7 @@
abacusai.api_endpoint
abacusai.api_key
abacusai.app_user_group
+abacusai.app_user_group_sign_in_token
abacusai.application_connector
abacusai.batch_prediction
abacusai.batch_prediction_version
diff --git a/docs/autoapi/abacusai/api_class/batch_prediction/index.html b/docs/autoapi/abacusai/api_class/batch_prediction/index.html
index 9df83c13..bfb4a24f 100644
--- a/docs/autoapi/abacusai/api_class/batch_prediction/index.html
+++ b/docs/autoapi/abacusai/api_class/batch_prediction/index.html
@@ -73,6 +73,7 @@
abacusai.api_endpoint
abacusai.api_key
abacusai.app_user_group
+abacusai.app_user_group_sign_in_token
abacusai.application_connector
abacusai.batch_prediction
abacusai.batch_prediction_version
diff --git a/docs/autoapi/abacusai/api_class/blob_input/index.html b/docs/autoapi/abacusai/api_class/blob_input/index.html
index dbda647c..e27b135a 100644
--- a/docs/autoapi/abacusai/api_class/blob_input/index.html
+++ b/docs/autoapi/abacusai/api_class/blob_input/index.html
@@ -73,6 +73,7 @@
abacusai.api_endpoint
abacusai.api_key
abacusai.app_user_group
+abacusai.app_user_group_sign_in_token
abacusai.application_connector
abacusai.batch_prediction
abacusai.batch_prediction_version
diff --git a/docs/autoapi/abacusai/api_class/connectors/index.html b/docs/autoapi/abacusai/api_class/connectors/index.html
index 2ac0756c..100c32a6 100644
--- a/docs/autoapi/abacusai/api_class/connectors/index.html
+++ b/docs/autoapi/abacusai/api_class/connectors/index.html
@@ -73,6 +73,7 @@
abacusai.api_endpoint
abacusai.api_key
abacusai.app_user_group
+abacusai.app_user_group_sign_in_token
abacusai.application_connector
abacusai.batch_prediction
abacusai.batch_prediction_version
diff --git a/docs/autoapi/abacusai/api_class/dataset/index.html b/docs/autoapi/abacusai/api_class/dataset/index.html
index 6bcc2f45..70137610 100644
--- a/docs/autoapi/abacusai/api_class/dataset/index.html
+++ b/docs/autoapi/abacusai/api_class/dataset/index.html
@@ -73,6 +73,7 @@
abacusai.api_endpoint
abacusai.api_key
abacusai.app_user_group
+abacusai.app_user_group_sign_in_token
abacusai.application_connector
abacusai.batch_prediction
abacusai.batch_prediction_version
diff --git a/docs/autoapi/abacusai/api_class/dataset_application_connector/index.html b/docs/autoapi/abacusai/api_class/dataset_application_connector/index.html
index cf11cdaa..2af79777 100644
--- a/docs/autoapi/abacusai/api_class/dataset_application_connector/index.html
+++ b/docs/autoapi/abacusai/api_class/dataset_application_connector/index.html
@@ -73,6 +73,7 @@
abacusai.api_endpoint
abacusai.api_key
abacusai.app_user_group
+abacusai.app_user_group_sign_in_token
abacusai.application_connector
abacusai.batch_prediction
abacusai.batch_prediction_version
diff --git a/docs/autoapi/abacusai/api_class/deployment/index.html b/docs/autoapi/abacusai/api_class/deployment/index.html
index b64c99f4..f83fcd01 100644
--- a/docs/autoapi/abacusai/api_class/deployment/index.html
+++ b/docs/autoapi/abacusai/api_class/deployment/index.html
@@ -73,6 +73,7 @@
abacusai.api_endpoint
abacusai.api_key
abacusai.app_user_group
+abacusai.app_user_group_sign_in_token
abacusai.application_connector
abacusai.batch_prediction
abacusai.batch_prediction_version
diff --git a/docs/autoapi/abacusai/api_class/document_retriever/index.html b/docs/autoapi/abacusai/api_class/document_retriever/index.html
index bdf74071..65d087af 100644
--- a/docs/autoapi/abacusai/api_class/document_retriever/index.html
+++ b/docs/autoapi/abacusai/api_class/document_retriever/index.html
@@ -73,6 +73,7 @@
abacusai.api_endpoint
abacusai.api_key
abacusai.app_user_group
+abacusai.app_user_group_sign_in_token
abacusai.application_connector
abacusai.batch_prediction
abacusai.batch_prediction_version
diff --git a/docs/autoapi/abacusai/api_class/enums/index.html b/docs/autoapi/abacusai/api_class/enums/index.html
index f66b214b..16083d01 100644
--- a/docs/autoapi/abacusai/api_class/enums/index.html
+++ b/docs/autoapi/abacusai/api_class/enums/index.html
@@ -73,6 +73,7 @@
abacusai.api_endpoint
abacusai.api_key
abacusai.app_user_group
+abacusai.app_user_group_sign_in_token
abacusai.application_connector
abacusai.batch_prediction
abacusai.batch_prediction_version
@@ -2146,11 +2147,6 @@ Module Contents
-
--
-ONEDRIVEUSER = 'ONEDRIVEUSER'
-
-
-
TEAMSSCRAPER = 'TEAMSSCRAPER'
@@ -2486,6 +2482,11 @@ Module Contents
+
+-
+XAI_GROK = 'XAI_GROK'
+
+
diff --git a/docs/autoapi/abacusai/api_class/feature_group/index.html b/docs/autoapi/abacusai/api_class/feature_group/index.html
index 68f2f8eb..b9dde7d9 100644
--- a/docs/autoapi/abacusai/api_class/feature_group/index.html
+++ b/docs/autoapi/abacusai/api_class/feature_group/index.html
@@ -73,6 +73,7 @@
abacusai.api_endpoint
abacusai.api_key
abacusai.app_user_group
+abacusai.app_user_group_sign_in_token
abacusai.application_connector
abacusai.batch_prediction
abacusai.batch_prediction_version
diff --git a/docs/autoapi/abacusai/api_class/index.html b/docs/autoapi/abacusai/api_class/index.html
index 992fd011..75a2756e 100644
--- a/docs/autoapi/abacusai/api_class/index.html
+++ b/docs/autoapi/abacusai/api_class/index.html
@@ -73,6 +73,7 @@
abacusai.api_endpoint
abacusai.api_key
abacusai.app_user_group
+abacusai.app_user_group_sign_in_token
abacusai.application_connector
abacusai.batch_prediction
abacusai.batch_prediction_version
@@ -6291,11 +6292,6 @@ Package Contents
-
--
-ONEDRIVEUSER = 'ONEDRIVEUSER'
-
-
-
TEAMSSCRAPER = 'TEAMSSCRAPER'
@@ -6631,6 +6627,11 @@ Package Contents
+
+-
+XAI_GROK = 'XAI_GROK'
+
+
@@ -11471,7 +11472,7 @@ Package Contents
@@ -4037,6 +4038,27 @@ Module Contents
+
-
query_feature_group_code_generator(query, language, project_id=None)
diff --git a/docs/autoapi/abacusai/code_autocomplete_response/index.html b/docs/autoapi/abacusai/code_autocomplete_response/index.html
index ff302a92..e2684b8b 100644
--- a/docs/autoapi/abacusai/code_autocomplete_response/index.html
+++ b/docs/autoapi/abacusai/code_autocomplete_response/index.html
@@ -66,6 +66,7 @@
abacusai.api_endpoint
abacusai.api_key
abacusai.app_user_group
+abacusai.app_user_group_sign_in_token
abacusai.application_connector
abacusai.batch_prediction
abacusai.batch_prediction_version
diff --git a/docs/autoapi/abacusai/code_edit_response/index.html b/docs/autoapi/abacusai/code_edit_response/index.html
index 1aa67175..683ab7b8 100644
--- a/docs/autoapi/abacusai/code_edit_response/index.html
+++ b/docs/autoapi/abacusai/code_edit_response/index.html
@@ -66,6 +66,7 @@
abacusai.api_endpoint
abacusai.api_key
abacusai.app_user_group
+abacusai.app_user_group_sign_in_token
abacusai.application_connector
abacusai.batch_prediction
abacusai.batch_prediction_version
@@ -329,7 +330,7 @@ Classes