abacusai 0.60.0__tar.gz → 0.62.0__tar.gz

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

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

{abacusai-0.60.0 → abacusai-0.62.0}/abacusai/__init__.py

@@ -1,6 +1,7 @@
 from .api_class import *
 from .client import ApiClient, ApiException, ClientOptions, ReadOnlyClient
 from .prediction_client import PredictionClient
+from .streaming_client import StreamingClient
 
 
-__version__ = "0.60.0"
+__version__ = "0.61.0"

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.60.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.60.0 → abacusai-0.62.0}/abacusai/api_class/enums.py

@@ -22,6 +22,7 @@ class ProblemType(ApiEnum):
     ANOMALY_DETECTION = 'anomaly_new'
     ANOMALY_OUTLIERS = 'anomaly'
     CLUSTERING = 'clustering'
+    CLUSTERING_TIMESERIES = 'clustering_timeseries'
     CUMULATIVE_FORECASTING = 'cumulative_forecasting'
     NAMED_ENTITY_EXTRACTION = 'nlp_ner'
     NATURAL_LANGUAGE_SEARCH = 'nlp_search'

abacusai-0.62.0/abacusai/api_endpoint.py

@@ -0,0 +1,29 @@
+from .return_class import AbstractApiClass
+
+
+class ApiEndpoint(AbstractApiClass):
+    """
+        An collection of endpoints which can be used to make requests to, such as api calls or predict calls
+
+        Args:
+            client (ApiClient): An authenticated API Client instance
+            apiEndpoint (str): The URI that can be used to make API calls
+            predictEndpoint (str): The URI that can be used to make predict calls against Deployments
+    """
+
+    def __init__(self, client, apiEndpoint=None, predictEndpoint=None):
+        super().__init__(client, None)
+        self.api_endpoint = apiEndpoint
+        self.predict_endpoint = predictEndpoint
+
+    def __repr__(self):
+        return f"ApiEndpoint(api_endpoint={repr(self.api_endpoint)},\n  predict_endpoint={repr(self.predict_endpoint)})"
+
+    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 {'api_endpoint': self.api_endpoint, 'predict_endpoint': self.predict_endpoint}

abacusai-0.62.0/abacusai/chat_bot_response.py

@@ -0,0 +1,29 @@
+from .return_class import AbstractApiClass
+
+
+class ChatBotResponse(AbstractApiClass):
+    """
+        A response from the chat bot
+
+        Args:
+            client (ApiClient): An authenticated API Client instance
+            answer (str): The response from the chat bot
+            chatHistory (list): The chat history as a list of dicts with is_user and text entries
+    """
+
+    def __init__(self, client, answer=None, chatHistory=None):
+        super().__init__(client, None)
+        self.answer = answer
+        self.chat_history = chatHistory
+
+    def __repr__(self):
+        return f"ChatBotResponse(answer={repr(self.answer)},\n  chat_history={repr(self.chat_history)})"
+
+    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 {'answer': self.answer, 'chat_history': self.chat_history}