crypticorn 2.7.5__py3-none-any.whl → 2.8.0__py3-none-any.whl

crypticorn/hive/client/models/data_options.py

@@ -0,0 +1,92 @@
+# coding: utf-8
+
+"""
+Hive AI API
+
+API for Hive AI model training and evaluation
+
+The version of the OpenAPI document: 1.0.0
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict, Field
+from typing import Any, ClassVar, Dict, List
+from crypticorn.hive.client.models.feature_size import FeatureSize
+from crypticorn.hive.client.models.target import Target
+from typing import Optional, Set
+from typing_extensions import Self
+
+
+class DataOptions(BaseModel):
+    """
+    DataOptions
+    """  # noqa: E501
+
+    targets: List[Target] = Field(
+        description="The targets available on the latest data version."
+    )
+    feature_sizes: List[FeatureSize] = Field(
+        description="The feature sizes available on the latest data version."
+    )
+    __properties: ClassVar[List[str]] = ["targets", "feature_sizes"]
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of DataOptions from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of DataOptions from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate(
+            {"targets": obj.get("targets"), "feature_sizes": obj.get("feature_sizes")}
+        )
+        return _obj

crypticorn/hive/client/models/exception_detail.py

@@ -19,6 +19,9 @@ import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictInt, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
+from crypticorn.hive.client.models.api_error_identifier import ApiErrorIdentifier
+from crypticorn.hive.client.models.api_error_level import ApiErrorLevel
+from crypticorn.hive.client.models.api_error_type import ApiErrorType
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -29,9 +32,9 @@ class ExceptionDetail(BaseModel):
     """  # noqa: E501
 
     message: Optional[StrictStr] = None
-    code: StrictStr = Field(description="API error identifiers")
-    type: StrictStr = Field(description="Type of API error")
-    level: StrictStr = Field(description="API error levels")
+    code: ApiErrorIdentifier = Field(description="The unique error code")
+    type: ApiErrorType = Field(description="The type of error")
+    level: ApiErrorLevel = Field(description="The level of the error")
     status_code: StrictInt = Field(description="The HTTP status code")
     details: Optional[Any] = None
     __properties: ClassVar[List[str]] = [

crypticorn/hive/client/models/log_level.py

@@ -0,0 +1,38 @@
+# coding: utf-8
+
+"""
+Hive AI API
+
+API for Hive AI model training and evaluation
+
+The version of the OpenAPI document: 1.0.0
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import json
+from enum import Enum
+from typing_extensions import Self
+
+
+class LogLevel(str, Enum):
+    """
+    LogLevel
+    """
+
+    """
+    allowed enum values
+    """
+    DEBUG = "DEBUG"
+    INFO = "INFO"
+    WARNING = "WARNING"
+    ERROR = "ERROR"
+    CRITICAL = "CRITICAL"
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of LogLevel from a JSON string"""
+        return cls(json.loads(json_str))

crypticorn/hive/client/models/model.py

@@ -33,7 +33,7 @@ class Model(BaseModel):
     Pydantic model for model response
     """  # noqa: E501
 
-    model_id: StrictInt = Field(description="Unique model identifier")
+    id: StrictInt = Field(description="Unique model identifier")
     name: StrictStr = Field(description="Model name")
     coin_id: Coins = Field(description="Coin ID")
     target: Target = Field(description="Target variable")
@@ -44,7 +44,7 @@ class Model(BaseModel):
     created_at: StrictInt = Field(description="Model creation unix timestamp")
     updated_at: StrictInt = Field(description="Model update unix timestamp")
     __properties: ClassVar[List[str]] = [
-        "model_id",
+        "id",
         "name",
         "coin_id",
         "target",
@@ -113,7 +113,7 @@ class Model(BaseModel):
 
         _obj = cls.model_validate(
             {
-                "model_id": obj.get("model_id"),
+                "id": obj.get("id"),
                 "name": obj.get("name"),
                 "coin_id": obj.get("coin_id"),
                 "target": obj.get("target"),

crypticorn/hive/main.py

@@ -8,9 +8,13 @@ from crypticorn.hive import (
     DataApi,
     StatusApi,
     Configuration,
+    AdminApi,
+    DataVersion,
+    FeatureSize,
 )
 from crypticorn.hive.utils import download_file
-import logging
+from typing import Optional
+from pydantic import StrictInt
 
 
 class DataApiWrapper(DataApi):
@@ -18,7 +22,15 @@ class DataApiWrapper(DataApi):
     A wrapper for the DataApi class.
     """
 
-    async def download_data(self, folder: Path = "data", *args, **kwargs) -> list[Path]:
+    async def download_data(
+        self,
+        model_id: StrictInt,
+        folder: Path = Path("data"),
+        version: Optional[DataVersion] = None,
+        feature_size: Optional[FeatureSize] = None,
+        *args,
+        **kwargs,
+    ) -> list[Path]:
         """
         Download data for model training. All three files (y_train, x_test, x_train) are downloaded and saved under e.g. FOLDER/v1/coin_1/*.feather.
         The folder will be created if it doesn't exist.
@@ -28,7 +40,13 @@ class DataApiWrapper(DataApi):
         :param feature_size: The number of features in the data. Default is LARGE. (optional) (type: FeatureSize)
         :return: A list of paths to the downloaded files.
         """
-        response = await super().download_data(*args, **kwargs)
+        response = await super().download_data(
+            model_id=model_id,
+            version=version,
+            feature_size=feature_size,
+            *args,
+            **kwargs,
+        )
         base_path = f"{folder}/v{response.version.value}/coin_{response.coin.value}/"
         os.makedirs(base_path, exist_ok=True)
 
@@ -73,3 +91,4 @@ class HiveClient:
         self.models = ModelsApi(self.base_client)
         self.data = DataApiWrapper(self.base_client)
         self.status = StatusApi(self.base_client)
+        self.admin = AdminApi(self.base_client)

crypticorn/hive/utils.py

@@ -3,8 +3,7 @@ import os
 import tqdm
 import logging
 
-logger = logging.getLogger(__name__)
-logger.setLevel(logging.INFO)
+logger = logging.getLogger("crypticorn")
 
 
 async def download_file(

crypticorn/metrics/client/__init__.py

@@ -17,11 +17,13 @@ Do not edit the class manually.
 __version__ = "1.0.0"
 
 # import apis into sdk package
+from crypticorn.metrics.client.api.admin_api import AdminApi
 from crypticorn.metrics.client.api.exchanges_api import ExchangesApi
 from crypticorn.metrics.client.api.indicators_api import IndicatorsApi
 from crypticorn.metrics.client.api.logs_api import LogsApi
 from crypticorn.metrics.client.api.marketcap_api import MarketcapApi
 from crypticorn.metrics.client.api.markets_api import MarketsApi
+from crypticorn.metrics.client.api.quote_currencies_api import QuoteCurrenciesApi
 from crypticorn.metrics.client.api.status_api import StatusApi
 from crypticorn.metrics.client.api.tokens_api import TokensApi
 
@@ -37,7 +39,16 @@ from crypticorn.metrics.client.exceptions import ApiAttributeError
 from crypticorn.metrics.client.exceptions import ApiException
 
 # import models into sdk package
+from crypticorn.metrics.client.models.api_error_identifier import ApiErrorIdentifier
+from crypticorn.metrics.client.models.api_error_level import ApiErrorLevel
+from crypticorn.metrics.client.models.api_error_type import ApiErrorType
 from crypticorn.metrics.client.models.exception_detail import ExceptionDetail
+from crypticorn.metrics.client.models.exchange_mapping import ExchangeMapping
+from crypticorn.metrics.client.models.internal_exchange import InternalExchange
+from crypticorn.metrics.client.models.log_level import LogLevel
+from crypticorn.metrics.client.models.market_type import MarketType
+from crypticorn.metrics.client.models.marketcap_ranking import MarketcapRanking
+from crypticorn.metrics.client.models.ohlcv import OHLCV
 from crypticorn.metrics.client.models.severity import Severity
 from crypticorn.metrics.client.models.time_interval import TimeInterval
 from crypticorn.metrics.client.models.trading_status import TradingStatus

crypticorn/metrics/client/api/__init__.py

@@ -1,10 +1,12 @@
 # flake8: noqa
 
 # import apis into api package
+from crypticorn.metrics.client.api.admin_api import AdminApi
 from crypticorn.metrics.client.api.exchanges_api import ExchangesApi
 from crypticorn.metrics.client.api.indicators_api import IndicatorsApi
 from crypticorn.metrics.client.api.logs_api import LogsApi
 from crypticorn.metrics.client.api.marketcap_api import MarketcapApi
 from crypticorn.metrics.client.api.markets_api import MarketsApi
+from crypticorn.metrics.client.api.quote_currencies_api import QuoteCurrenciesApi
 from crypticorn.metrics.client.api.status_api import StatusApi
 from crypticorn.metrics.client.api.tokens_api import TokensApi