remove user message wait trigger, move conv agent prompt

Author: mike-deem-uipath

src/uipath/_cli/_chat/_bridge.py

@@ -58,9 +58,9 @@ def __init__(
         self._client: AsyncClient | None = None
         self._connected_event = asyncio.Event()
 
-        # Set CAS_WEBSOCKET_DISABLED when using the debugger to prevent websocket errors from 
+        # Set CAS_WEBSOCKET_DISABLED when using the debugger to prevent websocket errors from
         # interrupting the debugging session. Events will be logged instead of being sent.
-        self._debug_mode = os.environ.get("CAS_WEBSOCKET_DISABLED") == "true"
+        self._websocket_disabled = os.environ.get("CAS_WEBSOCKET_DISABLED") == "true"
 
     async def connect(self, timeout: float = 10.0) -> None:
         """Establish WebSocket connection to the server.
@@ -94,8 +94,10 @@ async def connect(self, timeout: float = 10.0) -> None:
 
         self._connected_event.clear()
 
-        if self._debug_mode:
-            logger.warning("SocketIOChatBridge is in debug mode. Not connecting websocket.")
+        if self._websocket_disabled:
+            logger.warning(
+                "SocketIOChatBridge is in debug mode. Not connecting websocket."
+            )
         else:
             try:
                 # Attempt to connect with timeout
@@ -157,7 +159,7 @@ async def emit_message_event(
         if self._client is None:
             raise RuntimeError("WebSocket client not connected. Call connect() first.")
 
-        if not self._connected_event.is_set() and not self._debug_mode:
+        if not self._connected_event.is_set() and not self._websocket_disabled:
             raise RuntimeError("WebSocket client not in connected state")
 
         try:
@@ -174,8 +176,10 @@ async def emit_message_event(
                 mode="json", exclude_none=True, by_alias=True
             )
 
-            if self._debug_mode:
-                logger.info(f"SocketIOChatBridge is in debug mode. Not sending event: {json.dumps(event_data)}")
+            if self._websocket_disabled:
+                logger.info(
+                    f"SocketIOChatBridge is in debug mode. Not sending event: {json.dumps(event_data)}"
+                )
             else:
                 await self._client.emit("ConversationEvent", event_data)
 
@@ -195,7 +199,7 @@ async def emit_exchange_end_event(self) -> None:
         if self._client is None:
             raise RuntimeError("WebSocket client not connected. Call connect() first.")
 
-        if not self._connected_event.is_set() and not self._debug_mode:
+        if not self._connected_event.is_set() and not self._websocket_disabled:
             raise RuntimeError("WebSocket client not in connected state")
 
         try:
@@ -211,8 +215,10 @@ async def emit_exchange_end_event(self) -> None:
                 mode="json", exclude_none=True, by_alias=True
             )
 
-            if self._debug_mode:
-                logger.info(f"SocketIOChatBridge is in debug mode. Not sending event: {json.dumps(event_data)}")
+            if self._websocket_disabled:
+                logger.info(
+                    f"SocketIOChatBridge is in debug mode. Not sending event: {json.dumps(event_data)}"
+                )
             else:
                 await self._client.emit("ConversationEvent", event_data)
 
@@ -244,8 +250,10 @@ async def emit_interrupt_event(self, runtime_result: UiPathRuntimeResult):
                 event_data = interrupt_event.model_dump(
                     mode="json", exclude_none=True, by_alias=True
                 )
-                if self._debug_mode:
-                    logger.info(f"SocketIOChatBridge is in debug mode. Not sending event: {json.dumps(event_data)}")
+                if self._websocket_disabled:
+                    logger.info(
+                        f"SocketIOChatBridge is in debug mode. Not sending event: {json.dumps(event_data)}"
+                    )
                 else:
                     await self._client.emit("ConversationEvent", event_data)
             except Exception as e:
@@ -335,7 +343,9 @@ def get_chat_bridge(
     if os.environ.get("CAS_WEBSOCKET_HOST"):
         websocket_url = f"ws://{os.environ.get('CAS_WEBSOCKET_HOST')}?conversationId={context.conversation_id}"
         websocket_path = "/socket.io"
-        logger.warning(f"CAS_WEBSOCKET_HOST is set. Using websocket_url '{websocket_url}{websocket_path}'.")
+        logger.warning(
+            f"CAS_WEBSOCKET_HOST is set. Using websocket_url '{websocket_url}{websocket_path}'."
+        )
 
     # Build headers from context
     headers = {

src/uipath/agent/react/__init__.py

@@ -3,6 +3,10 @@
 This module includes UiPath ReAct Agent Loop constructs such as prompts, tools
 """
 
+from .conversational_prompts import (
+    PromptUserSettings,
+    generate_conversational_agent_system_prompt,
+)
 from .prompts import AGENT_SYSTEM_PROMPT_TEMPLATE
 from .tools import (
     END_EXECUTION_TOOL,
@@ -19,4 +23,6 @@
     "RAISE_ERROR_TOOL",
     "EndExecutionToolSchemaModel",
     "RaiseErrorToolSchemaModel",
+    "PromptUserSettings",
+    "generate_conversational_agent_system_prompt",
 ]