yaicli 0.0.12__py3-none-any.whl → 0.0.14__py3-none-any.whl

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "yaicli"
-version = "0.0.12"
+version = "0.0.14"
 description = "A simple CLI tool to interact with LLM"
 authors = [{ name = "belingud", email = "im.victor@qq.com" }]
 readme = "README.md"
@@ -46,10 +46,6 @@ Documentation = "https://github.com/belingud/yaicli"
 [project.scripts]
 ai = "yaicli:app"
 
-[tool.pdm.scripts]
-bump = "bump2version {args}"
-changelog = "just changelog"
-
 [tool.uv]
 resolution = "highest"
 

{yaicli-0.0.12.dist-info → yaicli-0.0.14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: yaicli
-Version: 0.0.12
+Version: 0.0.14
 Summary: A simple CLI tool to interact with LLM
 Project-URL: Homepage, https://github.com/belingud/yaicli
 Project-URL: Repository, https://github.com/belingud/yaicli
@@ -329,14 +329,17 @@ MAX_TOKENS=1024
 
 Below are the available configuration options and override environment variables:
 
-- **BASE_URL**: API endpoint URL (default: OpenAI API), env: AI_BASE_URL
-- **API_KEY**: Your API key for the LLM provider, env: AI_API_KEY
-- **MODEL**: The model to use (e.g., gpt-4o, gpt-3.5-turbo), default: gpt-4o, env: AI_MODEL
-- **SHELL_NAME**: Shell to use (auto for automatic detection), default: auto, env: AI_SHELL_NAME
-- **OS_NAME**: OS to use (auto for automatic detection), default: auto, env: AI_OS_NAME
-- **COMPLETION_PATH**: Path for completions endpoint, default: /chat/completions, env: AI_COMPLETION_PATH
-- **ANSWER_PATH**: Json path expression to extract answer from response, default: choices[0].message.content, env: AI_ANSWER_PATH
-- **STREAM**: Enable/disable streaming responses, default: true, env: AI_STREAM
+- **BASE_URL**: API endpoint URL (default: OpenAI API), env: YAI_BASE_URL
+- **API_KEY**: Your API key for the LLM provider, env: YAI_API_KEY
+- **MODEL**: The model to use (e.g., gpt-4o, gpt-3.5-turbo), default: gpt-4o, env: YAI_MODEL
+- **SHELL_NAME**: Shell to use (auto for automatic detection), default: auto, env: YAI_SHELL_NAME
+- **OS_NAME**: OS to use (auto for automatic detection), default: auto, env: YAI_OS_NAME
+- **COMPLETION_PATH**: Path for completions endpoint, default: /chat/completions, env: YAI_COMPLETION_PATH
+- **ANSWER_PATH**: Json path expression to extract answer from response, default: choices[0].message.content, env: YAI_ANSWER_PATH
+- **STREAM**: Enable/disable streaming responses, default: true, env: YAI_STREAM
+- **TEMPERATURE**: Temperature for response generation (default: 0.7), env: YAI_TEMPERATURE
+- **TOP_P**: Top-p sampling for response generation (default: 1.0), env: YAI_TOP_P
+- **MAX_TOKENS**: Maximum number of tokens for response generation (default: 1024), env: YAI_MAX_TOKENS
 
 Default config of `COMPLETION_PATH` and `ANSWER_PATH` is OpenAI compatible. If you are using OpenAI or other OpenAI compatible LLM provider, you can use the default config.
 
@@ -467,10 +470,45 @@ In Execute mode:
 
 ## Examples
 
+### Have a Chat
+
+```bash
+$ ai "What is the capital of France?"
+Assistant:
+The capital of France is Paris.
+```
+
+### Command Gen and Run
+
+```bash
+$ ai -s 'Check the current directory size'
+Assistant:
+du -sh .
+
+Generated command: du -sh .
+Execute this command? [y/n/e] (n): e
+Edit command, press enter to execute:
+du -sh ./
+Output:
+109M    ./
+```
+
 ### Chat Mode Example
 
 ```bash
 $ ai --chat
+
+██    ██  █████  ██  ██████ ██      ██
+ ██  ██  ██   ██ ██ ██      ██      ██
+  ████   ███████ ██ ██      ██      ██
+   ██    ██   ██ ██ ██      ██      ██
+   ██    ██   ██ ██  ██████ ███████ ██
+
+Press TAB to change in chat and exec mode
+Type /clear to clear chat history
+Type /his to see chat history
+Press Ctrl+C or type /exit to exit
+
 💬 > Tell me about the solar system
 
 Assistant:
@@ -489,7 +527,17 @@ Certainly! Here’s a brief overview of the solar system:
  • Dwarf Planets:
     • Pluto: Once considered the ninth planet, now classified as
 
-💬 >
+🚀 > Check the current directory size
+Assistant:
+du -sh .
+
+Generated command: du -sh .
+Execute this command? [y/n/e] (n): e
+Edit command, press enter to execute:
+du -sh ./
+Output:
+109M    ./
+🚀 >
 ```
 
 ### Execute Mode Example

yaicli-0.0.14.dist-info/RECORD

@@ -0,0 +1,7 @@
+pyproject.toml,sha256=D0K1CAn3p9Tf-E9VMq82Ml0jX44Eh6gYRzn5qKQD5xs,1452
+yaicli.py,sha256=5tQBYifI2CFJj_7BpRfRox7yeEYhDu9T9rDkKBppKzE,21392
+yaicli-0.0.14.dist-info/METADATA,sha256=738uxK3JVEOXwcR3m_KZqni9CRSSnj9RLbEOVIzXgCQ,29445
+yaicli-0.0.14.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+yaicli-0.0.14.dist-info/entry_points.txt,sha256=gdduQwAuu_LeDqnDU81Fv3NPmD2tRQ1FffvolIP3S1Q,34
+yaicli-0.0.14.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+yaicli-0.0.14.dist-info/RECORD,,

yaicli.py

@@ -2,6 +2,7 @@ import configparser
 import json
 import platform
 import subprocess
+import time
 from os import getenv
 from os.path import basename, pathsep
 from pathlib import Path
@@ -11,7 +12,7 @@ import httpx
 import jmespath
 import typer
 from distro import name as distro_name
-from prompt_toolkit import PromptSession
+from prompt_toolkit import PromptSession, prompt
 from prompt_toolkit.completion import WordCompleter
 from prompt_toolkit.history import FileHistory
 from prompt_toolkit.key_binding import KeyBindings, KeyPressEvent
@@ -19,7 +20,8 @@ from prompt_toolkit.keys import Keys
 from rich.console import Console
 from rich.live import Live
 from rich.markdown import Markdown
-from rich.prompt import Confirm
+from rich.panel import Panel
+from rich.prompt import Prompt
 
 SHELL_PROMPT = """Your are a Shell Command Generator.
 Generate a command EXCLUSIVELY for {_os} OS with {_shell} shell.
@@ -46,14 +48,17 @@ CHAT_MODE = "chat"
 TEMP_MODE = "temp"
 
 DEFAULT_CONFIG_MAP = {
-    "BASE_URL": {"value": "https://api.openai.com/v1", "env_key": "AI_BASE_URL"},
-    "API_KEY": {"value": "", "env_key": "AI_API_KEY"},
-    "MODEL": {"value": "gpt-4o", "env_key": "AI_MODEL"},
-    "SHELL_NAME": {"value": "auto", "env_key": "AI_SHELL_NAME"},
-    "OS_NAME": {"value": "auto", "env_key": "AI_OS_NAME"},
-    "COMPLETION_PATH": {"value": "chat/completions", "env_key": "AI_COMPLETION_PATH"},
-    "ANSWER_PATH": {"value": "choices[0].message.content", "env_key": "AI_ANSWER_PATH"},
-    "STREAM": {"value": "true", "env_key": "AI_STREAM"},
+    "BASE_URL": {"value": "https://api.openai.com/v1", "env_key": "YAI_BASE_URL"},
+    "API_KEY": {"value": "", "env_key": "YAI_API_KEY"},
+    "MODEL": {"value": "gpt-4o", "env_key": "YAI_MODEL"},
+    "SHELL_NAME": {"value": "auto", "env_key": "YAI_SHELL_NAME"},
+    "OS_NAME": {"value": "auto", "env_key": "YAI_OS_NAME"},
+    "COMPLETION_PATH": {"value": "chat/completions", "env_key": "YAI_COMPLETION_PATH"},
+    "ANSWER_PATH": {"value": "choices[0].message.content", "env_key": "YAI_ANSWER_PATH"},
+    "STREAM": {"value": "true", "env_key": "YAI_STREAM"},
+    "TEMPERATURE": {"value": "0.7", "env_key": "YAI_TEMPERATURE"},
+    "TOP_P": {"value": "1.0", "env_key": "YAI_TOP_P"},
+    "MAX_TOKENS": {"value": "1024", "env_key": "YAI_MAX_TOKENS"},
 }
 
 DEFAULT_CONFIG_INI = """[core]
@@ -102,10 +107,14 @@ class CLI:
         self.bindings = KeyBindings()
         self.session = PromptSession(key_bindings=self.bindings)
         self.config = {}
-        self.history = []
+        self.history: list[dict[str, str]] = []
         self.max_history_length = 25
         self.current_mode = TEMP_MODE
 
+    def is_stream(self) -> bool:
+        """Check if streaming is enabled"""
+        return self.config["STREAM"] == "true"
+
     def prepare_chat_loop(self) -> None:
         """Setup key bindings and history for chat mode"""
         self._setup_key_bindings()
@@ -232,7 +241,7 @@ class CLI:
             # Join the remaining lines and strip any extra whitespace
             return "\n".join(line.strip() for line in content_lines if line.strip())
 
-    def _get_type_number(self, key, _type: type, default=None):
+    def _get_number_with_type(self, key, _type: type, default=None):
         """Get number with type from config"""
         try:
             return _type(self.config.get(key, default))
@@ -245,10 +254,10 @@ class CLI:
         body = {
             "messages": message,
             "model": self.config.get("MODEL", "gpt-4o"),
-            "stream": self.config.get("STREAM", "true") == "true",
-            "temperature": self._get_type_number(key="TEMPERATURE", _type=float, default="0.7"),
-            "top_p": self._get_type_number(key="TOP_P", _type=float, default="1.0"),
-            "max_tokens": self._get_type_number(key="MAX_TOKENS", _type=int, default="1024"),
+            "stream": self.is_stream(),
+            "temperature": self._get_number_with_type(key="TEMPERATURE", _type=float, default="0.7"),
+            "top_p": self._get_number_with_type(key="TOP_P", _type=float, default="1.0"),
+            "max_tokens": self._get_number_with_type(key="MAX_TOKENS", _type=int, default="1024"),
         }
         with httpx.Client(timeout=120.0) as client:
             response = client.post(
@@ -315,8 +324,11 @@ class CLI:
 
     def _print_stream(self, response: httpx.Response) -> str:
         """Print response from LLM in streaming mode"""
+        self.console.print("Assistant:", style="bold green")
         full_completion = ""
         in_reasoning = False
+        cursor_chars = ["_", " "]
+        cursor_index = 0
 
         with Live() as live:
             for line in response.iter_lines():
@@ -332,30 +344,21 @@ class CLI:
                         reason, full_completion, in_reasoning
                     )
                 else:
-                    content = delta.get("content", "") or ""
                     full_completion, in_reasoning = self._process_regular_content(
-                        content, full_completion, in_reasoning
+                        delta.get("content", "") or "", full_completion, in_reasoning
                     )
 
-                live.update(Markdown(markup=full_completion), refresh=True)
-
+                live.update(Markdown(markup=full_completion + cursor_chars[cursor_index]), refresh=True)
+                cursor_index = (cursor_index + 1) % 2
+                time.sleep(0.005)  # Slow down the printing speed, avoiding screen flickering
+            live.update(Markdown(markup=full_completion), refresh=True)
         return full_completion
 
-    def _print_non_stream(self, response: httpx.Response) -> str:
+    def _print_normal(self, response: httpx.Response) -> str:
         """Print response from LLM in non-streaming mode"""
+        self.console.print("Assistant:", style="bold green")
         full_completion = jmespath.search(self.config.get("ANSWER_PATH", "choices[0].message.content"), response.json())
-        self.console.print(Markdown(full_completion))
-        return full_completion
-
-    def _print(self, response: httpx.Response, stream: bool = True) -> str:
-        """Print response from LLM and return full completion"""
-        if stream:
-            # Streaming response
-            full_completion = self._print_stream(response)
-        else:
-            # Non-streaming response
-            full_completion = self._print_non_stream(response)
-        self.console.print()  # Add a newline after the response to separate from the next input
+        self.console.print(Markdown(full_completion + "\n"))
         return full_completion
 
     def get_prompt_tokens(self) -> list[tuple[str, str]]:
@@ -368,9 +371,74 @@ class CLI:
         if len(self.history) > self.max_history_length:
             self.history = self.history[-self.max_history_length :]
 
+    def _handle_special_commands(self, user_input: str) -> Optional[bool]:
+        """Handle special command return: True-continue loop, False-exit loop, None-non-special command"""
+        if user_input.lower() == CMD_EXIT:
+            return False
+        if user_input.lower() == CMD_CLEAR and self.current_mode == CHAT_MODE:
+            self.history.clear()
+            self.console.print("Chat history cleared\n", style="bold yellow")
+            return True
+        if user_input.lower() == CMD_HISTORY:
+            self.console.print(self.history)
+            return True
+        return None
+
+    def _confirm_and_execute(self, content: str) -> None:
+        """Review, edit and execute the command"""
+        cmd = self._filter_command(content)
+        if not cmd:
+            self.console.print("No command generated", style="bold red")
+            return
+        self.console.print(Panel(cmd, title="Command", title_align="left", border_style="bold magenta", expand=False))
+        _input = Prompt.ask(
+            r"Execute command? \[e]dit, \[y]es, \[n]o",
+            choices=["y", "n", "e"],
+            default="n",
+            case_sensitive=False,
+            show_choices=False,
+        )
+        if _input == "y":  # execute cmd
+            self.console.print("Output:", style="bold green")
+            subprocess.call(cmd, shell=True)
+        elif _input == "e":  # edit cmd
+            cmd = prompt("Edit command, press enter to execute:\n", default=cmd)
+            self.console.print("Output:", style="bold green")
+            subprocess.call(cmd, shell=True)
+
+    def _build_messages(self, user_input: str) -> list[dict[str, str]]:
+        return [
+            {"role": "system", "content": self.get_system_prompt()},
+            *self.history,
+            {"role": "user", "content": user_input},
+        ]
+
+    def _handle_llm_response(self, response: httpx.Response, user_input: str) -> str:
+        """Print LLM response and update history"""
+        content = self._print_stream(response) if self.is_stream() else self._print_normal(response)
+        self.history.extend([{"role": "user", "content": user_input}, {"role": "assistant", "content": content}])
+        self._check_history_len()
+        return content
+
+    def _process_user_input(self, user_input: str) -> bool:
+        """Process user input and generate response"""
+        try:
+            response = self.post(self._build_messages(user_input))
+            content = self._handle_llm_response(response, user_input)
+            if self.current_mode == EXEC_MODE:
+                self._confirm_and_execute(content)
+            return True
+        except Exception as e:
+            self.console.print(f"Error: {e}", style="red")
+            return False
+
+    def get_system_prompt(self) -> str:
+        """Return system prompt for current mode"""
+        prompt = SHELL_PROMPT if self.current_mode == EXEC_MODE else DEFAULT_PROMPT
+        return prompt.format(_os=self.detect_os(), _shell=self.detect_shell())
+
     def _run_repl(self) -> None:
         """Run REPL loop, handling user input and generating responses, saving history, and executing commands"""
-        # Show REPL instructions
         self.prepare_chat_loop()
         self.console.print("""
 ██    ██  █████  ██  ██████ ██      ██
@@ -379,13 +447,13 @@ class CLI:
    ██    ██   ██ ██ ██      ██      ██
    ██    ██   ██ ██  ██████ ███████ ██
 """)
-        self.console.print("[bold]Press TAB to change in chat and exec mode[/bold]")
-        self.console.print("[bold]Type /clear to clear chat history[/bold]")
-        self.console.print("[bold]Type /his to see chat history[/bold]")
-        self.console.print("[bold]Press Ctrl+C or type /exit to exit[/bold]\n")
+        self.console.print("Press TAB to change in chat and exec mode", style="bold")
+        self.console.print("Type /clear to clear chat history", style="bold")
+        self.console.print("Type /his to see chat history", style="bold")
+        self.console.print("Press Ctrl+C or type /exit to exit\n", style="bold")
 
         while True:
-            # Get user input
+            self.console.print(Markdown("---"))
             user_input = self.session.prompt(self.get_prompt_tokens).strip()
             if not user_input:
                 continue
@@ -402,88 +470,21 @@ class CLI:
             elif user_input.lower() == CMD_HISTORY:
                 self.console.print(self.history)
                 continue
-            # Create appropriate system prompt based on mode
-            system_prompt = SHELL_PROMPT if self.current_mode == EXEC_MODE else DEFAULT_PROMPT
-            system_content = system_prompt.format(_os=self.detect_os(), _shell=self.detect_shell())
-
-            # Create message with system prompt and history
-            message = [{"role": "system", "content": system_content}]
-            message.extend(self.history)
-
-            # Add current user message
-            message.append({"role": "user", "content": user_input})
-
-            # Get response from LLM
-            try:
-                response = self.post(message)
-            except ValueError as e:
-                self.console.print(f"[red]Error: {e}[/red]")
-                return
-            except (httpx.ConnectError, httpx.HTTPStatusError) as e:
-                self.console.print(f"[red]Error: {e}[/red]")
+            if not self._process_user_input(user_input):
                 continue
-            self.console.print("\n[bold green]Assistant:[/bold green]")
-            try:
-                content = self._print(response, stream=self.config["STREAM"] == "true")
-            except Exception as e:
-                self.console.print(f"[red]Unknown Error: {e}[/red]")
-                continue
-
-            # Add user input and assistant response to history
-            self.history.append({"role": "user", "content": user_input})
-            self.history.append({"role": "assistant", "content": content})
-
-            self._check_history_len()
-
-            # Handle command execution in exec mode
-            if self.current_mode == EXEC_MODE:
-                content = self._filter_command(content)
-                if not content:
-                    self.console.print("[bold red]No command generated[/bold red]")
-                    continue
-                self.console.print(f"\n[bold magenta]Generated command:[/bold magenta] {content}")
-                if Confirm.ask("Execute this command?", default=False):
-                    subprocess.call(content, shell=True)
 
         self.console.print("[bold green]Exiting...[/bold green]")
 
     def _run_once(self, prompt: str, shell: bool = False) -> None:
         """Run once with given prompt"""
-        _os = self.detect_os()
-        _shell = self.detect_shell()
-        # Create appropriate system prompt based on mode
-        system_prompt = SHELL_PROMPT if shell else DEFAULT_PROMPT
-        system_content = system_prompt.format(_os=_os, _shell=_shell)
-
-        # Create message with system prompt and user input
-        message = [
-            {"role": "system", "content": system_content},
-            {"role": "user", "content": prompt},
-        ]
 
-        # Get response from LLM
         try:
-            response = self.post(message)
-        except (ValueError, httpx.ConnectError, httpx.HTTPStatusError) as e:
-            self.console.print(f"[red]Error: {e}[/red]")
-            return
+            response = self.post(self._build_messages(prompt))
+            content = self._handle_llm_response(response, prompt)
+            if shell:
+                self._confirm_and_execute(content)
         except Exception as e:
-            self.console.print(f"[red]Unknown Error: {e}[/red]")
-            return
-        self.console.print("\n[bold green]Assistant:[/bold green]")
-        content = self._print(response, stream=self.config["STREAM"] == "true")
-
-        # Handle shell mode execution
-        if shell:
-            content = self._filter_command(content)
-            if not content:
-                self.console.print("[bold red]No command generated[/bold red]")
-                return
-            self.console.print(f"\n[bold magenta]Generated command:[/bold magenta] {content}")
-            if Confirm.ask("Execute this command?", default=False):
-                returncode = subprocess.call(content, shell=True)
-                if returncode != 0:
-                    self.console.print(f"[bold red]Command failed with return code {returncode}[/bold red]")
+            self.console.print(f"[red]Error: {e}[/red]")
 
     def run(self, chat: bool, shell: bool, prompt: str) -> None:
         """Run the CLI"""
@@ -493,12 +494,11 @@ class CLI:
                 "[yellow]API key not set. Please set in ~/.config/yaicli/config.ini or AI_API_KEY env[/]"
             )
             raise typer.Exit(code=1)
-
-        # Handle chat mode
         if chat:
             self.current_mode = CHAT_MODE
             self._run_repl()
         else:
+            self.current_mode = EXEC_MODE if shell else TEMP_MODE
             self._run_once(prompt, shell)
 
 
@@ -507,13 +507,13 @@ def main(
     ctx: typer.Context,
     prompt: Annotated[Optional[str], typer.Argument(show_default=False, help="The prompt send to the LLM")] = None,
     chat: Annotated[
-        bool, typer.Option("--chat", "-c", help="Start in chat mode", rich_help_panel="Run Option")
+        bool, typer.Option("--chat", "-c", help="Start in chat mode", rich_help_panel="Run Options")
     ] = False,
     shell: Annotated[
-        bool, typer.Option("--shell", "-s", help="Generate and execute shell command", rich_help_panel="Run Option")
+        bool, typer.Option("--shell", "-s", help="Generate and execute shell command", rich_help_panel="Run Options")
     ] = False,
     verbose: Annotated[
-        bool, typer.Option("--verbose", "-V", help="Show verbose information", rich_help_panel="Run Option")
+        bool, typer.Option("--verbose", "-V", help="Show verbose information", rich_help_panel="Run Options")
     ] = False,
     template: Annotated[bool, typer.Option("--template", help="Show the config template.")] = False,
 ):

yaicli-0.0.12.dist-info/RECORD

@@ -1,7 +0,0 @@
-pyproject.toml,sha256=NZNzmBt0C_yyfCeVmJeLv05N-WyhNhI2oZCWP6A_nZo,1530
-yaicli.py,sha256=aYb-EIxPpT6hfQJ9gulTb1lBWaB7IanoNW4U3ZcFqi0,21263
-yaicli-0.0.12.dist-info/METADATA,sha256=KXtToNAbzOpGOtq7Wt-ofFj0_LM7kwvSUlrETWR8uwY,28095
-yaicli-0.0.12.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-yaicli-0.0.12.dist-info/entry_points.txt,sha256=gdduQwAuu_LeDqnDU81Fv3NPmD2tRQ1FffvolIP3S1Q,34
-yaicli-0.0.12.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-yaicli-0.0.12.dist-info/RECORD,,