yaicli 0.0.2__py3-none-any.whl → 0.0.4__py3-none-any.whl

{yaicli-0.0.2.dist-info → yaicli-0.0.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: yaicli
-Version: 0.0.2
+Version: 0.0.4
 Summary: A simple CLI tool to interact with LLM
 License-File: LICENSE
 Requires-Python: >=3.8

yaicli-0.0.4.dist-info/RECORD

@@ -0,0 +1,6 @@
+yaicli.py,sha256=CY5TnPQAypGJ1FQLiryGBV4UFc8E1ZQyaNmT7ghw7NU,16741
+yaicli-0.0.4.dist-info/METADATA,sha256=fY81CYJR2jcWmwElOe3rSk-wM4Lk27SZhTPdK0eHw4E,379
+yaicli-0.0.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+yaicli-0.0.4.dist-info/entry_points.txt,sha256=gdduQwAuu_LeDqnDU81Fv3NPmD2tRQ1FffvolIP3S1Q,34
+yaicli-0.0.4.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+yaicli-0.0.4.dist-info/RECORD,,

yaicli.py

@@ -79,9 +79,7 @@ STREAM=true"""
         @self.bindings.add(Keys.ControlI)  # Bind Ctrl+I to switch modes
         def _(event: KeyPressEvent):
             self.current_mode = (
-                ModeEnum.CHAT.value
-                if self.current_mode == ModeEnum.EXECUTE.value
-                else ModeEnum.EXECUTE.value
+                ModeEnum.CHAT.value if self.current_mode == ModeEnum.EXECUTE.value else ModeEnum.EXECUTE.value
             )
 
     def detect_os(self):
@@ -126,9 +124,8 @@ Rules:
         _os = self.detect_os()
         _shell = self.detect_shell()
         return (
-            "You are a system and code assistant, "
-            f"focusing on {_os} and {_shell}. "
-            "Assist with system management, script writing, and coding tasks. "
+            "You are yaili, a system management and programing assistant, "
+            f"You are managing {_os} operating system with {_shell} shell. "
             "Your responses should be concise and use Markdown format, "
             "unless the user explicitly requests more details."
         )
@@ -235,6 +232,8 @@ Rules:
             response = self._call_api(url, headers, data)
         except requests.exceptions.RequestException as e:
             self.console.print(f"[red]Error calling API: {e}[/red]")
+            if self.verbose and e.response:
+                self.console.print(f"{e.response.text}")
             raise typer.Exit(code=1) from None
         if not response:
             raise typer.Exit(code=1)
@@ -271,9 +270,7 @@ Rules:
         self.console.print(f"\n[bold green]Executing command: [/bold green] {command}\n")
         result = subprocess.run(command, shell=True)
         if result.returncode != 0:
-            self.console.print(
-                f"\n[bold red]Command failed with return code: {result.returncode}[/bold red]"
-            )
+            self.console.print(f"\n[bold red]Command failed with return code: {result.returncode}[/bold red]")
 
     def get_prompt_tokens(self):
         """Get prompt tokens based on current mode"""
@@ -382,9 +379,7 @@ Rules:
         # Load configuration
         self.config = self.load_config()
         if not self.config.get("API_KEY", None):
-            self.console.print(
-                "[red]API key not found. Please set it in the configuration file.[/red]"
-            )
+            self.console.print("[red]API key not found. Please set it in the configuration file.[/red]")
             return
 
         # Set initial mode
@@ -417,10 +412,10 @@ CONTEXT_SETTINGS = {
 }
 
 app = typer.Typer(
-    name="ShellAI",
+    name="yaicli",
     context_settings=CONTEXT_SETTINGS,
     pretty_exceptions_enable=False,
-    short_help="ShellAI Command Line Tool",
+    short_help="yaicli. Your AI interface in cli.",
     no_args_is_help=True,
     invoke_without_command=True,
 )
@@ -428,17 +423,18 @@ app = typer.Typer(
 
 @app.command()
 def main(
-    prompt: Annotated[str, typer.Argument(show_default=False, help="The prompt send to the LLM")],
-    verbose: Annotated[
-        bool, typer.Option("--verbose", "-V", help="Show verbose information")
-    ] = False,
+    ctx: typer.Context,
+    prompt: Annotated[str, typer.Argument(show_default=False, help="The prompt send to the LLM")] = "",
+    verbose: Annotated[bool, typer.Option("--verbose", "-V", help="Show verbose information")] = False,
     chat: Annotated[bool, typer.Option("--chat", "-c", help="Start in chat mode")] = False,
-    shell: Annotated[
-        bool, typer.Option("--shell", "-s", help="Generate and execute shell command")
-    ] = False,
+    shell: Annotated[bool, typer.Option("--shell", "-s", help="Generate and execute shell command")] = False,
 ):
-    """LLM CLI Tool"""
-    cli = ShellAI()
+    """yaicli. Your AI interface in cli."""
+    if not prompt and not chat:
+        typer.echo(ctx.get_help())
+        raise typer.Exit()
+
+    cli = ShellAI(verbose=verbose)
     cli.run(chat=chat, shell=shell, prompt=prompt)
 
 

yaicli-0.0.2.dist-info/RECORD

@@ -1,6 +0,0 @@
-yaicli.py,sha256=4c5UFeIUFI7lQERM31drdaj-bR6Q1DP3WBhm5qnu6b4,16620
-yaicli-0.0.2.dist-info/METADATA,sha256=N-j9mcia0VG5XmxoXfrLz5ziXWKE8lqbVgSMfv-qnCQ,379
-yaicli-0.0.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-yaicli-0.0.2.dist-info/entry_points.txt,sha256=gdduQwAuu_LeDqnDU81Fv3NPmD2tRQ1FffvolIP3S1Q,34
-yaicli-0.0.2.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-yaicli-0.0.2.dist-info/RECORD,,