aiauto-client 0.1.3__tar.gz → 0.1.5__tar.gz

{aiauto_client-0.1.3 → aiauto_client-0.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: aiauto-client
-Version: 0.1.3
+Version: 0.1.5
 Summary: AI Auto HPO (Hyperparameter Optimization) Client Library
 Author-email: AIAuto Team <ainode@zeroone.ai>
 Project-URL: Homepage, https://aiauto.cloude.ainode.ai
@@ -31,15 +31,8 @@ AIAuto는 Kubernetes 기반의 분산 HPO(Hyperparameter Optimization) 시스템
 사용자 python lib <-> Next.js 서버 사이 gRPC 통신 담당
 
 ## lib build
-- pypi build, upload 종속성 다운로드 `uv add --dev twine`
-- build lib `uv build`
-  - `aiauto_client-0.1.1-py3-none-any.whl` 생성
-  - `aiauto_client-0.1.1.tar.gz` 생성
-  - `aiauto_client.egg-info` 생성
-- `~/.pypirc` 파일에 설정 확인
-- `uv run twine upload --repository aiauto-client dist/*`
-  - upload 시 pypi token 을 입력하라고 나옴, pypi 로그인 계정 설정가면 있다 
-
+- `make build push` 
+  
 ## 설치
 - `uv add aiauto-client`
 

{aiauto_client-0.1.3 → aiauto_client-0.1.5}/README.md

@@ -4,15 +4,8 @@ AIAuto는 Kubernetes 기반의 분산 HPO(Hyperparameter Optimization) 시스템
 사용자 python lib <-> Next.js 서버 사이 gRPC 통신 담당
 
 ## lib build
-- pypi build, upload 종속성 다운로드 `uv add --dev twine`
-- build lib `uv build`
-  - `aiauto_client-0.1.1-py3-none-any.whl` 생성
-  - `aiauto_client-0.1.1.tar.gz` 생성
-  - `aiauto_client.egg-info` 생성
-- `~/.pypirc` 파일에 설정 확인
-- `uv run twine upload --repository aiauto-client dist/*`
-  - upload 시 pypi token 을 입력하라고 나옴, pypi 로그인 계정 설정가면 있다 
-
+- `make build push` 
+  
 ## 설치
 - `uv add aiauto-client`
 

{aiauto_client-0.1.3 → aiauto_client-0.1.5}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "aiauto-client"
-version = "0.1.3"
+version = "0.1.5"
 description = "AI Auto HPO (Hyperparameter Optimization) Client Library"
 readme = "README.md"
 requires-python = ">=3.8"

{aiauto_client-0.1.3 → aiauto_client-0.1.5}/src/aiauto/__init__.py

@@ -1,5 +1,4 @@
 from .core import AIAutoController, TrialController, CallbackTopNArtifact, StudyWrapper
-from .api import create_study
 from ._config import AIAUTO_API_TARGET
 from .constants import RUNTIME_IMAGES
 
@@ -10,7 +9,6 @@ __all__ = [
     'TrialController', 
     'CallbackTopNArtifact',
     'StudyWrapper',
-    'create_study',
     'AIAUTO_API_TARGET',
     'RUNTIME_IMAGES',
 ]

{aiauto_client-0.1.3 → aiauto_client-0.1.5}/src/aiauto/core.py

@@ -3,7 +3,7 @@ import tempfile
 from typing import Union, Optional, List, Dict, Callable
 import optuna
 from .http_client import ConnectRPCClient
-from .serializer import serialize, build_requirements
+from .serializer import serialize, build_requirements, object_to_json
 from ._config import AIAUTO_API_TARGET
 
 
@@ -39,7 +39,10 @@ class AIAutoController:
             self.dashboard_url = response.get('dashboardUrl', '')
 
         except Exception as e:
-            raise RuntimeError(f"Failed to initialize workspace: {e}") from e
+            raise RuntimeError(
+                f"Failed to initialize workspace: {e}\n"
+                "Please delete and reissue your token from the web dashboard at https://dashboard.aiauto.pangyo.ainode.ai"
+            ) from e
 
         # artifact storage
         makedirs('./artifacts', exist_ok=True)
@@ -59,6 +62,46 @@ class AIAutoController:
     def get_artifact_tmp_dir(self):
         return self.tmp_dir
 
+    def create_study(
+        self,
+        study_name: str,
+        direction: Optional[str] = None,
+        directions: Optional[List[str]] = None,
+        sampler: Union[object, dict, None] = None,
+        pruner: Union[object, dict, None] = None
+    ) -> 'StudyWrapper':
+        """Create a new study using the controller's token."""
+        if not direction and not directions:
+            raise ValueError("Either 'direction' or 'directions' must be specified")
+
+        if direction and directions:
+            raise ValueError("Cannot specify both 'direction' and 'directions'")
+
+        try:
+            # Prepare request data for CreateStudy
+            request_data = {
+                "spec": {
+                    "studyName": study_name,
+                    "direction": direction or "",
+                    "directions": directions or [],
+                    "samplerJson": object_to_json(sampler),
+                    "prunerJson": object_to_json(pruner)
+                }
+            }
+            
+            # Call CreateStudy RPC
+            response = self.client.call_rpc("CreateStudy", request_data)
+            
+            # Return StudyWrapper
+            return StudyWrapper(
+                study_name=response.get("studyName", study_name),
+                storage=self.storage,
+                controller=self
+            )
+
+        except Exception as e:
+            raise RuntimeError(f"Failed to create study: {e}") from e
+
 
 class TrialController:
     def __init__(self, trial: optuna.trial.Trial):
@@ -132,7 +175,10 @@ class StudyWrapper:
                     load_if_exists=True
                 )
             except Exception as e:
-                raise RuntimeError("Study not ready. Call get_status() and wait for phase=Ready.") from e
+                raise RuntimeError(
+                    "Failed to get study. If this persists, please delete and reissue your token "
+                    "from the web dashboard at https://dashboard.aiauto.pangyo.ainode.ai"
+                ) from e
         return self._study
 
     def optimize(

{aiauto_client-0.1.3 → aiauto_client-0.1.5}/src/aiauto_client.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: aiauto-client
-Version: 0.1.3
+Version: 0.1.5
 Summary: AI Auto HPO (Hyperparameter Optimization) Client Library
 Author-email: AIAuto Team <ainode@zeroone.ai>
 Project-URL: Homepage, https://aiauto.cloude.ainode.ai
@@ -31,15 +31,8 @@ AIAuto는 Kubernetes 기반의 분산 HPO(Hyperparameter Optimization) 시스템
 사용자 python lib <-> Next.js 서버 사이 gRPC 통신 담당
 
 ## lib build
-- pypi build, upload 종속성 다운로드 `uv add --dev twine`
-- build lib `uv build`
-  - `aiauto_client-0.1.1-py3-none-any.whl` 생성
-  - `aiauto_client-0.1.1.tar.gz` 생성
-  - `aiauto_client.egg-info` 생성
-- `~/.pypirc` 파일에 설정 확인
-- `uv run twine upload --repository aiauto-client dist/*`
-  - upload 시 pypi token 을 입력하라고 나옴, pypi 로그인 계정 설정가면 있다 
-
+- `make build push` 
+  
 ## 설치
 - `uv add aiauto-client`
 

{aiauto_client-0.1.3 → aiauto_client-0.1.5}/src/aiauto_client.egg-info/SOURCES.txt

@@ -2,7 +2,6 @@ README.md
 pyproject.toml
 src/aiauto/__init__.py
 src/aiauto/_config.py
-src/aiauto/api.py
 src/aiauto/constants.py
 src/aiauto/core.py
 src/aiauto/http_client.py

aiauto_client-0.1.3/src/aiauto/api.py

@@ -1,46 +0,0 @@
-from typing import Optional, List, Union
-from .serializer import object_to_json
-from .core import StudyWrapper, AIAutoController
-
-
-def create_study(
-    study_name: str,
-    token: str,
-    direction: Optional[str] = None,
-    directions: Optional[List[str]] = None,
-    sampler: Union[object, dict, None] = None,
-    pruner: Union[object, dict, None] = None
-) -> StudyWrapper:
-    if not direction and not directions:
-        raise ValueError("Either 'direction' or 'directions' must be specified")
-
-    if direction and directions:
-        raise ValueError("Cannot specify both 'direction' and 'directions'")
-
-    try:
-        # Initialize controller (which ensures workspace)
-        controller = AIAutoController(token)
-        
-        # Prepare request data for CreateStudy
-        request_data = {
-            "spec": {
-                "studyName": study_name,
-                "direction": direction or "",
-                "directions": directions or [],
-                "samplerJson": object_to_json(sampler),
-                "prunerJson": object_to_json(pruner)
-            }
-        }
-        
-        # Call CreateStudy RPC
-        response = controller.client.call_rpc("CreateStudy", request_data)
-        
-        # Return StudyWrapper
-        return StudyWrapper(
-            study_name=response.get("studyName", study_name),
-            storage=controller.storage,
-            controller=controller
-        )
-
-    except Exception as e:
-        raise RuntimeError(f"Failed to create study: {e}") from e