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

aicosmos_client/cli.py

@@ -1,6 +1,7 @@
 import asyncio
 from typing import Dict
 
+from client import AICosmosClient
 from textual import on, work
 from textual.app import App, ComposeResult
 from textual.containers import Container
@@ -18,7 +19,6 @@ from textual.widgets import (
     RichLog,
     Static,
 )
-from .client import AICosmosClient
 
 
 class LoginScreen(Screen):
@@ -176,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,9 +193,9 @@ class SessionScreen(Screen):
         for session in self.sessions:
             session_id = session["session_id"]
             clean_id = self.sanitize_id(session_id)
-            title = session["environment_info"].get(
-                "title", f"Session {session_id[:6]}"
-            )
+            title = session["title"]
+            if not title:
+                title = f"Session {session_id[:6]}"
             self.query_one("#session-list").append(
                 ListItem(Label(title), id=f"session-{clean_id}")
             )
@@ -203,15 +203,16 @@ class SessionScreen(Screen):
     @on(Button.Pressed, "#create-session")
     @work(exclusive=True)
     async def create_session(self) -> None:
-        session_id = 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("Failed to create session")
+        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:
@@ -399,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()
 
@@ -433,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

@@ -9,9 +9,7 @@ class AICosmosClient:
         self.password: str = password
         self.access_token: str = None
 
-        login, message = self._login()
-        if not login:
-            raise ValueError(f"Failed to login. {message}")
+        self._login()
 
     def _login(self):
         login_data = {
@@ -27,11 +25,11 @@ class AICosmosClient:
             if response.status_code == 200:
                 token_data = response.json()
                 self.access_token = token_data["access_token"]
-                return True, "Success"
+                return
             else:
-                return False, f"Status code: {response.status_code}"
+                raise ValueError(f"Status code: {response.status_code}")
         except Exception as e:
-            return False, f"Error: {e}"
+            raise ValueError(f"Error: {e}")
 
     def _get_auth_headers(self):
         if not self.access_token:
@@ -49,11 +47,11 @@ class AICosmosClient:
             )
             success = response.status_code == 200
             if success:
-                return response.json(), "Success"
+                return response.json()
             else:
-                return None, f"Status code: {response.status_code}"
+                raise ValueError(f"Status code: {response.status_code}")
         except Exception as e:
-            return None, f"Error: {e}"
+            raise ValueError(f"Error: {e}")
 
     def create_session(self):
         if not self.access_token:
@@ -64,11 +62,11 @@ class AICosmosClient:
             )
             if response.status_code == 200:
                 response_json = response.json()
-                return response_json["session_id"], "Success"
+                return response_json["session_id"]
             else:
-                return None, f"Status code: {response.status_code}"
+                raise ValueError(f"Status code: {response.status_code}")
         except Exception as e:
-            return None, f"Error: {e}"
+            raise ValueError(f"Error: {e}")
 
     def delete_session(self, session_id: str):
         if not self.access_token:
@@ -79,11 +77,11 @@ class AICosmosClient:
                 headers=self._get_auth_headers(),
             )
             if response.status_code == 200:
-                return True, "Success"
+                return
             else:
-                return False, f"Status code: {response.status_code}"
+                raise ValueError(f"Status code: {response.status_code}")
         except Exception as e:
-            return False, f"Error: {e}"
+            raise ValueError(f"Error: {e}")
 
     def get_my_sessions(self):
         if not self.access_token:
@@ -99,21 +97,18 @@ class AICosmosClient:
                 return [
                     {
                         "session_id": session["session_id"],
-                        "title": session["environment_info"].get("title", "unknown"),
+                        "title": session["environment_info"].get("title", None),
                     }
                     for session in sessions
-                ], "Success"
+                ]
             else:
-                return None, f"Status code: {response.status_code}"
+                raise ValueError(f"Status code: {response.status_code}")
         except Exception as e:
-            return None, f"Error: {e}"
+            raise ValueError(f"Error: {e}")
 
     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:
@@ -130,8 +125,8 @@ class AICosmosClient:
             )
             success = response.status_code == 200
             if success:
-                return response.json()["conversation_history"], "Success"
+                return response.json()["conversation_history"]
             else:
-                return [], f"Status code: {response.status_code}"
+                raise ValueError(f"Status code: {response.status_code}")
         except Exception as e:
-            return [], f"Error: {e}"
+            raise ValueError(f"Error: {e}")

aicosmos_client-0.0.5.dist-info/METADATA

@@ -0,0 +1,66 @@
+Metadata-Version: 2.4
+Name: aicosmos_client
+Version: 0.0.5
+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="http://xxx.xxx", username="xxx", password="xxxxxx"
+)
+
+# 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
+
+AICosmosCLI().run()
+```

aicosmos_client-0.0.5.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=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,,

aicosmos_client-0.0.3.dist-info/METADATA

@@ -1,21 +0,0 @@
-Metadata-Version: 2.4
-Name: aicosmos_client
-Version: 0.0.3
-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.3.dist-info/RECORD

@@ -1,7 +0,0 @@
-aicosmos_client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-aicosmos_client/cli.py,sha256=8VzGfvUGPmEbnbK_yqC3-UEcTMCMg8xs_Zy-xQzeDj0,13939
-aicosmos_client/client.py,sha256=t3qlfanWG_NngNp9VPwHcJh_JC-2sQR-RRMvXWVQVcM,4875
-aicosmos_client-0.0.3.dist-info/METADATA,sha256=mhPwxEWtskK0Umu592yqdgTMf-gstv7zjzL-KuSs17w,712
-aicosmos_client-0.0.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-aicosmos_client-0.0.3.dist-info/licenses/LICENSE,sha256=XBdpsYae127l7YQyMSVQwUUo22FPis7sMts7oBjkN_g,1056
-aicosmos_client-0.0.3.dist-info/RECORD,,