pylogue 0.1.0__py3-none-any.whl → 0.2.0__py3-none-any.whl

pylogue/_modidx.py

@@ -45,8 +45,6 @@ d = { 'settings': { 'branch': 'main',
             'pylogue.renderer': { 'pylogue.renderer.ChatRenderer': ('3-renderer.html#chatrenderer', 'pylogue/renderer.py'),
                                   'pylogue.renderer.ChatRenderer.__init__': ( '3-renderer.html#chatrenderer.__init__',
                                                                               'pylogue/renderer.py'),
-                                  'pylogue.renderer.ChatRenderer.render_chat_interface': ( '3-renderer.html#chatrenderer.render_chat_interface',
-                                                                                           'pylogue/renderer.py'),
                                   'pylogue.renderer.ChatRenderer.render_form': ( '3-renderer.html#chatrenderer.render_form',
                                                                                  'pylogue/renderer.py'),
                                   'pylogue.renderer.ChatRenderer.render_input': ( '3-renderer.html#chatrenderer.render_input',
@@ -59,8 +57,12 @@ d = { 'settings': { 'branch': 'main',
                                                                                                 'pylogue/renderer.py')},
             'pylogue.service': { 'pylogue.service.ChatService': ('2-service.html#chatservice', 'pylogue/service.py'),
                                  'pylogue.service.ChatService.__init__': ('2-service.html#chatservice.__init__', 'pylogue/service.py'),
+                                 'pylogue.service.ChatService._is_async_generator': ( '2-service.html#chatservice._is_async_generator',
+                                                                                      'pylogue/service.py'),
                                  'pylogue.service.ChatService.process_message': ( '2-service.html#chatservice.process_message',
                                                                                   'pylogue/service.py'),
+                                 'pylogue.service.ChatService.process_message_stream': ( '2-service.html#chatservice.process_message_stream',
+                                                                                         'pylogue/service.py'),
                                  'pylogue.service.ChatService.process_session_message': ( '2-service.html#chatservice.process_session_message',
                                                                                           'pylogue/service.py'),
                                  'pylogue.service.ContextAwareResponder': ('2-service.html#contextawareresponder', 'pylogue/service.py'),

pylogue/chatapp.py

@@ -78,15 +78,7 @@ class ChatAppConfig:
 
 # %% ../../nbs/4-ChatApp.ipynb 4
 class ChatApp:
-    """
-    Complete chat application with full dependency injection.
-
-    This is the main orchestration layer that ties together:
-    - Session management (state)
-    - Chat service (business logic)
-    - Renderer (presentation)
-    - FastHTML + WebSocket (infrastructure)
-    """
+    """Main chat application composing all components."""
 
     def __init__(
         self,
@@ -96,7 +88,7 @@ class ChatApp:
         config: Optional[ChatAppConfig] = None,
     ):
         """
-        Initialize ChatApp with injected dependencies.
+        Initialize ChatApp with dependency injection.
 
         Args:
             session_manager: Manages chat sessions
@@ -155,12 +147,11 @@ class ChatApp:
                     H1(self.config.app_title, style=self.config.header_style),
                     self.renderer.render_messages(initial_messages),
                     self.renderer.render_form(),
-                    hx_ext="ws",
-                    ws_connect=self.config.ws_endpoint,
                     style=container_style,
                 ),
             )
 
+        # Register WebSocket route with message handler
         @self.app.ws(
             self.config.ws_endpoint, conn=self._on_connect, disconn=self._on_disconnect
         )
@@ -183,12 +174,11 @@ class ChatApp:
 
     async def _handle_websocket_message(self, msg: str, send, ws):
         """
-        Handle incoming WebSocket message with full lifecycle:
+        Handle incoming WebSocket message with streaming support:
         1. Add user message
-        2. Show pending assistant message with spinner
-        3. Process with chat service
-        4. Update with actual response
-        5. Clear input
+        2. Add empty assistant message for streaming
+        3. Stream response tokens and update message progressively
+        4. Clear input
         """
         session_id = str(id(ws))
         session = self.session_manager.get_session(session_id)
@@ -204,21 +194,32 @@ class ChatApp:
         session.add_message("User", msg)
         await send(self.renderer.render_messages(session.get_messages()))
 
-        # Step 2: Add pending assistant message with spinner
-        pending_msg = session.add_message("Assistant", "", pending=True)
-        await send(self.renderer.render_messages(session.get_messages()))
+        # Step 2: Add empty assistant message for streaming
+        assistant_msg = session.add_message("Assistant", "", pending=False)
 
-        # Step 3: Process message with chat service
+        # Step 3: Stream response and update progressively
         try:
-            response = await self.chat_service.process_message(msg, session)
-        except Exception as e:
-            response = f"Error: {str(e)}"
+            response_chunks = []
+            chunk_count = 0
+            async for chunk in self.chat_service.process_message_stream(msg, session):
+                response_chunks.append(chunk)
+                chunk_count += 1
+                # Update the assistant message with accumulated response
+                full_response = "".join(response_chunks)
+                session.update_message(
+                    assistant_msg.id, content=full_response, pending=False
+                )
+                # Send updated message list to UI
+                print(f"📤 Sending chunk #{chunk_count}: {repr(chunk)}")  # Debug
+                await send(self.renderer.render_messages(session.get_messages()))
 
-        # Step 4: Update pending message with actual response
-        session.update_message(pending_msg.id, content=response, pending=False)
-        await send(self.renderer.render_messages(session.get_messages()))
+        except Exception as e:
+            # Handle errors
+            error_msg = f"Error: {str(e)}"
+            session.update_message(assistant_msg.id, content=error_msg, pending=False)
+            await send(self.renderer.render_messages(session.get_messages()))
 
-        # Step 5: Clear input field
+        # Step 4: Clear input field
         await send(self.renderer.render_input())
 
     def run(

pylogue/renderer.py

@@ -21,6 +21,7 @@ class ChatRenderer:
         input_placeholder: str = "Type a message...",
         input_style: Optional[str] = None,
         chat_container_style: Optional[str] = None,
+        ws_endpoint: str = "/ws",
     ):
         """
         Initialize ChatRenderer.
@@ -30,9 +31,11 @@ class ChatRenderer:
             input_placeholder: Placeholder text for input field
             input_style: Custom CSS style for input field
             chat_container_style: Custom CSS style for chat container
+            ws_endpoint: WebSocket endpoint path
         """
         self.card = card or ChatCard()
         self.input_placeholder = input_placeholder
+        self.ws_endpoint = ws_endpoint
         self.input_style = input_style or (
             "width: 60%; max-width: 600px; padding: 0.75em; "
             "font-size: 1em; border-radius: 0.5em"
@@ -100,7 +103,7 @@ class ChatRenderer:
         ws_send: bool = True,
     ) -> Any:
         """
-        Render the input form with styling.
+        Render the input form with WebSocket connection.
 
         Args:
             form_id: HTML ID for the form
@@ -118,34 +121,8 @@ class ChatRenderer:
         return Form(
             self.render_input(),
             id=form_id,
+            hx_ext="ws",
+            ws_connect=self.ws_endpoint,
             ws_send=ws_send,
             style=form_style,
         )
-
-    def render_chat_interface(
-        self,
-        messages: List[Message],
-        title: str = "Chat",
-        header_style: Optional[str] = None,
-        container_style: Optional[str] = None,
-    ) -> Any:
-        """
-        Render complete chat interface with messages and input.
-
-        Args:
-            messages: List of messages to display
-            title: Chat title/header
-            header_style: Custom style for header
-            container_style: Custom style for main container
-
-        Returns:
-            FastHTML Div with complete chat interface
-        """
-        header_style = header_style or "text-align: center; padding: 1em;"
-
-        return Div(
-            H1(title, style=header_style),
-            self.render_messages(messages),
-            self.render_form(),
-            style=container_style,
-        )

pylogue/service.py

@@ -60,6 +60,12 @@ class ChatService:
         self.error_handler = error_handler or DefaultErrorHandler()
         self.context_provider = context_provider
 
+    def _is_async_generator(self, obj):
+        """Check if object is an async generator."""
+        return inspect.isasyncgenfunction(obj) or (
+            hasattr(obj, "__call__") and inspect.isasyncgenfunction(obj.__call__)
+        )
+
     async def process_message(
         self, user_message: str, session: Optional[ChatSession] = None
     ) -> str:
@@ -79,7 +85,15 @@ class ChatService:
             if self.context_provider and session:
                 context = self.context_provider(session)
 
-            # Call responder
+            # Check if responder is an async generator (streaming)
+            if self._is_async_generator(self.responder):
+                # For streaming responders, collect all chunks
+                chunks = []
+                async for chunk in self.responder(user_message, context):
+                    chunks.append(str(chunk))
+                return "".join(chunks)
+
+            # Non-streaming responder
             if inspect.iscoroutinefunction(self.responder):
                 result = await self.responder(user_message, context)
             else:
@@ -92,6 +106,42 @@ class ChatService:
         except Exception as e:
             return self.error_handler(e, user_message)
 
+    async def process_message_stream(
+        self, user_message: str, session: Optional[ChatSession] = None
+    ):
+        """
+        Process a user message and stream the response token by token.
+
+        Args:
+            user_message: The user's input message
+            session: Optional chat session for context
+
+        Yields:
+            Response chunks as they are generated
+        """
+        try:
+            # Extract context if provider exists
+            context = None
+            if self.context_provider and session:
+                context = self.context_provider(session)
+
+            # Check if responder supports streaming
+            if self._is_async_generator(self.responder):
+                async for chunk in self.responder(user_message, context):
+                    yield str(chunk)
+            else:
+                # Non-streaming responder - yield full response
+                if inspect.iscoroutinefunction(self.responder):
+                    result = await self.responder(user_message, context)
+                else:
+                    result = self.responder(user_message, context)
+                    if inspect.isawaitable(result):
+                        result = await result
+                yield str(result)
+
+        except Exception as e:
+            yield self.error_handler(e, user_message)
+
     async def process_session_message(
         self, session: ChatSession, user_message: str, add_to_session: bool = True
     ) -> Message:

{pylogue-0.1.0.dist-info → pylogue-0.2.0.dist-info}/METADATA

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: pylogue
-Version: 0.1.0
-Summary: A Chatbot UI build for Pydantic-AI agents
+Version: 0.2.0
+Summary: A Chatbot UI built for AI agents
 Author-email: Yeshwanth Reddy <yyeshr@gmail.com>
 Maintainer-email: Yeshwanth Reddy <yyeshr@gmail.com>
 License: MIT license

pylogue-0.2.0.dist-info/RECORD

@@ -0,0 +1,17 @@
+pylogue/__init__.py,sha256=5_YsGHUfQ5L1Eh4uZjSnsgszN9JSrWnO1VwEzBGN6Z4,273
+pylogue/__pre_init__.py,sha256=NiQUAz7feHFR47QTpMwF324j61IVrOreZsk9wY11cds,153
+pylogue/_modidx.py,sha256=apvAlYvaQgShSh_ZrX7fXSpxDdwPBYIKwEQWdEkVJg8,14209
+pylogue/cards.py,sha256=wsOVNbMczFbyoATkujC7xJAeskYgTqsJmxmkHqzHA-g,5385
+pylogue/chat.py,sha256=RyRADCZWxTjCM3W7FgIj0jQ8WrqhRBzbeFEvS9mTZq4,3982
+pylogue/chatapp.py,sha256=khtkhbjWvkZIHmEYQPLogFlDeK5KjItWXOHVC6_9tkg,9918
+pylogue/health.py,sha256=8LWhpVzdj2g6qUcErrg455_6WfOsv3lp82OLLwJPtug,443
+pylogue/renderer.py,sha256=KGo0l-Jo8i2pe8NqwlZsM9Qo3ZBA2EV-osrHnBUWzBA,3862
+pylogue/service.py,sha256=e3GfbQrOPlRIj7j0CZZx_qN2ZI_polOuWqBMm7ySXqs,6917
+pylogue/session.py,sha256=fUyGOFIG8b1clhyyw5c_dapdwrJyzd9K2bOc_qKBQcs,5382
+pylogue-0.2.0.dist-info/licenses/AUTHORS.md,sha256=5Viska6TOr9uggc2olSr8VVnu6BFw_F6_64gP42CwQ4,157
+pylogue-0.2.0.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pylogue-0.2.0.dist-info/METADATA,sha256=upCfS03poXnbZyeCISnAKqKmsvIORSyhEWSPtLJOEfk,1059
+pylogue-0.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pylogue-0.2.0.dist-info/entry_points.txt,sha256=ha0gwcGgtciKEhmFcMKJO1efTY9Spu1wcTHUopB1bec,40
+pylogue-0.2.0.dist-info/top_level.txt,sha256=oEueWVdlRAUPQt8VfQdqFEqOWxvUObx3UIa3UYa3s6o,8
+pylogue-0.2.0.dist-info/RECORD,,

pylogue-0.1.0.dist-info/RECORD

@@ -1,17 +0,0 @@
-pylogue/__init__.py,sha256=5_YsGHUfQ5L1Eh4uZjSnsgszN9JSrWnO1VwEzBGN6Z4,273
-pylogue/__pre_init__.py,sha256=NiQUAz7feHFR47QTpMwF324j61IVrOreZsk9wY11cds,153
-pylogue/_modidx.py,sha256=S-NkgQExGm8mDtWiZztOY9DhMtvieuDNXg1PG8RFTMc,13970
-pylogue/cards.py,sha256=wsOVNbMczFbyoATkujC7xJAeskYgTqsJmxmkHqzHA-g,5385
-pylogue/chat.py,sha256=RyRADCZWxTjCM3W7FgIj0jQ8WrqhRBzbeFEvS9mTZq4,3982
-pylogue/chatapp.py,sha256=V0edaNeQXy2wKwDr5MZ5IcKC3a_AakXPcfN4sqGpDhU,9643
-pylogue/health.py,sha256=8LWhpVzdj2g6qUcErrg455_6WfOsv3lp82OLLwJPtug,443
-pylogue/renderer.py,sha256=7ZiGNJ83xwCVgvw4pnEEJPB3E9lO60vK6qobAttryNk,4507
-pylogue/service.py,sha256=0hdL8Eg7FDpX_NXSuTzC7bJ_wvEnnRsomDR-kNrs2i4,4952
-pylogue/session.py,sha256=fUyGOFIG8b1clhyyw5c_dapdwrJyzd9K2bOc_qKBQcs,5382
-pylogue-0.1.0.dist-info/licenses/AUTHORS.md,sha256=5Viska6TOr9uggc2olSr8VVnu6BFw_F6_64gP42CwQ4,157
-pylogue-0.1.0.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pylogue-0.1.0.dist-info/METADATA,sha256=0qS4tbqcJYe-ZmLOc9LX2NAG1LTmnWHmghaKjKHUCbM,1068
-pylogue-0.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pylogue-0.1.0.dist-info/entry_points.txt,sha256=ha0gwcGgtciKEhmFcMKJO1efTY9Spu1wcTHUopB1bec,40
-pylogue-0.1.0.dist-info/top_level.txt,sha256=oEueWVdlRAUPQt8VfQdqFEqOWxvUObx3UIa3UYa3s6o,8
-pylogue-0.1.0.dist-info/RECORD,,