aiauto-client 0.1.7__py3-none-any.whl → 0.1.9__py3-none-any.whl

aiauto/core.py

@@ -31,8 +31,16 @@ class AIAutoController:
             if not host_external:
                 raise RuntimeError("No storage host returned from EnsureWorkspace")
 
-            host, port = host_external.split(':')
-            self.storage = optuna.storages.GrpcStorageProxy(host=host, port=int(port))
+            # Parse host and port
+            if ':' in host_external:
+                host, port_str = host_external.rsplit(':', 1)
+                port = int(port_str)
+            else:
+                host = host_external
+                port = 443  # Default to HTTPS port
+            
+            # Create storage with TLS support for port 443
+            self.storage = self._create_storage_with_tls(host, port)
 
             # Store the internal host for CRD usage (if needed later)
             self.storage_host_internal = response.get('journalGrpcStorageProxyHostInternal', '')
@@ -49,6 +57,74 @@ class AIAutoController:
         self.artifact_store = optuna.artifacts.FileSystemArtifactStore('./artifacts')
         self.tmp_dir = tempfile.mkdtemp(prefix=f'ai_auto_tmp_')
 
+    def _create_storage_with_tls(self, host: str, port: int):
+        """Create GrpcStorageProxy with automatic TLS detection based on port"""
+        # Port 13000 = internal (plain), Port 443 = external (TLS)
+        if port != 443:
+            # Plain gRPC for internal connections
+            return optuna.storages.GrpcStorageProxy(host=host, port=port)
+        
+        # TLS connection for external access (port 443)
+        import grpc
+        creds = grpc.ssl_channel_credentials()
+        
+        # Try different TLS parameter names for Optuna version compatibility
+        # Try 1: channel_credentials parameter (newer Optuna versions)
+        try:
+            return optuna.storages.GrpcStorageProxy(
+                host=host, 
+                port=port, 
+                channel_credentials=creds
+            )
+        except TypeError:
+            pass
+        
+        # Try 2: ssl boolean parameter
+        try:
+            return optuna.storages.GrpcStorageProxy(
+                host=host, 
+                port=port, 
+                ssl=True
+            )
+        except TypeError:
+            pass
+        
+        # Try 3: use_tls parameter
+        try:
+            return optuna.storages.GrpcStorageProxy(
+                host=host,
+                port=port,
+                use_tls=True
+            )
+        except TypeError:
+            pass
+        
+        # Try 4: secure parameter
+        try:
+            return optuna.storages.GrpcStorageProxy(
+                host=host,
+                port=port,
+                secure=True
+            )
+        except TypeError:
+            pass
+        
+        # Fallback: try to create secure channel manually
+        try:
+            channel = grpc.secure_channel(f"{host}:{port}", creds)
+            # Some Optuna versions might accept a channel directly
+            return optuna.storages.GrpcStorageProxy(channel=channel)
+        except (TypeError, AttributeError):
+            pass
+        
+        # If all attempts fail, raise informative error
+        raise RuntimeError(
+            f"Failed to create TLS connection to {host}:{port}. "
+            "GrpcStorageProxy TLS parameters not recognized. "
+            "Please check Optuna version compatibility. "
+            "Tried: channel_credentials, ssl, use_tls, secure, channel parameters."
+        )
+    
     def get_storage(self):
         return self.storage
 
@@ -172,7 +248,7 @@ class StudyWrapper:
                 self._study = optuna.create_study(
                     study_name=self.study_name,
                     storage=self._storage,
-                    load_if_exists=True
+                    load_if_exists=True,
                 )
             except Exception as e:
                 raise RuntimeError(
@@ -202,7 +278,8 @@ class StudyWrapper:
             request_data = {
                 "objective": {
                     "sourceCode": serialize(objective),
-                    "requirementsTxt": build_requirements(requirements_file, requirements_list)
+                    "requirementsTxt": build_requirements(requirements_file, requirements_list),
+                    "objectiveName": objective.__name__  # 함수명 추출하여 전달
                 },
                 "batch": {
                     "studyName": self.study_name,

{aiauto_client-0.1.7.dist-info → aiauto_client-0.1.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: aiauto-client
-Version: 0.1.7
+Version: 0.1.9
 Summary: AI Auto HPO (Hyperparameter Optimization) Client Library
 Author-email: AIAuto Team <ainode@zeroone.ai>
 Project-URL: Homepage, https://dashboard.aiauto.pangyo.ainode.ai
@@ -94,6 +94,17 @@ study_wrapper.optimize(
     runtime_image=aiauto.RUNTIME_IMAGES[0],
 )
 ```
+- 종료 됐는지 optuna-dashboard 가 아닌 코드로 확인하는 법
+```python
+study_wrapper.get_status()
+# {'study_name': 'test', 'count_active': 0, 'count_succeeded': 10, 'count_pruned': 0, 'count_failed': 0, 'count_total': 10, 'count_completed': 10, 'dashboard_url': 'https://optuna-dashboard-10f804bb-52be-48e8-aa06-9f5411ed4b0d.aiauto.pangyo.ainode.ai', 'last_error': '', 'updated_at': '2025-09-01T11:31:49.375Z'}
+while study_wrapper.get_status()['count_completed'] <= study_wrapper.get_status()['count_total']:
+    sleep(10)  # 10초 마다 한 번 씩
+```
+- best trial 을 가져오는 법
+```python
+TODO
+```
 
 ## Jupyter Notebook 사용 시 주의사항
 

{aiauto_client-0.1.7.dist-info → aiauto_client-0.1.9.dist-info}/RECORD

@@ -1,10 +1,10 @@
 aiauto/__init__.py,sha256=sF7sJaXg7-MqolSYLxsaXAir1dBzARhXLrHo7zLsupg,345
 aiauto/_config.py,sha256=DaRTIZlph9T3iuW-Cy4fkw8i3bXB--gMtW947SLZZNs,159
 aiauto/constants.py,sha256=rBibGOQHHrdkwaai92-3I8-N0cu-B4CoCoQbG9-Cl8k,821
-aiauto/core.py,sha256=DQW9uvVNqP9J9s1IFx969upw10NQLHJMNsudHnauk6A,9840
+aiauto/core.py,sha256=caGUQEi4KTaov7Brak7o3O_27LZDNdYqUWmXDpusQaU,12553
 aiauto/http_client.py,sha256=t1gxeM5-d5bsVoFWgaNcTrt_WWUXuMuxge9gDlEqhoA,2086
 aiauto/serializer.py,sha256=KqQeH0xp4LQuZE6r8kzXQsWY6QgC3hqn8MSuWTt4QmU,1938
-aiauto_client-0.1.7.dist-info/METADATA,sha256=xy8ZGkJYOwMUgA_8J4ag2mepwBzGZQwsyIOkMgQ8Cpw,18450
-aiauto_client-0.1.7.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
-aiauto_client-0.1.7.dist-info/top_level.txt,sha256=Sk2ctO9_Bf_tAPwq1x6Vfl6OuL29XzwMTO4F_KG6oJE,7
-aiauto_client-0.1.7.dist-info/RECORD,,
+aiauto_client-0.1.9.dist-info/METADATA,sha256=sIl-_c3kOSgxyO2wL3apNAK5_5t5KRZoBPCIeBkoxz8,19071
+aiauto_client-0.1.9.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
+aiauto_client-0.1.9.dist-info/top_level.txt,sha256=Sk2ctO9_Bf_tAPwq1x6Vfl6OuL29XzwMTO4F_KG6oJE,7
+aiauto_client-0.1.9.dist-info/RECORD,,