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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: yaicli
-Version: 0.0.3
+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):
@@ -272,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"""
@@ -383,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
@@ -418,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,
 )
@@ -429,16 +423,17 @@ 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"""
+    """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.3.dist-info/RECORD

@@ -1,6 +0,0 @@
-yaicli.py,sha256=4c-cGPzZlUPdqv8uZrWmiimlCp8Q8S9UH7jKHRWYI8U,16709
-yaicli-0.0.3.dist-info/METADATA,sha256=3C60zYZfELyBiKNPqXyPKKqAh8cKe7FG42oc_Esr01Y,379
-yaicli-0.0.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-yaicli-0.0.3.dist-info/entry_points.txt,sha256=gdduQwAuu_LeDqnDU81Fv3NPmD2tRQ1FffvolIP3S1Q,34
-yaicli-0.0.3.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-yaicli-0.0.3.dist-info/RECORD,,