aicosmos-client 0.0.3__tar.gz → 0.0.4__tar.gz

{aicosmos_client-0.0.3 → aicosmos_client-0.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aicosmos_client
-Version: 0.0.3
+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

{aicosmos_client-0.0.3 → aicosmos_client-0.0.4}/aicosmos_client/cli.py

@@ -1,24 +1,14 @@
 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
 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 .client import AICosmosClient
+from textual.widgets import (Button, Footer, Header, Input, Label, ListItem,
+                             ListView, LoadingIndicator, RichLog, Static)
 
 
 class LoginScreen(Screen):
@@ -178,7 +168,7 @@ class SessionScreen(Screen):
         self.query_one("#session-list").clear()
         self.sessions, message = self.app.client.get_my_sessions()
 
-        if message != "success":
+        if message != "Success":
             self.query_one("#session-list").append(
                 ListItem(Label(f"{message}"), id="no-sessions")
             )
@@ -193,9 +183,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,7 +193,7 @@ class SessionScreen(Screen):
     @on(Button.Pressed, "#create-session")
     @work(exclusive=True)
     async def create_session(self) -> None:
-        session_id = self.app.client.create_session()
+        session_id, message = self.app.client.create_session()
 
         if session_id:
             self.app.current_session = session_id
@@ -211,7 +201,7 @@ class SessionScreen(Screen):
             self.app.install_screen(ChatScreen(), "chat")
             self.app.push_screen("chat")
         else:
-            self.query_one("#session-error").update("Failed to create session")
+            self.query_one("#session-error").update(f"Failed to create session: {message}")
 
     @on(ListView.Selected)
     def session_selected(self, event: ListView.Selected) -> None:
@@ -402,7 +392,7 @@ class ChatScreen(Screen):
         self.conversation, message = self.app.client.get_session_history(
             self.app.current_session
         )
-        if message != "success":
+        if message != "Success":
             self.query_one("#chat-log").write(f"[red]Error: {message}[/red]")
             return
         self.update_message_display()
@@ -437,7 +427,7 @@ class ChatScreen(Screen):
                 self.app.client.chat, self.app.current_session, message
             )
 
-            if message != "success":
+            if message != "Success":
                 self.query_one("#chat-log").write(f"[red]Error: {message}[/red]")
                 return
             new_messages = conversation_history[len(self.conversation) :]

{aicosmos_client-0.0.3 → aicosmos_client-0.0.4}/aicosmos_client/client.py

@@ -99,7 +99,7 @@ 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"

{aicosmos_client-0.0.3 → aicosmos_client-0.0.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "aicosmos_client"
-version = "0.0.3"
+version = "0.0.4"
 authors = [{ name = "Example Author", email = "author@example.com" }]
 description = "client for AICosmos platform"
 readme = "README.md"