fix: improved inquire namespace prompts

Author: evilsocket

nerve/runtime/logging.py

@@ -24,7 +24,7 @@ def init(log_path: pathlib.Path | None = None, level: str = "INFO", litellm_debu
         logger.add(
             sys.stdout,
             colorize=True,
-            format="{time} <level>{message}</level>",
+            format="<level>[{time:MM-DD-YY HH:mm:ss}] {level}</level> {message}",
             level=level,
         )
 

nerve/tools/mcp/client.py

@@ -28,15 +28,13 @@ async def _create_streams(
             async with stdio_client(server=self.server_params) as (read_stream, write_stream):
                 try:
                     yield read_stream, write_stream
-                except GeneratorExit:
-                    pass
+                except Exception as e:
+                    logger.debug("error yielding streams: {}", e)
         except Exception as e:
             # TODO: there's a weird bug, if we don't do this when the process exits
             # we will see an exception
-            if "anext(): asynchronous generator is already running" in str(e):
-                exit(0)
-            else:
-                raise e
+            logger.debug("error creating streams: {}", e)
+            exit(0)
 
     async def connect(self) -> None:
         if self._session:

nerve/tools/namespaces/inquire.py

@@ -17,7 +17,7 @@ def ask_question(
         Field(description="The question to ask the user.", examples=["What is your name?", "What is your age?"]),
     ],
 ) -> str:
-    """Ask a question to the user."""
+    """Use this tool to ask a question to the user."""
 
     print()
     return str(inquirer.prompt([inquirer.Text("question", message=question)])["question"]).strip()
@@ -32,7 +32,7 @@ def ask_for_confirmation(
     ],
     default: Annotated[bool, Field(description="The default answer to the question.", examples=[True, False])] = False,
 ) -> str:
-    """Ask a confirmation question to the user."""
+    """Use this tool to ask a confirmation question to the user."""
 
     print()
     return (
@@ -53,7 +53,7 @@ def ask_for_single_choice(
         Field(description="The choices to offer the user.", examples=["red", "blue", "green"]),
     ],
 ) -> str:
-    """Ask a single choice question to the user."""
+    """Use this tool to ask a single choice question to the user."""
 
     print()
     return str(inquirer.prompt([inquirer.List("choice", message=question, choices=choices)])["choice"]).strip()
@@ -72,7 +72,7 @@ def ask_for_multiple_choice(
         Field(description="The choices to offer the user.", examples=["red", "blue", "green"]),
     ],
 ) -> str:
-    """Ask a multiple choice question to the user."""
+    """Use this tool to ask a multiple choice question to the user."""
 
     print()
     return ", ".join(