aicosmos-client 0.0.5__py3-none-any.whl → 0.0.6__py3-none-any.whl

aicosmos_client/client.py

@@ -1,16 +1,91 @@
+import os
+import socket
+import ssl
+from urllib.parse import urlparse
+
+import certifi
 import requests
+from requests.adapters import HTTPAdapter
+
+
+class SSLAdapter(HTTPAdapter):
+    """HTTPS adapter that allows dynamic CA bundle injection."""
+
+    def __init__(self, cafile=None, *args, **kwargs):
+        self.cafile = cafile
+        super().__init__(*args, **kwargs)
+
+    def init_poolmanager(self, *args, **kwargs):
+        context = ssl.create_default_context()
+        context.load_verify_locations(cafile=certifi.where())
+        if self.cafile and os.path.exists(self.cafile):
+            context.load_verify_locations(cafile=self.cafile)
+        kwargs["ssl_context"] = context
+        return super().init_poolmanager(*args, **kwargs)
 
 
 class AICosmosClient:
-    def __init__(self, base_url: str, username: str, password: str):
-        self.session = requests.Session()
-        self.base_url = base_url
-        self.username: str = username
-        self.password: str = password
+    def __init__(
+        self,
+        base_url: str,
+        username: str,
+        password: str,
+        certs_dir: str = None,
+        auto_trust: bool = False,
+    ):
+        """
+        :param base_url: API base URL, e.g. 'https://aicosmos.ai/api'
+        :param username: Username for login
+        :param password: Password for login
+        :param certs_dir: Directory for storing trusted self-signed certs
+        :param auto_trust: If True, will automatically trust self-signed certs
+        """
+        self.base_url = base_url.rstrip("/")
+        self.username = username
+        self.password = password
         self.access_token: str = None
+        self.auto_trust = auto_trust
+
+        host = urlparse(self.base_url).hostname
+        self.certs_dir = certs_dir or os.path.join(
+            os.path.expanduser("~"), ".aicosmos", "certs"
+        )
+        os.makedirs(self.certs_dir, exist_ok=True)
+        self.cert_file = os.path.join(self.certs_dir, f"{host}.pem")
+
+        self.session = requests.Session()
+        self.session.mount("https://", SSLAdapter(cafile=self.cert_file))
 
         self._login()
 
+    def _fetch_server_cert(self, hostname, port=443):
+        """Download server's SSL certificate and save locally."""
+        pem_path = self.cert_file
+        conn = socket.create_connection((hostname, port))
+        context = ssl.SSLContext(ssl.PROTOCOL_TLS_CLIENT)
+        context.check_hostname = False
+        context.verify_mode = ssl.CERT_NONE
+        with context.wrap_socket(conn, server_hostname=hostname) as sock:
+            der_cert = sock.getpeercert(True)
+            pem_cert = ssl.DER_cert_to_PEM_cert(der_cert)
+            with open(pem_path, "w") as f:
+                f.write(pem_cert)
+        return pem_path
+
+    def _robust_request(self, method, url, **kwargs):
+        try:
+            return self.session.request(method, url, **kwargs)
+        except requests.exceptions.SSLError as e:
+            if not self.auto_trust:
+                raise RuntimeError(
+                    f"SSL verification failed for {url}. "
+                    f"Set auto_trust=True to accept and store the server's certificate."
+                ) from e
+            host = urlparse(self.base_url).hostname
+            self._fetch_server_cert(host)
+            self.session.mount("https://", SSLAdapter(cafile=self.cert_file))
+            return self.session.request(method, url, **kwargs)
+
     def _login(self):
         login_data = {
             "username": self.username,
@@ -18,18 +93,14 @@ class AICosmosClient:
             "grant_type": "password",
         }
         headers = {"Content-Type": "application/x-www-form-urlencoded"}
-        try:
-            response = self.session.post(
-                f"{self.base_url}/user/login", data=login_data, headers=headers
-            )
-            if response.status_code == 200:
-                token_data = response.json()
-                self.access_token = token_data["access_token"]
-                return
-            else:
-                raise ValueError(f"Status code: {response.status_code}")
-        except Exception as e:
-            raise ValueError(f"Error: {e}")
+        response = self._robust_request(
+            "POST", f"{self.base_url}/user/login", data=login_data, headers=headers
+        )
+        if response.status_code == 200:
+            token_data = response.json()
+            self.access_token = token_data["access_token"]
+        else:
+            raise ValueError(f"Login failed: {response.status_code} {response.text}")
 
     def _get_auth_headers(self):
         if not self.access_token:
@@ -40,93 +111,68 @@ class AICosmosClient:
         }
 
     def _get_session_status(self, session_id):
-        try:
-            response = self.session.get(
-                f"{self.base_url}/sessions/{session_id}/status",
-                headers=self._get_auth_headers(),
-            )
-            success = response.status_code == 200
-            if success:
-                return response.json()
-            else:
-                raise ValueError(f"Status code: {response.status_code}")
-        except Exception as e:
-            raise ValueError(f"Error: {e}")
+        response = self._robust_request(
+            "GET",
+            f"{self.base_url}/sessions/{session_id}/status",
+            headers=self._get_auth_headers(),
+        )
+        if response.status_code == 200:
+            return response.json()
+        else:
+            raise ValueError(f"Status code: {response.status_code}")
 
     def create_session(self):
-        if not self.access_token:
-            raise ValueError("Not logged in")
-        try:
-            response = self.session.post(
-                f"{self.base_url}/sessions/create", headers=self._get_auth_headers()
-            )
-            if response.status_code == 200:
-                response_json = response.json()
-                return response_json["session_id"]
-            else:
-                raise ValueError(f"Status code: {response.status_code}")
-        except Exception as e:
-            raise ValueError(f"Error: {e}")
+        response = self._robust_request(
+            "POST",
+            f"{self.base_url}/sessions/create",
+            headers=self._get_auth_headers(),
+        )
+        if response.status_code == 200:
+            return response.json()["session_id"]
+        else:
+            raise ValueError(f"Status code: {response.status_code}")
 
     def delete_session(self, session_id: str):
-        if not self.access_token:
-            raise ValueError("Not logged in")
-        try:
-            response = self.session.delete(
-                f"{self.base_url}/sessions/{session_id}",
-                headers=self._get_auth_headers(),
-            )
-            if response.status_code == 200:
-                return
-            else:
-                raise ValueError(f"Status code: {response.status_code}")
-        except Exception as e:
-            raise ValueError(f"Error: {e}")
+        response = self._robust_request(
+            "DELETE",
+            f"{self.base_url}/sessions/{session_id}",
+            headers=self._get_auth_headers(),
+        )
+        if response.status_code != 200:
+            raise ValueError(f"Status code: {response.status_code}")
 
     def get_my_sessions(self):
-        if not self.access_token:
-            raise ValueError("Not logged in")
-        try:
-            response = self.session.get(
-                f"{self.base_url}/sessions/my_sessions",
-                headers=self._get_auth_headers(),
-            )
-            if response.status_code == 200:
-                sessions = response.json()
-                self.active_sessions = sessions
-                return [
-                    {
-                        "session_id": session["session_id"],
-                        "title": session["environment_info"].get("title", None),
-                    }
-                    for session in sessions
-                ]
-            else:
-                raise ValueError(f"Status code: {response.status_code}")
-        except Exception as e:
-            raise ValueError(f"Error: {e}")
+        response = self._robust_request(
+            "GET",
+            f"{self.base_url}/sessions/my_sessions",
+            headers=self._get_auth_headers(),
+        )
+        if response.status_code == 200:
+            sessions = response.json()
+            self.active_sessions = sessions
+            return [
+                {
+                    "session_id": s["session_id"],
+                    "title": s["environment_info"].get("title"),
+                }
+                for s in sessions
+            ]
+        else:
+            raise ValueError(f"Status code: {response.status_code}")
 
     def get_session_history(self, session_id: str):
         session = self._get_session_status(session_id)
         return session.get("conversation", [])
 
     def chat(self, session_id: str, prompt: str):
-        if not self.access_token:
-            raise ValueError("Not logged in")
-        data = {
-            "user_input": prompt,
-            "session_id": session_id,
-        }
-        try:
-            response = self.session.post(
-                f"{self.base_url}/chat",
-                json=data,
-                headers=self._get_auth_headers(),
-            )
-            success = response.status_code == 200
-            if success:
-                return response.json()["conversation_history"]
-            else:
-                raise ValueError(f"Status code: {response.status_code}")
-        except Exception as e:
-            raise ValueError(f"Error: {e}")
+        data = {"user_input": prompt, "session_id": session_id}
+        response = self._robust_request(
+            "POST",
+            f"{self.base_url}/chat",
+            json=data,
+            headers=self._get_auth_headers(),
+        )
+        if response.status_code == 200:
+            return response.json()["conversation_history"]
+        else:
+            raise ValueError(f"Status code: {response.status_code}")

{aicosmos_client-0.0.5.dist-info → aicosmos_client-0.0.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aicosmos_client
-Version: 0.0.5
+Version: 0.0.6
 Summary: client for AICosmos platform
 Project-URL: Homepage, https://github.com/pypa/sampleproject
 Project-URL: Issues, https://github.com/pypa/sampleproject/issues
@@ -28,7 +28,10 @@ from aicosmos_client.client import AICosmosClient
 
 # login
 client = AICosmosClient(
-    base_url="http://xxx.xxx", username="xxx", password="xxxxxx"
+    base_url="https://aicosmos.ai/api",
+    username="xxx",
+    password="xxx",
+    auto_trust=True,
 )
 
 # create a new session
@@ -62,5 +65,6 @@ To show that the client is enough to build an application, we offer you an comma
 ```Python
 from aicosmos_client.cli import AICosmosCLI
 
+# url: https://aicosmos.ai/api
 AICosmosCLI().run()
 ```

aicosmos_client-0.0.6.dist-info/RECORD

@@ -0,0 +1,7 @@
+aicosmos_client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+aicosmos_client/cli.py,sha256=V6fewmdU2Im3NNbVw0OrVbxd1mkTGYKTdfgNKJl0CwY,13979
+aicosmos_client/client.py,sha256=dn58ztTJVSOP9munQcYVB8AAhUiTsJoNnPvngQceSGc,6388
+aicosmos_client-0.0.6.dist-info/METADATA,sha256=EsoT3FQDaqbmGH2bEdQCCcCAUkKQKz8e4xxH74d4Sj0,2242
+aicosmos_client-0.0.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+aicosmos_client-0.0.6.dist-info/licenses/LICENSE,sha256=XBdpsYae127l7YQyMSVQwUUo22FPis7sMts7oBjkN_g,1056
+aicosmos_client-0.0.6.dist-info/RECORD,,

aicosmos_client-0.0.5.dist-info/RECORD

@@ -1,7 +0,0 @@
-aicosmos_client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-aicosmos_client/cli.py,sha256=V6fewmdU2Im3NNbVw0OrVbxd1mkTGYKTdfgNKJl0CwY,13979
-aicosmos_client/client.py,sha256=akVMmyDrAqOg0JI4mv_MBIA6I3VhLwDkstJlTEwiF5w,4664
-aicosmos_client-0.0.5.dist-info/METADATA,sha256=7mdmOAHYtWfyEuYOiVsJPiDT4PzPBBF3LyZ83L-rC6c,2175
-aicosmos_client-0.0.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-aicosmos_client-0.0.5.dist-info/licenses/LICENSE,sha256=XBdpsYae127l7YQyMSVQwUUo22FPis7sMts7oBjkN_g,1056
-aicosmos_client-0.0.5.dist-info/RECORD,,