abacusai 0.61.0__tar.gz → 0.62.0__tar.gz

{abacusai-0.61.0/abacusai.egg-info → abacusai-0.62.0}/PKG-INFO

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

abacusai-0.62.0/abacusai/annotation_config.py

@@ -0,0 +1,31 @@
+from .return_class import AbstractApiClass
+
+
+class AnnotationConfig(AbstractApiClass):
+    """
+        Annotation config for a feature group
+
+        Args:
+            client (ApiClient): An authenticated API Client instance
+            featureAnnotationConfigs (list): List of feature annotation configs
+            labels (list): List of labels
+            statusFeature (str): Name of the feature that contains the status of the annotation (Optional)
+    """
+
+    def __init__(self, client, featureAnnotationConfigs=None, labels=None, statusFeature=None):
+        super().__init__(client, None)
+        self.feature_annotation_configs = featureAnnotationConfigs
+        self.labels = labels
+        self.status_feature = statusFeature
+
+    def __repr__(self):
+        return f"AnnotationConfig(feature_annotation_configs={repr(self.feature_annotation_configs)},\n  labels={repr(self.labels)},\n  status_feature={repr(self.status_feature)})"
+
+    def to_dict(self):
+        """
+        Get a dict representation of the parameters in this class
+
+        Returns:
+            dict: The dict value representation of the class parameters
+        """
+        return {'feature_annotation_configs': self.feature_annotation_configs, 'labels': self.labels, 'status_feature': self.status_feature}

{abacusai-0.61.0 → abacusai-0.62.0}/abacusai/api_class/dataset.py

@@ -6,3 +6,5 @@ from .abstract import ApiClass
 @dataclass
 class ParsingConfig(ApiClass):
     escape: str = field(default='"')
+    csv_delimiter: str = field(default=None)
+    file_path_with_schema: str = field(default=None)

{abacusai-0.61.0 → abacusai-0.62.0}/abacusai/client.py

@@ -79,6 +79,9 @@ from .monitor_alert_version import MonitorAlertVersion
 from .natural_language_explanation import NaturalLanguageExplanation
 from .notebook_completion import NotebookCompletion
 from .organization_group import OrganizationGroup
+from .pipeline import Pipeline
+from .pipeline_step import PipelineStep
+from .pipeline_version import PipelineVersion
 from .problem_type import ProblemType
 from .project import Project
 from .project_dataset import ProjectDataset
@@ -1535,6 +1538,37 @@ class ReadOnlyClient(BaseApiClient):
             PythonFunction: A list of PythonFunction objects."""
         return self._call_api('listPythonFunctions', 'GET', query_params={'functionType': function_type}, parse_type=PythonFunction)
 
+    def list_pipelines(self, project_id: str = None) -> Pipeline:
+        """Lists the pipelines for an organization or a project
+
+        Args:
+            project_id (str): Unique string identifier for the project to list graph dashboards from.
+
+        Returns:
+            Pipeline: A list of pipelines."""
+        return self._call_api('listPipelines', 'GET', query_params={'projectId': project_id}, parse_type=Pipeline)
+
+    def describe_pipeline_version(self, pipeline_version: str) -> PipelineVersion:
+        """Describes a specified pipeline version
+
+        Args:
+            pipeline_version (str): Unique string identifier for the pipeline version
+
+        Returns:
+            PipelineVersion: Object describing the pipeline version"""
+        return self._call_api('describePipelineVersion', 'GET', query_params={'pipelineVersion': pipeline_version}, parse_type=PipelineVersion)
+
+    def describe_pipeline_step(self, pipeline_name: str, step_name: str) -> PipelineStep:
+        """Deletes a step from a pipeline.
+
+        Args:
+            pipeline_name (str): The name of the pipeline.
+            step_name (str): The name of the step.
+
+        Returns:
+            PipelineStep: An object describing the pipeline step."""
+        return self._call_api('describePipelineStep', 'GET', query_params={'pipelineName': pipeline_name, 'stepName': step_name}, parse_type=PipelineStep)
+
     def describe_graph_dashboard(self, graph_dashboard_id: str) -> GraphDashboard:
         """Describes a given graph dashboard.
 
@@ -2365,7 +2399,7 @@ class ApiClient(ReadOnlyClient):
 
         Args:
             name (str): The project's name.
-            use_case (str): 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,  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_SEQUENCE_CLASSIFICATION_2,  PRETRAINED_TEXT_CLASSIFICATION.
+            use_case (str): 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,  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.
 
         Returns:
             Project: This object represents the newly created project. For details, refer to."""
@@ -3349,7 +3383,7 @@ Creates a new feature group defined as the union of other feature group versions
             location_date_format (str): The date format in which the data is partitioned in the cloud storage location. For example, if the data is partitioned as s3://bucket1/dir1/dir2/event_date=YYYY-MM-DD/dir4/filename.parquet, then the `location_date_format` is YYYY-MM-DD. This format needs to be consistent across all files within the specified location.
             date_format_lookback_days (int): The number of days to look back from the current day for import locations that are date partitioned. For example, import date 2021-06-04 with `date_format_lookback_days` = 3 will retrieve data for all the dates in the range [2021-06-02, 2021-06-04].
             incremental (bool): Signifies if the dataset is an incremental dataset.
-            is_documentset (bool): Signifies if the dataset is docstore dataset.
+            is_documentset (bool): Signifies if the dataset is docstore dataset. A docstore dataset contains documents like images, PDFs, audio files etc. or is tabular data with links to such files.
             extract_bounding_boxes (bool): Signifies whether to extract bounding boxes out of the documents. Only valid if is_documentset if True.
             merge_file_schemas (bool): Signifies if the merge file schema policy is enabled.
             parsing_config (ParsingConfig): Custom config for dataset parsing.
@@ -3440,7 +3474,7 @@ Creates a new feature group defined as the union of other feature group versions
             table_name (str): Organization-unique table name for this dataset.
             file_format (str): The file format of the dataset.
             csv_delimiter (str): If the file format is CSV, use a specific CSV delimiter.
-            is_documentset (bool): Signifies if the dataset is a docstore dataset.
+            is_documentset (bool): Signifies if the dataset is a docstore dataset. A docstore dataset contains documents like images, PDFs, audio files etc. or is tabular data with links to such files.
             extract_bounding_boxes (bool): Signifies whether to extract bounding boxes out of the documents. Only valid if is_documentset if True.
             parsing_config (ParsingConfig): Custom config for dataset parsing.
 
@@ -4441,7 +4475,7 @@ Creates a new feature group defined as the union of other feature group versions
             deployment_id, deployment_token,)
         return self._call_api('getForecast', 'POST', query_params={'deploymentToken': deployment_token, 'deploymentId': deployment_id}, body={'queryData': query_data, 'futureData': future_data, 'numPredictions': num_predictions, 'predictionStart': prediction_start, 'explainPredictions': explain_predictions, 'explainerType': explainer_type}, server_override=prediction_url)
 
-    def get_k_nearest(self, deployment_token: str, deployment_id: str, vector: list, k: int = None, distance: str = None, include_score: bool = False) -> Dict:
+    def get_k_nearest(self, deployment_token: str, deployment_id: str, vector: list, k: int = None, distance: str = None, include_score: bool = False, catalog_id: str = None) -> Dict:
         """Returns the k nearest neighbors for the provided embedding vector.
 
         Args:
@@ -4450,10 +4484,11 @@ Creates a new feature group defined as the union of other feature group versions
             vector (list): Input vector to perform the k nearest neighbors with.
             k (int): Overrideable number of items to return.
             distance (str): Specify the distance function to use when finding nearest neighbors.
-            include_score (bool): If True, will return the score alongside the resulting embedding value."""
+            include_score (bool): If True, will return the score alongside the resulting embedding value.
+            catalog_id (str): An optional parameter honored only for embeddings that provide a catalog id"""
         prediction_url = self._get_prediction_endpoint(
             deployment_id, deployment_token,)
-        return self._call_api('getKNearest', 'POST', query_params={'deploymentToken': deployment_token, 'deploymentId': deployment_id}, body={'vector': vector, 'k': k, 'distance': distance, 'includeScore': include_score}, server_override=prediction_url)
+        return self._call_api('getKNearest', 'POST', query_params={'deploymentToken': deployment_token, 'deploymentId': deployment_id}, body={'vector': vector, 'k': k, 'distance': distance, 'includeScore': include_score, 'catalogId': catalog_id}, server_override=prediction_url)
 
     def get_multiple_k_nearest(self, deployment_token: str, deployment_id: str, queries: list):
         """Returns the k nearest neighbors for the queries provided.
@@ -5094,6 +5129,97 @@ Creates a new feature group defined as the union of other feature group versions
             name (str): The name to identify the Python function."""
         return self._call_api('deletePythonFunction', 'DELETE', query_params={'name': name})
 
+    def create_pipeline(self, pipeline_name: str, project_id: str = None) -> Pipeline:
+        """Creates a pipeline for executing multiple steps.
+
+        Args:
+            pipeline_name (str): The name of the pipeline, which should be unique to the organization.
+            project_id (str): A unique string identifier for the plot dashboard.
+
+        Returns:
+            Pipeline: An object that describes a Pipeline."""
+        return self._call_api('createPipeline', 'POST', query_params={}, body={'pipelineName': pipeline_name, 'projectId': project_id}, parse_type=Pipeline)
+
+    def describe_pipeline(self, pipeline_name: str) -> Pipeline:
+        """Describes a given pipeline.
+
+        Args:
+            pipeline_name (str): The name of the pipeline to describe.
+
+        Returns:
+            Pipeline: An object describing a Pipeline"""
+        return self._call_api('describePipeline', 'POST', query_params={}, body={'pipelineName': pipeline_name}, parse_type=Pipeline)
+
+    def delete_pipeline(self, pipeline_name: None):
+        """Deletes a pipeline.
+
+        Args:
+            pipeline_name (None): The name of the pipeline to describe."""
+        return self._call_api('deletePipeline', 'DELETE', query_params={'pipelineName': pipeline_name})
+
+    def list_pipeline_versions(self, pipeline_name: str) -> List[PipelineVersion]:
+        """Lists the pipeline versions for a specified pipeline
+
+        Args:
+            pipeline_name (str): The name of the pipeline to describe.
+
+        Returns:
+            PipelineVersion: A list of pipeline versions."""
+        return self._call_api('listPipelineVersions', 'POST', query_params={}, body={'pipelineName': pipeline_name}, parse_type=PipelineVersion)
+
+    def run_pipeline(self, pipeline_name: str, pipeline_variable_mappings: list = None) -> PipelineVersion:
+        """Runs a specified pipeline with the arguments provided.
+
+        Args:
+            pipeline_name (str): The name of the pipeline to run.
+            pipeline_variable_mappings (list): List of Python function arguments for the pipeline.
+
+        Returns:
+            PipelineVersion: The object describing the pipeline"""
+        return self._call_api('runPipeline', 'POST', query_params={}, body={'pipelineName': pipeline_name, 'pipelineVariableMappings': pipeline_variable_mappings}, parse_type=PipelineVersion)
+
+    def create_pipeline_step(self, pipeline_name: str, step_name: str, function_name: str = None, source_code: str = None, step_input_mappings: list = None, output_variable_mappings: list = None, step_dependencies: list = None, package_requirements: list = None) -> Pipeline:
+        """Creates a step in a given pipeline.
+
+        Args:
+            pipeline_name (str): The name of the pipeline to run.
+            step_name (str): The name of the step.
+            function_name (str): The name of the Python function.
+            source_code (str): 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.
+            step_input_mappings (list): List of Python function arguments.
+            output_variable_mappings (list): List of Python function ouputs.
+            step_dependencies (list): List of step names this step depends on.
+            package_requirements (list): List of package requirement strings. For example: ['numpy==1.2.3', 'pandas>=1.4.0'].
+
+        Returns:
+            Pipeline: Object describing the pipeline."""
+        return self._call_api('createPipelineStep', 'POST', query_params={}, body={'pipelineName': pipeline_name, 'stepName': step_name, 'functionName': function_name, 'sourceCode': source_code, 'stepInputMappings': step_input_mappings, 'outputVariableMappings': output_variable_mappings, 'stepDependencies': step_dependencies, 'packageRequirements': package_requirements}, parse_type=Pipeline)
+
+    def delete_pipeline_step(self, pipeline_name: str, step_name: str):
+        """Deletes a step from a pipeline.
+
+        Args:
+            pipeline_name (str): The name of the pipeline.
+            step_name (str): The name of the step."""
+        return self._call_api('deletePipelineStep', 'DELETE', query_params={'pipelineName': pipeline_name, 'stepName': step_name})
+
+    def update_pipeline_step(self, pipeline_name: str, step_name: str, function_name: str = None, source_code: str = None, step_input_mappings: list = None, output_variable_mappings: list = None, step_dependencies: list = None, package_requirements: list = None) -> Pipeline:
+        """Creates a step in a given pipeline.
+
+        Args:
+            pipeline_name (str): The name of the pipeline to run.
+            step_name (str): The name of the step.
+            function_name (str): The name of the Python function.
+            source_code (str): 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.
+            step_input_mappings (list): List of Python function arguments.
+            output_variable_mappings (list): List of Python function ouputs.
+            step_dependencies (list): List of step names this step depends on.
+            package_requirements (list): List of package requirement strings. For example: ['numpy==1.2.3', 'pandas>=1.4.0'].
+
+        Returns:
+            Pipeline: Object describing the pipeline."""
+        return self._call_api('updatePipelineStep', 'POST', query_params={}, body={'pipelineName': pipeline_name, 'stepName': step_name, 'functionName': function_name, 'sourceCode': source_code, 'stepInputMappings': step_input_mappings, 'outputVariableMappings': output_variable_mappings, 'stepDependencies': step_dependencies, 'packageRequirements': package_requirements}, parse_type=Pipeline)
+
     def create_graph_dashboard(self, project_id: str, name: str, python_function_ids: list = None) -> GraphDashboard:
         """Create a plot dashboard given selected python plots
 

{abacusai-0.61.0 → abacusai-0.62.0}/abacusai/feature_group.py

@@ -1,5 +1,6 @@
 from typing import Union
 
+from .annotation_config import AnnotationConfig
 from .api_class import SamplingConfig
 from .code_source import CodeSource
 from .concatenation_config import ConcatenationConfig
@@ -54,11 +55,11 @@ class FeatureGroup(AbstractApiClass):
             useOriginalCsvNames (bool): If true, the feature group will use the original column names in the source dataset.
             pythonFunctionBindings (dict): Config specifying variable names, types, and values to use when resolving a Python feature group.
             pythonFunctionName (str): Name of the Python function the feature group was built from.
-            annotationConfig (annotationconfig): Annotation config for this feature
             projectConfig (dict): Project-specific config for this feature group.
             features (Feature): List of resolved features.
             duplicateFeatures (Feature): List of duplicate features.
             pointInTimeGroups (PointInTimeGroup): List of Point In Time Groups.
+            annotationConfig (AnnotationConfig): Annotation config for this feature
             latestFeatureGroupVersion (FeatureGroupVersion): Latest feature group version.
             concatenationConfig (ConcatenationConfig): Feature group ID whose data will be concatenated into this feature group.
             indexingConfig (IndexingConfig): Indexing config for the feature group for feature store
@@ -67,7 +68,7 @@ class FeatureGroup(AbstractApiClass):
             explanation (NaturalLanguageExplanation): Natural language explanation of the feature group
     """
 
-    def __init__(self, client, modificationLock=None, featureGroupId=None, name=None, featureGroupSourceType=None, tableName=None, sql=None, datasetId=None, functionSourceCode=None, functionName=None, sourceTables=None, createdAt=None, description=None, featureGroupType=None, sqlError=None, latestVersionOutdated=None, referencedFeatureGroups=None, tags=None, primaryKey=None, updateTimestampKey=None, lookupKeys=None, streamingEnabled=None, featureGroupUse=None, incremental=None, mergeConfig=None, transformConfig=None, samplingConfig=None, cpuSize=None, memory=None, streamingReady=None, featureTags=None, moduleName=None, templateBindings=None, featureExpression=None, useOriginalCsvNames=None, pythonFunctionBindings=None, pythonFunctionName=None, annotationConfig=None, projectConfig=None, features={}, duplicateFeatures={}, pointInTimeGroups={}, concatenationConfig={}, indexingConfig={}, codeSource={}, featureGroupTemplate={}, explanation={}, latestFeatureGroupVersion={}):
+    def __init__(self, client, modificationLock=None, featureGroupId=None, name=None, featureGroupSourceType=None, tableName=None, sql=None, datasetId=None, functionSourceCode=None, functionName=None, sourceTables=None, createdAt=None, description=None, featureGroupType=None, sqlError=None, latestVersionOutdated=None, referencedFeatureGroups=None, tags=None, primaryKey=None, updateTimestampKey=None, lookupKeys=None, streamingEnabled=None, featureGroupUse=None, incremental=None, mergeConfig=None, transformConfig=None, samplingConfig=None, cpuSize=None, memory=None, streamingReady=None, featureTags=None, moduleName=None, templateBindings=None, featureExpression=None, useOriginalCsvNames=None, pythonFunctionBindings=None, pythonFunctionName=None, projectConfig=None, features={}, duplicateFeatures={}, pointInTimeGroups={}, annotationConfig={}, concatenationConfig={}, indexingConfig={}, codeSource={}, featureGroupTemplate={}, explanation={}, latestFeatureGroupVersion={}):
         super().__init__(client, featureGroupId)
         self.modification_lock = modificationLock
         self.feature_group_id = featureGroupId
@@ -105,13 +106,14 @@ class FeatureGroup(AbstractApiClass):
         self.use_original_csv_names = useOriginalCsvNames
         self.python_function_bindings = pythonFunctionBindings
         self.python_function_name = pythonFunctionName
-        self.annotation_config = annotationConfig
         self.project_config = projectConfig
         self.features = client._build_class(Feature, features)
         self.duplicate_features = client._build_class(
             Feature, duplicateFeatures)
         self.point_in_time_groups = client._build_class(
             PointInTimeGroup, pointInTimeGroups)
+        self.annotation_config = client._build_class(
+            AnnotationConfig, annotationConfig)
         self.concatenation_config = client._build_class(
             ConcatenationConfig, concatenationConfig)
         self.indexing_config = client._build_class(
@@ -125,7 +127,7 @@ class FeatureGroup(AbstractApiClass):
             FeatureGroupVersion, latestFeatureGroupVersion)
 
     def __repr__(self):
-        return f"FeatureGroup(modification_lock={repr(self.modification_lock)},\n  feature_group_id={repr(self.feature_group_id)},\n  name={repr(self.name)},\n  feature_group_source_type={repr(self.feature_group_source_type)},\n  table_name={repr(self.table_name)},\n  sql={repr(self.sql)},\n  dataset_id={repr(self.dataset_id)},\n  function_source_code={repr(self.function_source_code)},\n  function_name={repr(self.function_name)},\n  source_tables={repr(self.source_tables)},\n  created_at={repr(self.created_at)},\n  description={repr(self.description)},\n  feature_group_type={repr(self.feature_group_type)},\n  sql_error={repr(self.sql_error)},\n  latest_version_outdated={repr(self.latest_version_outdated)},\n  referenced_feature_groups={repr(self.referenced_feature_groups)},\n  tags={repr(self.tags)},\n  primary_key={repr(self.primary_key)},\n  update_timestamp_key={repr(self.update_timestamp_key)},\n  lookup_keys={repr(self.lookup_keys)},\n  streaming_enabled={repr(self.streaming_enabled)},\n  feature_group_use={repr(self.feature_group_use)},\n  incremental={repr(self.incremental)},\n  merge_config={repr(self.merge_config)},\n  transform_config={repr(self.transform_config)},\n  sampling_config={repr(self.sampling_config)},\n  cpu_size={repr(self.cpu_size)},\n  memory={repr(self.memory)},\n  streaming_ready={repr(self.streaming_ready)},\n  feature_tags={repr(self.feature_tags)},\n  module_name={repr(self.module_name)},\n  template_bindings={repr(self.template_bindings)},\n  feature_expression={repr(self.feature_expression)},\n  use_original_csv_names={repr(self.use_original_csv_names)},\n  python_function_bindings={repr(self.python_function_bindings)},\n  python_function_name={repr(self.python_function_name)},\n  annotation_config={repr(self.annotation_config)},\n  project_config={repr(self.project_config)},\n  features={repr(self.features)},\n  duplicate_features={repr(self.duplicate_features)},\n  point_in_time_groups={repr(self.point_in_time_groups)},\n  concatenation_config={repr(self.concatenation_config)},\n  indexing_config={repr(self.indexing_config)},\n  code_source={repr(self.code_source)},\n  feature_group_template={repr(self.feature_group_template)},\n  explanation={repr(self.explanation)},\n  latest_feature_group_version={repr(self.latest_feature_group_version)})"
+        return f"FeatureGroup(modification_lock={repr(self.modification_lock)},\n  feature_group_id={repr(self.feature_group_id)},\n  name={repr(self.name)},\n  feature_group_source_type={repr(self.feature_group_source_type)},\n  table_name={repr(self.table_name)},\n  sql={repr(self.sql)},\n  dataset_id={repr(self.dataset_id)},\n  function_source_code={repr(self.function_source_code)},\n  function_name={repr(self.function_name)},\n  source_tables={repr(self.source_tables)},\n  created_at={repr(self.created_at)},\n  description={repr(self.description)},\n  feature_group_type={repr(self.feature_group_type)},\n  sql_error={repr(self.sql_error)},\n  latest_version_outdated={repr(self.latest_version_outdated)},\n  referenced_feature_groups={repr(self.referenced_feature_groups)},\n  tags={repr(self.tags)},\n  primary_key={repr(self.primary_key)},\n  update_timestamp_key={repr(self.update_timestamp_key)},\n  lookup_keys={repr(self.lookup_keys)},\n  streaming_enabled={repr(self.streaming_enabled)},\n  feature_group_use={repr(self.feature_group_use)},\n  incremental={repr(self.incremental)},\n  merge_config={repr(self.merge_config)},\n  transform_config={repr(self.transform_config)},\n  sampling_config={repr(self.sampling_config)},\n  cpu_size={repr(self.cpu_size)},\n  memory={repr(self.memory)},\n  streaming_ready={repr(self.streaming_ready)},\n  feature_tags={repr(self.feature_tags)},\n  module_name={repr(self.module_name)},\n  template_bindings={repr(self.template_bindings)},\n  feature_expression={repr(self.feature_expression)},\n  use_original_csv_names={repr(self.use_original_csv_names)},\n  python_function_bindings={repr(self.python_function_bindings)},\n  python_function_name={repr(self.python_function_name)},\n  project_config={repr(self.project_config)},\n  features={repr(self.features)},\n  duplicate_features={repr(self.duplicate_features)},\n  point_in_time_groups={repr(self.point_in_time_groups)},\n  annotation_config={repr(self.annotation_config)},\n  concatenation_config={repr(self.concatenation_config)},\n  indexing_config={repr(self.indexing_config)},\n  code_source={repr(self.code_source)},\n  feature_group_template={repr(self.feature_group_template)},\n  explanation={repr(self.explanation)},\n  latest_feature_group_version={repr(self.latest_feature_group_version)})"
 
     def to_dict(self):
         """
@@ -134,7 +136,7 @@ class FeatureGroup(AbstractApiClass):
         Returns:
             dict: The dict value representation of the class parameters
         """
-        return {'modification_lock': self.modification_lock, 'feature_group_id': self.feature_group_id, 'name': self.name, 'feature_group_source_type': self.feature_group_source_type, 'table_name': self.table_name, 'sql': self.sql, 'dataset_id': self.dataset_id, 'function_source_code': self.function_source_code, 'function_name': self.function_name, 'source_tables': self.source_tables, 'created_at': self.created_at, 'description': self.description, 'feature_group_type': self.feature_group_type, 'sql_error': self.sql_error, 'latest_version_outdated': self.latest_version_outdated, 'referenced_feature_groups': self.referenced_feature_groups, 'tags': self.tags, 'primary_key': self.primary_key, 'update_timestamp_key': self.update_timestamp_key, 'lookup_keys': self.lookup_keys, 'streaming_enabled': self.streaming_enabled, 'feature_group_use': self.feature_group_use, 'incremental': self.incremental, 'merge_config': self.merge_config, 'transform_config': self.transform_config, 'sampling_config': self.sampling_config, 'cpu_size': self.cpu_size, 'memory': self.memory, 'streaming_ready': self.streaming_ready, 'feature_tags': self.feature_tags, 'module_name': self.module_name, 'template_bindings': self.template_bindings, 'feature_expression': self.feature_expression, 'use_original_csv_names': self.use_original_csv_names, 'python_function_bindings': self.python_function_bindings, 'python_function_name': self.python_function_name, 'annotation_config': self.annotation_config, 'project_config': self.project_config, 'features': self._get_attribute_as_dict(self.features), 'duplicate_features': self._get_attribute_as_dict(self.duplicate_features), 'point_in_time_groups': self._get_attribute_as_dict(self.point_in_time_groups), 'concatenation_config': self._get_attribute_as_dict(self.concatenation_config), 'indexing_config': self._get_attribute_as_dict(self.indexing_config), 'code_source': self._get_attribute_as_dict(self.code_source), 'feature_group_template': self._get_attribute_as_dict(self.feature_group_template), 'explanation': self._get_attribute_as_dict(self.explanation), 'latest_feature_group_version': self._get_attribute_as_dict(self.latest_feature_group_version)}
+        return {'modification_lock': self.modification_lock, 'feature_group_id': self.feature_group_id, 'name': self.name, 'feature_group_source_type': self.feature_group_source_type, 'table_name': self.table_name, 'sql': self.sql, 'dataset_id': self.dataset_id, 'function_source_code': self.function_source_code, 'function_name': self.function_name, 'source_tables': self.source_tables, 'created_at': self.created_at, 'description': self.description, 'feature_group_type': self.feature_group_type, 'sql_error': self.sql_error, 'latest_version_outdated': self.latest_version_outdated, 'referenced_feature_groups': self.referenced_feature_groups, 'tags': self.tags, 'primary_key': self.primary_key, 'update_timestamp_key': self.update_timestamp_key, 'lookup_keys': self.lookup_keys, 'streaming_enabled': self.streaming_enabled, 'feature_group_use': self.feature_group_use, 'incremental': self.incremental, 'merge_config': self.merge_config, 'transform_config': self.transform_config, 'sampling_config': self.sampling_config, 'cpu_size': self.cpu_size, 'memory': self.memory, 'streaming_ready': self.streaming_ready, 'feature_tags': self.feature_tags, 'module_name': self.module_name, 'template_bindings': self.template_bindings, 'feature_expression': self.feature_expression, 'use_original_csv_names': self.use_original_csv_names, 'python_function_bindings': self.python_function_bindings, 'python_function_name': self.python_function_name, 'project_config': self.project_config, 'features': self._get_attribute_as_dict(self.features), 'duplicate_features': self._get_attribute_as_dict(self.duplicate_features), 'point_in_time_groups': self._get_attribute_as_dict(self.point_in_time_groups), 'annotation_config': self._get_attribute_as_dict(self.annotation_config), 'concatenation_config': self._get_attribute_as_dict(self.concatenation_config), 'indexing_config': self._get_attribute_as_dict(self.indexing_config), 'code_source': self._get_attribute_as_dict(self.code_source), 'feature_group_template': self._get_attribute_as_dict(self.feature_group_template), 'explanation': self._get_attribute_as_dict(self.explanation), 'latest_feature_group_version': self._get_attribute_as_dict(self.latest_feature_group_version)}
 
     def add_to_project(self, project_id: str, feature_group_type: str = 'CUSTOM_TABLE'):
         """

abacusai-0.62.0/abacusai/feature_group_metrics.py

@@ -0,0 +1,29 @@
+from .return_class import AbstractApiClass
+
+
+class FeatureGroupMetrics(AbstractApiClass):
+    """
+        Metrics for a feature group
+
+        Args:
+            client (ApiClient): An authenticated API Client instance
+            metrics (list[dict]): A list of dicts with metrics for each columns in the feature group
+            schema (list[dict]): A list of dicts with the schema for each metric
+    """
+
+    def __init__(self, client, metrics=None, schema=None):
+        super().__init__(client, None)
+        self.metrics = metrics
+        self.schema = schema
+
+    def __repr__(self):
+        return f"FeatureGroupMetrics(metrics={repr(self.metrics)},\n  schema={repr(self.schema)})"
+
+    def to_dict(self):
+        """
+        Get a dict representation of the parameters in this class
+
+        Returns:
+            dict: The dict value representation of the class parameters
+        """
+        return {'metrics': self.metrics, 'schema': self.schema}

{abacusai-0.61.0 → abacusai-0.62.0}/abacusai/feature_group_version.py

@@ -2,6 +2,7 @@ import os
 import tempfile
 from concurrent.futures import ThreadPoolExecutor
 
+from .annotation_config import AnnotationConfig
 from .code_source import CodeSource
 from .feature import Feature
 from .point_in_time_group import PointInTimeGroup
@@ -26,13 +27,13 @@ class FeatureGroupVersion(AbstractApiClass):
             memory (int): Memory in GB specified for the python feature group.
             useOriginalCsvNames (bool): If true, the feature group will use the original column names in the source dataset.
             pythonFunctionBindings (list): Config specifying variable names, types, and values to use when resolving a Python feature group.
-            annotationConfig (dict): The annotations config for the feature group.
             features (Feature): List of features.
             pointInTimeGroups (PointInTimeGroup): List of Point In Time Groups
             codeSource (CodeSource): If a python feature group, information on the source code
+            annotationConfig (AnnotationConfig): The annotations config for the feature group.
     """
 
-    def __init__(self, client, featureGroupVersion=None, featureGroupId=None, sql=None, sourceTables=None, createdAt=None, status=None, error=None, deployable=None, cpuSize=None, memory=None, useOriginalCsvNames=None, pythonFunctionBindings=None, annotationConfig=None, features={}, pointInTimeGroups={}, codeSource={}):
+    def __init__(self, client, featureGroupVersion=None, featureGroupId=None, sql=None, sourceTables=None, createdAt=None, status=None, error=None, deployable=None, cpuSize=None, memory=None, useOriginalCsvNames=None, pythonFunctionBindings=None, features={}, pointInTimeGroups={}, codeSource={}, annotationConfig={}):
         super().__init__(client, featureGroupVersion)
         self.feature_group_version = featureGroupVersion
         self.feature_group_id = featureGroupId
@@ -46,14 +47,15 @@ class FeatureGroupVersion(AbstractApiClass):
         self.memory = memory
         self.use_original_csv_names = useOriginalCsvNames
         self.python_function_bindings = pythonFunctionBindings
-        self.annotation_config = annotationConfig
         self.features = client._build_class(Feature, features)
         self.point_in_time_groups = client._build_class(
             PointInTimeGroup, pointInTimeGroups)
         self.code_source = client._build_class(CodeSource, codeSource)
+        self.annotation_config = client._build_class(
+            AnnotationConfig, annotationConfig)
 
     def __repr__(self):
-        return f"FeatureGroupVersion(feature_group_version={repr(self.feature_group_version)},\n  feature_group_id={repr(self.feature_group_id)},\n  sql={repr(self.sql)},\n  source_tables={repr(self.source_tables)},\n  created_at={repr(self.created_at)},\n  status={repr(self.status)},\n  error={repr(self.error)},\n  deployable={repr(self.deployable)},\n  cpu_size={repr(self.cpu_size)},\n  memory={repr(self.memory)},\n  use_original_csv_names={repr(self.use_original_csv_names)},\n  python_function_bindings={repr(self.python_function_bindings)},\n  annotation_config={repr(self.annotation_config)},\n  features={repr(self.features)},\n  point_in_time_groups={repr(self.point_in_time_groups)},\n  code_source={repr(self.code_source)})"
+        return f"FeatureGroupVersion(feature_group_version={repr(self.feature_group_version)},\n  feature_group_id={repr(self.feature_group_id)},\n  sql={repr(self.sql)},\n  source_tables={repr(self.source_tables)},\n  created_at={repr(self.created_at)},\n  status={repr(self.status)},\n  error={repr(self.error)},\n  deployable={repr(self.deployable)},\n  cpu_size={repr(self.cpu_size)},\n  memory={repr(self.memory)},\n  use_original_csv_names={repr(self.use_original_csv_names)},\n  python_function_bindings={repr(self.python_function_bindings)},\n  features={repr(self.features)},\n  point_in_time_groups={repr(self.point_in_time_groups)},\n  code_source={repr(self.code_source)},\n  annotation_config={repr(self.annotation_config)})"
 
     def to_dict(self):
         """
@@ -62,7 +64,7 @@ class FeatureGroupVersion(AbstractApiClass):
         Returns:
             dict: The dict value representation of the class parameters
         """
-        return {'feature_group_version': self.feature_group_version, 'feature_group_id': self.feature_group_id, 'sql': self.sql, 'source_tables': self.source_tables, 'created_at': self.created_at, 'status': self.status, 'error': self.error, 'deployable': self.deployable, 'cpu_size': self.cpu_size, 'memory': self.memory, 'use_original_csv_names': self.use_original_csv_names, 'python_function_bindings': self.python_function_bindings, 'annotation_config': self.annotation_config, 'features': self._get_attribute_as_dict(self.features), 'point_in_time_groups': self._get_attribute_as_dict(self.point_in_time_groups), 'code_source': self._get_attribute_as_dict(self.code_source)}
+        return {'feature_group_version': self.feature_group_version, 'feature_group_id': self.feature_group_id, 'sql': self.sql, 'source_tables': self.source_tables, 'created_at': self.created_at, 'status': self.status, 'error': self.error, 'deployable': self.deployable, 'cpu_size': self.cpu_size, 'memory': self.memory, 'use_original_csv_names': self.use_original_csv_names, 'python_function_bindings': self.python_function_bindings, 'features': self._get_attribute_as_dict(self.features), 'point_in_time_groups': self._get_attribute_as_dict(self.point_in_time_groups), 'code_source': self._get_attribute_as_dict(self.code_source), 'annotation_config': self._get_attribute_as_dict(self.annotation_config)}
 
     def create_snapshot_feature_group(self, table_name: str):
         """

{abacusai-0.61.0 → abacusai-0.62.0}/abacusai/model_upload.py

@@ -21,9 +21,10 @@ class ModelUpload(AbstractApiClass):
             inputPreprocessorUploadId (str): An upload identifier to be used when uploading the input preprocessor JSON file.
             requirementsUploadId (str): An upload identifier to be used when uploading the requirements JSON file.
             resourcesUploadId (str): An upload identifier to be used when uploading the resources JSON file.
+            multiCatalogEmbeddingsUploadId (str): An upload identifier to be used when upload the multi-catalog embeddings CSV file.
     """
 
-    def __init__(self, client, modelId=None, modelVersion=None, status=None, createdAt=None, modelUploadId=None, embeddingsUploadId=None, artifactsUploadId=None, verificationsUploadId=None, defaultItemsUploadId=None, modelFileUploadId=None, modelStateUploadId=None, inputPreprocessorUploadId=None, requirementsUploadId=None, resourcesUploadId=None):
+    def __init__(self, client, modelId=None, modelVersion=None, status=None, createdAt=None, modelUploadId=None, embeddingsUploadId=None, artifactsUploadId=None, verificationsUploadId=None, defaultItemsUploadId=None, modelFileUploadId=None, modelStateUploadId=None, inputPreprocessorUploadId=None, requirementsUploadId=None, resourcesUploadId=None, multiCatalogEmbeddingsUploadId=None):
         super().__init__(client, modelUploadId)
         self.model_id = modelId
         self.model_version = modelVersion
@@ -39,9 +40,10 @@ class ModelUpload(AbstractApiClass):
         self.input_preprocessor_upload_id = inputPreprocessorUploadId
         self.requirements_upload_id = requirementsUploadId
         self.resources_upload_id = resourcesUploadId
+        self.multi_catalog_embeddings_upload_id = multiCatalogEmbeddingsUploadId
 
     def __repr__(self):
-        return f"ModelUpload(model_id={repr(self.model_id)},\n  model_version={repr(self.model_version)},\n  status={repr(self.status)},\n  created_at={repr(self.created_at)},\n  model_upload_id={repr(self.model_upload_id)},\n  embeddings_upload_id={repr(self.embeddings_upload_id)},\n  artifacts_upload_id={repr(self.artifacts_upload_id)},\n  verifications_upload_id={repr(self.verifications_upload_id)},\n  default_items_upload_id={repr(self.default_items_upload_id)},\n  model_file_upload_id={repr(self.model_file_upload_id)},\n  model_state_upload_id={repr(self.model_state_upload_id)},\n  input_preprocessor_upload_id={repr(self.input_preprocessor_upload_id)},\n  requirements_upload_id={repr(self.requirements_upload_id)},\n  resources_upload_id={repr(self.resources_upload_id)})"
+        return f"ModelUpload(model_id={repr(self.model_id)},\n  model_version={repr(self.model_version)},\n  status={repr(self.status)},\n  created_at={repr(self.created_at)},\n  model_upload_id={repr(self.model_upload_id)},\n  embeddings_upload_id={repr(self.embeddings_upload_id)},\n  artifacts_upload_id={repr(self.artifacts_upload_id)},\n  verifications_upload_id={repr(self.verifications_upload_id)},\n  default_items_upload_id={repr(self.default_items_upload_id)},\n  model_file_upload_id={repr(self.model_file_upload_id)},\n  model_state_upload_id={repr(self.model_state_upload_id)},\n  input_preprocessor_upload_id={repr(self.input_preprocessor_upload_id)},\n  requirements_upload_id={repr(self.requirements_upload_id)},\n  resources_upload_id={repr(self.resources_upload_id)},\n  multi_catalog_embeddings_upload_id={repr(self.multi_catalog_embeddings_upload_id)})"
 
     def to_dict(self):
         """
@@ -50,4 +52,4 @@ class ModelUpload(AbstractApiClass):
         Returns:
             dict: The dict value representation of the class parameters
         """
-        return {'model_id': self.model_id, 'model_version': self.model_version, 'status': self.status, 'created_at': self.created_at, 'model_upload_id': self.model_upload_id, 'embeddings_upload_id': self.embeddings_upload_id, 'artifacts_upload_id': self.artifacts_upload_id, 'verifications_upload_id': self.verifications_upload_id, 'default_items_upload_id': self.default_items_upload_id, 'model_file_upload_id': self.model_file_upload_id, 'model_state_upload_id': self.model_state_upload_id, 'input_preprocessor_upload_id': self.input_preprocessor_upload_id, 'requirements_upload_id': self.requirements_upload_id, 'resources_upload_id': self.resources_upload_id}
+        return {'model_id': self.model_id, 'model_version': self.model_version, 'status': self.status, 'created_at': self.created_at, 'model_upload_id': self.model_upload_id, 'embeddings_upload_id': self.embeddings_upload_id, 'artifacts_upload_id': self.artifacts_upload_id, 'verifications_upload_id': self.verifications_upload_id, 'default_items_upload_id': self.default_items_upload_id, 'model_file_upload_id': self.model_file_upload_id, 'model_state_upload_id': self.model_state_upload_id, 'input_preprocessor_upload_id': self.input_preprocessor_upload_id, 'requirements_upload_id': self.requirements_upload_id, 'resources_upload_id': self.resources_upload_id, 'multi_catalog_embeddings_upload_id': self.multi_catalog_embeddings_upload_id}

abacusai-0.62.0/abacusai/pipeline.py

@@ -0,0 +1,36 @@
+from .pipeline_step import PipelineStep
+from .return_class import AbstractApiClass
+
+
+class Pipeline(AbstractApiClass):
+    """
+        A Pipeline For Steps.
+
+        Args:
+            client (ApiClient): An authenticated API Client instance
+            pipelineName (str): The name of the pipeline this step is a part of.
+            pipelineId (str): The reference to the pipeline this step belongs to.
+            createdAt (str): The date and time which the pipeline was created.
+            pipelineVariableMappings (dict): A description of the function variables into the pipeline.
+            steps (PipelineStep): A list of the pipeline steps attached to the pipeline.
+    """
+
+    def __init__(self, client, pipelineName=None, pipelineId=None, createdAt=None, pipelineVariableMappings=None, steps={}):
+        super().__init__(client, pipelineId)
+        self.pipeline_name = pipelineName
+        self.pipeline_id = pipelineId
+        self.created_at = createdAt
+        self.pipeline_variable_mappings = pipelineVariableMappings
+        self.steps = client._build_class(PipelineStep, steps)
+
+    def __repr__(self):
+        return f"Pipeline(pipeline_name={repr(self.pipeline_name)},\n  pipeline_id={repr(self.pipeline_id)},\n  created_at={repr(self.created_at)},\n  pipeline_variable_mappings={repr(self.pipeline_variable_mappings)},\n  steps={repr(self.steps)})"
+
+    def to_dict(self):
+        """
+        Get a dict representation of the parameters in this class
+
+        Returns:
+            dict: The dict value representation of the class parameters
+        """
+        return {'pipeline_name': self.pipeline_name, 'pipeline_id': self.pipeline_id, 'created_at': self.created_at, 'pipeline_variable_mappings': self.pipeline_variable_mappings, 'steps': self._get_attribute_as_dict(self.steps)}

abacusai-0.62.0/abacusai/pipeline_step.py

@@ -0,0 +1,45 @@
+from .python_function import PythonFunction
+from .return_class import AbstractApiClass
+
+
+class PipelineStep(AbstractApiClass):
+    """
+        A step in a pipeline.
+
+        Args:
+            client (ApiClient): An authenticated API Client instance
+            pipelineStepId (str): The reference to this step.
+            pipelineId (str): The reference to the pipeline this step belongs to.
+            stepName (str): The name of the step.
+            pipelineName (str): The name of the pipeline this step is a part of.
+            createdAt (str): The date and time which this step was created.
+            updatedAt (str): The date and time when this step was last updated.
+            pythonFunctionId (str): The python function_id.
+            stepDependencies (list[str]): List of steps this step depends on.
+            pythonFunction (PythonFunction): Information about the python function for the step.
+    """
+
+    def __init__(self, client, pipelineStepId=None, pipelineId=None, stepName=None, pipelineName=None, createdAt=None, updatedAt=None, pythonFunctionId=None, stepDependencies=None, pythonFunction={}):
+        super().__init__(client, pipelineStepId)
+        self.pipeline_step_id = pipelineStepId
+        self.pipeline_id = pipelineId
+        self.step_name = stepName
+        self.pipeline_name = pipelineName
+        self.created_at = createdAt
+        self.updated_at = updatedAt
+        self.python_function_id = pythonFunctionId
+        self.step_dependencies = stepDependencies
+        self.python_function = client._build_class(
+            PythonFunction, pythonFunction)
+
+    def __repr__(self):
+        return f"PipelineStep(pipeline_step_id={repr(self.pipeline_step_id)},\n  pipeline_id={repr(self.pipeline_id)},\n  step_name={repr(self.step_name)},\n  pipeline_name={repr(self.pipeline_name)},\n  created_at={repr(self.created_at)},\n  updated_at={repr(self.updated_at)},\n  python_function_id={repr(self.python_function_id)},\n  step_dependencies={repr(self.step_dependencies)},\n  python_function={repr(self.python_function)})"
+
+    def to_dict(self):
+        """
+        Get a dict representation of the parameters in this class
+
+        Returns:
+            dict: The dict value representation of the class parameters
+        """
+        return {'pipeline_step_id': self.pipeline_step_id, 'pipeline_id': self.pipeline_id, 'step_name': self.step_name, 'pipeline_name': self.pipeline_name, 'created_at': self.created_at, 'updated_at': self.updated_at, 'python_function_id': self.python_function_id, 'step_dependencies': self.step_dependencies, 'python_function': self._get_attribute_as_dict(self.python_function)}

abacusai-0.62.0/abacusai/pipeline_step_version.py

@@ -0,0 +1,49 @@
+from .return_class import AbstractApiClass
+
+
+class PipelineStepVersion(AbstractApiClass):
+    """
+        A version of a pipeline step.
+
+        Args:
+            client (ApiClient): An authenticated API Client instance
+            stepName (str): The name of the step.
+            pipelineStepVersion (str): The reference to the pipeline step version.
+            pipelineStepId (str): The reference to this step.
+            pipelineId (str): The reference to the pipeline this step belongs to.
+            pipelineVersion (str): The reference to the pipeline version.
+            createdAt (str): The date and time which this step was created.
+            updatedAt (str): The date and time when this step was last updated.
+            status (str): The status of the pipeline version.
+            pythonFunctionId (str): The reference to the python function
+            functionVariableMappings (dict): The mappings for function parameters' names.
+            stepDependencies (list[str]): List of steps this step depends on.
+            outputVariableMappings (dict): The mappings for the output variables to the step.
+    """
+
+    def __init__(self, client, stepName=None, pipelineStepVersion=None, pipelineStepId=None, pipelineId=None, pipelineVersion=None, createdAt=None, updatedAt=None, status=None, pythonFunctionId=None, functionVariableMappings=None, stepDependencies=None, outputVariableMappings=None):
+        super().__init__(client, pipelineStepVersion)
+        self.step_name = stepName
+        self.pipeline_step_version = pipelineStepVersion
+        self.pipeline_step_id = pipelineStepId
+        self.pipeline_id = pipelineId
+        self.pipeline_version = pipelineVersion
+        self.created_at = createdAt
+        self.updated_at = updatedAt
+        self.status = status
+        self.python_function_id = pythonFunctionId
+        self.function_variable_mappings = functionVariableMappings
+        self.step_dependencies = stepDependencies
+        self.output_variable_mappings = outputVariableMappings
+
+    def __repr__(self):
+        return f"PipelineStepVersion(step_name={repr(self.step_name)},\n  pipeline_step_version={repr(self.pipeline_step_version)},\n  pipeline_step_id={repr(self.pipeline_step_id)},\n  pipeline_id={repr(self.pipeline_id)},\n  pipeline_version={repr(self.pipeline_version)},\n  created_at={repr(self.created_at)},\n  updated_at={repr(self.updated_at)},\n  status={repr(self.status)},\n  python_function_id={repr(self.python_function_id)},\n  function_variable_mappings={repr(self.function_variable_mappings)},\n  step_dependencies={repr(self.step_dependencies)},\n  output_variable_mappings={repr(self.output_variable_mappings)})"
+
+    def to_dict(self):
+        """
+        Get a dict representation of the parameters in this class
+
+        Returns:
+            dict: The dict value representation of the class parameters
+        """
+        return {'step_name': self.step_name, 'pipeline_step_version': self.pipeline_step_version, 'pipeline_step_id': self.pipeline_step_id, 'pipeline_id': self.pipeline_id, 'pipeline_version': self.pipeline_version, 'created_at': self.created_at, 'updated_at': self.updated_at, 'status': self.status, 'python_function_id': self.python_function_id, 'function_variable_mappings': self.function_variable_mappings, 'step_dependencies': self.step_dependencies, 'output_variable_mappings': self.output_variable_mappings}