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

aicosmos_client/cli.py

@@ -7,8 +7,18 @@ from textual.app import App, ComposeResult
 from textual.containers import Container
 from textual.reactive import reactive
 from textual.screen import Screen
-from textual.widgets import (Button, Footer, Header, Input, Label, ListItem,
-                             ListView, LoadingIndicator, RichLog, Static)
+from textual.widgets import (
+    Button,
+    Footer,
+    Header,
+    Input,
+    Label,
+    ListItem,
+    ListView,
+    LoadingIndicator,
+    RichLog,
+    Static,
+)
 
 
 class LoginScreen(Screen):
@@ -166,11 +176,11 @@ class SessionScreen(Screen):
     @work(exclusive=True)
     async def load_sessions(self) -> None:
         self.query_one("#session-list").clear()
-        self.sessions, message = self.app.client.get_my_sessions()
-
-        if message != "Success":
+        try:
+            self.sessions = self.app.client.get_my_sessions()
+        except Exception as e:
             self.query_one("#session-list").append(
-                ListItem(Label(f"{message}"), id="no-sessions")
+                ListItem(Label(f"Failed: {e}"), id="no-sessions")
             )
             return
 
@@ -193,15 +203,16 @@ class SessionScreen(Screen):
     @on(Button.Pressed, "#create-session")
     @work(exclusive=True)
     async def create_session(self) -> None:
-        session_id, message = self.app.client.create_session()
+        try:
+            session_id = self.app.client.create_session()
+        except Exception as e:
+            self.query_one("#session-error").update(f"Failed to create session: {e}")
+            return
 
-        if session_id:
-            self.app.current_session = session_id
-            self.app.uninstall_screen("chat")
-            self.app.install_screen(ChatScreen(), "chat")
-            self.app.push_screen("chat")
-        else:
-            self.query_one("#session-error").update(f"Failed to create session: {message}")
+        self.app.current_session = session_id
+        self.app.uninstall_screen("chat")
+        self.app.install_screen(ChatScreen(), "chat")
+        self.app.push_screen("chat")
 
     @on(ListView.Selected)
     def session_selected(self, event: ListView.Selected) -> None:
@@ -389,11 +400,12 @@ class ChatScreen(Screen):
     @work(exclusive=True)
     async def load_conversation(self) -> None:
         """Fetch conversation history"""
-        self.conversation, message = self.app.client.get_session_history(
-            self.app.current_session
-        )
-        if message != "Success":
-            self.query_one("#chat-log").write(f"[red]Error: {message}[/red]")
+        try:
+            self.conversation = self.app.client.get_session_history(
+                self.app.current_session
+            )
+        except Exception as e:
+            self.query_one("#chat-log").write(f"[red]Error: {e}[/red]")
             return
         self.update_message_display()
 
@@ -423,12 +435,12 @@ class ChatScreen(Screen):
         send_button.disabled = True
 
         try:
-            conversation_history, message = await asyncio.to_thread(
-                self.app.client.chat, self.app.current_session, message
-            )
-
-            if message != "Success":
-                self.query_one("#chat-log").write(f"[red]Error: {message}[/red]")
+            try:
+                conversation_history = await asyncio.to_thread(
+                    self.app.client.chat, self.app.current_session, message
+                )
+            except Exception as e:
+                self.query_one("#chat-log").write(f"[red]Error: {e}[/red]")
                 return
             new_messages = conversation_history[len(self.conversation) :]
             for msg in new_messages:

aicosmos_client/client.py

@@ -1,17 +1,90 @@
+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))
 
-        login, message = self._login()
-        if not login:
-            raise ValueError(f"Failed to login. {message}")
+        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 = {
@@ -20,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 True, "Success"
-            else:
-                return False, f"Status code: {response.status_code}"
-        except Exception as e:
-            return False, 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:
@@ -42,96 +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(), "Success"
-            else:
-                return None, f"Status code: {response.status_code}"
-        except Exception as e:
-            return None, 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"], "Success"
-            else:
-                return None, f"Status code: {response.status_code}"
-        except Exception as e:
-            return None, 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 True, "Success"
-            else:
-                return False, f"Status code: {response.status_code}"
-        except Exception as e:
-            return False, 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
-                ], "Success"
-            else:
-                return None, f"Status code: {response.status_code}"
-        except Exception as e:
-            return None, 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, message = self._get_session_status(session_id)
-        if not session:
-            return [], message
-        else:
-            return session.get("conversation", []), message
+        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"], "Success"
-            else:
-                return [], f"Status code: {response.status_code}"
-        except Exception as e:
-            return [], 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.6.dist-info/METADATA

@@ -0,0 +1,70 @@
+Metadata-Version: 2.4
+Name: aicosmos_client
+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
+Author-email: Example Author <author@example.com>
+License-Expression: MIT
+License-File: LICENSE
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Requires-Python: >=3.9
+Requires-Dist: requests>=2.25.0
+Requires-Dist: textual>=5.0.0
+Description-Content-Type: text/markdown
+
+# Client for AICosmos
+
+This package implements the client for AICosmos. Before using this package, please make sure that you have a valid account for AICosmos. 
+
+### AICosmosClient
+By using this client, you can chat with our backend in "base" mode. To login, you will need the server's address, your username and your password. You can either start a new session, or use an existing one.
+
+Our framework is a little bit different from "chat completions", where you give an llm the conversation history. Instead, your conversation history, along with other tool execution results, are stored in our database. This gives your a clean and simple interface to use, without worrying about constructing complicated contexts.
+
+```Python
+from aicosmos_client.client import AICosmosClient 
+
+# login
+client = AICosmosClient(
+    base_url="https://aicosmos.ai/api",
+    username="xxx",
+    password="xxx",
+    auto_trust=True,
+)
+
+# create a new session
+try:
+    new_session_id = client.create_session()
+except Exception as e:
+    print(f"Error creating new session: {e}")
+    exit(0)
+
+# lookup all the sessions
+try:
+    my_sessions = client.get_my_sessions()
+except Exception as e:
+    print(f"Error getting my sessions: {e}")
+    exit(0)
+# [{"session_id", "title"}, ...]
+print(my_sessions)
+
+# enjoy the conversation
+try:
+    conversation_history = client.chat(new_session_id, "Hello")
+except Exception as e:
+    print(f"Error chatting: {e}")
+    exit(0)
+print(conversation_history)
+```
+
+## AICosmosCLI
+To show that the client is enough to build an application, we offer you an command-line interface!
+
+```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.4.dist-info/METADATA

@@ -1,21 +0,0 @@
-Metadata-Version: 2.4
-Name: aicosmos_client
-Version: 0.0.4
-Summary: client for AICosmos platform
-Project-URL: Homepage, https://github.com/pypa/sampleproject
-Project-URL: Issues, https://github.com/pypa/sampleproject/issues
-Author-email: Example Author <author@example.com>
-License-Expression: MIT
-License-File: LICENSE
-Classifier: Operating System :: OS Independent
-Classifier: Programming Language :: Python :: 3
-Requires-Python: >=3.9
-Requires-Dist: requests>=2.25.0
-Requires-Dist: textual>=5.0.0
-Description-Content-Type: text/markdown
-
-# Example Package
-
-This is a simple example package. You can use
-[GitHub-flavored Markdown](https://guides.github.com/features/mastering-markdown/)
-to write your content.

aicosmos_client-0.0.4.dist-info/RECORD

@@ -1,7 +0,0 @@
-aicosmos_client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-aicosmos_client/cli.py,sha256=Dv9uNOcc0rMXjCVfaElDxS8p1mnDwFfTt_vnpAwLUuI,13940
-aicosmos_client/client.py,sha256=SSR7THQDvSj9Bnm2ytdJ5Twe3Vy8YdGslbmx71UgLw4,4870
-aicosmos_client-0.0.4.dist-info/METADATA,sha256=a85Gf6H1HTMtUY-YvMuzm_QFnDKT48L8Ngwp_1iwq64,712
-aicosmos_client-0.0.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-aicosmos_client-0.0.4.dist-info/licenses/LICENSE,sha256=XBdpsYae127l7YQyMSVQwUUo22FPis7sMts7oBjkN_g,1056
-aicosmos_client-0.0.4.dist-info/RECORD,,