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

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "yaicli"
-version = "0.0.14"
+version = "0.0.16"
 description = "A simple CLI tool to interact with LLM"
 authors = [{ name = "belingud", email = "im.victor@qq.com" }]
 readme = "README.md"

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: yaicli
-Version: 0.0.14
+Version: 0.0.16
 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
@@ -233,7 +233,7 @@ YAICLI is a compact yet potent command-line AI assistant, allowing you to engage
 
 Support regular and deep thinking models.
 
-> [!WARNING] 
+> [!WARNING]
 > This is a work in progress, some features could change or be removed in the future.
 
 ## Features
@@ -259,6 +259,8 @@ Support regular and deep thinking models.
 
 - **Keyboard Shortcuts**:
   - Tab to switch between Chat and Execute modes
+  - `↑/↓` to navigate history
+  - `Ctrl+R` to search history
 
 - **History**:
   - Save and recall previous queries
@@ -319,6 +321,7 @@ ANSWER_PATH=choices[0].message.content
 # true: streaming response
 # false: non-streaming response
 STREAM=true
+CODE_THEME=monokia
 
 TEMPERATURE=0.7
 TOP_P=1.0
@@ -337,9 +340,11 @@ Below are the available configuration options and override environment variables
 - **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
+- **CODE_THEME**: Theme for code blocks, default: monokia, env: YAI_CODE_THEME
 - **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
+- **MAX_HISTORY**: Max history size, default: 500, env: YAI_MAX_HISTORY
 
 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.
 
@@ -392,6 +397,12 @@ If you not sure how to config `COMPLETION_PATH` and `ANSWER_PATH`, here is a gui
    ```
     We are looking for the `text` field, so the path should be 1.Key `content`, 2.First obj `[0]`, 3.Key `text`. So it should be `content.[0].text`.
 
+**CODE_THEME**
+
+You can find the list of code theme here: https://pygments.org/styles/
+
+Default: monokia
+![monikia](artwork/monokia.png)
 
 ## Usage
 
@@ -430,10 +441,10 @@ Run Options:
 ```bash
 ai -h
 
-Usage: ai [OPTIONS] [PROMPT]                                                                                                                                              
-                                                                                                                                                                           
- yaicli - Your AI interface in cli.                                                                                                                                        
-                                                                                                                                                                           
+Usage: ai [OPTIONS] [PROMPT]
+
+ yaicli - Your AI interface in cli.
+
 ╭─ Arguments ─────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
 │   prompt      [PROMPT]  The prompt send to the LLM                                                                                                                      │
 ╰─────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
@@ -468,6 +479,25 @@ In Execute mode:
 3. Review the command
 4. Confirm to execute or reject
 
+### Keyboard Shortcuts
+- `Tab`: Switch between Chat and Execute modes
+- `Ctrl+C`: Exit
+- `Ctrl+R`: Search history
+- `↑/↓`: Navigate history
+
+### Stdin
+You can also pipe input to YAICLI:
+```bash
+echo "What is the capital of France?" | ai
+```
+
+```bash
+cat demo.py | ai "How to use this tool?"
+```
+
+### History
+Support max history size. Set MAX_HISTORY in config file. Default is 500.
+
 ## Examples
 
 ### Have a Chat
@@ -484,9 +514,10 @@ The capital of France is Paris.
 $ ai -s 'Check the current directory size'
 Assistant:
 du -sh .
-
-Generated command: du -sh .
-Execute this command? [y/n/e] (n): e
+╭─ Command ─╮
+│ du -sh .  │
+╰───────────╯
+Execute command? [e]dit, [y]es, [n]o (n): e
 Edit command, press enter to execute:
 du -sh ./
 Output:
@@ -530,9 +561,10 @@ Certainly! Here’s a brief overview of the solar system:
 🚀 > Check the current directory size
 Assistant:
 du -sh .
-
-Generated command: du -sh .
-Execute this command? [y/n/e] (n): e
+╭─ Command ─╮
+│ du -sh .  │
+╰───────────╯
+Execute command? [e]dit, [y]es, [n]o (n): e
 Edit command, press enter to execute:
 du -sh ./
 Output:
@@ -544,11 +576,13 @@ Output:
 
 ```bash
 $ ai --shell "Find all PDF files in my Downloads folder"
-
-Generated command: find ~/Downloads -type f -name "*.pdf"
-Execute this command? [y/n]: y
-
-Executing command: find ~/Downloads -type f -name "*.pdf"
+Assistant:
+find ~/Downloads -type f -name "*.pdf"
+╭─ Command ──────────────────────────────╮
+│ find ~/Downloads -type f -name "*.pdf" │
+╰────────────────────────────────────────╯
+Execute command? [e]dit, [y]es, [n]o (n): y
+Output:
 
 /Users/username/Downloads/document1.pdf
 /Users/username/Downloads/report.pdf
@@ -562,7 +596,7 @@ YAICLI is built using several Python libraries:
 - **Typer**: Provides the command-line interface
 - **Rich**: Provides terminal content formatting and beautiful display
 - **prompt_toolkit**: Provides interactive command-line input experience
-- **requests**: Handles API requests
+- **httpx**: Handles API requests
 - **jmespath**: Parses JSON responses
 
 ## Contributing

yaicli-0.0.16.dist-info/RECORD

@@ -0,0 +1,7 @@
+pyproject.toml,sha256=95wyxycfXtMy_hraLxEt3JI4Ps4IcDWWn5cyvARlopo,1452
+yaicli.py,sha256=ZJTejElstF1BEywXV5Hb5gOONXMbHdj7sHrEJzBqCe8,25153
+yaicli-0.0.16.dist-info/METADATA,sha256=xS7MlrbSU7N_CYp1Urbgr3iYu6eP8YeVWCbo2Zi9aAk,29932
+yaicli-0.0.16.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+yaicli-0.0.16.dist-info/entry_points.txt,sha256=gdduQwAuu_LeDqnDU81Fv3NPmD2tRQ1FffvolIP3S1Q,34
+yaicli-0.0.16.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+yaicli-0.0.16.dist-info/RECORD,,

yaicli.py

@@ -2,9 +2,10 @@ import configparser
 import json
 import platform
 import subprocess
+import sys
 import time
 from os import getenv
-from os.path import basename, pathsep
+from os.path import basename, exists, pathsep, devnull
 from pathlib import Path
 from typing import Annotated, Optional, Union
 
@@ -14,7 +15,7 @@ import typer
 from distro import name as distro_name
 from prompt_toolkit import PromptSession, prompt
 from prompt_toolkit.completion import WordCompleter
-from prompt_toolkit.history import FileHistory
+from prompt_toolkit.history import FileHistory, _StrOrBytesPath
 from prompt_toolkit.key_binding import KeyBindings, KeyPressEvent
 from prompt_toolkit.keys import Keys
 from rich.console import Console
@@ -56,9 +57,11 @@ DEFAULT_CONFIG_MAP = {
     "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"},
+    "CODE_THEME": {"value": "monokia", "env_key": "YAI_CODE_THEME"},
     "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"},
+    "MAX_HISTORY": {"value": "500", "env_key": "YAI_MAX_HISTORY"},
 }
 
 DEFAULT_CONFIG_INI = """[core]
@@ -79,10 +82,13 @@ ANSWER_PATH=choices[0].message.content
 # true: streaming response
 # false: non-streaming response
 STREAM=true
+CODE_THEME=monokia
 
 TEMPERATURE=0.7
 TOP_P=1.0
-MAX_TOKENS=1024"""
+MAX_TOKENS=1024
+
+MAX_HISTORY=500"""
 
 app = typer.Typer(
     name="yaicli",
@@ -98,6 +104,64 @@ class CasePreservingConfigParser(configparser.RawConfigParser):
         return optionstr
 
 
+class LimitedFileHistory(FileHistory):
+    def __init__(self, filename: _StrOrBytesPath, max_entries: int = 500, trim_every: int = 5):
+        """Limited file history
+        Args:
+            filename (str): path to history file
+            max_entries (int): maximum number of entries to keep
+            trim_every (int): trim history every `trim_every` appends
+
+        Example:
+            >>> history = LimitedFileHistory("~/.yaicli_history", max_entries=500, trim_every=10)
+            >>> history.append_string("echo hello")
+            >>> history.append_string("echo world")
+            >>> session = PromptSession(history=history)
+        """
+        self.max_entries = max_entries
+        self._append_count = 0
+        self._trim_every = trim_every
+        super().__init__(filename)
+
+    def store_string(self, string: str) -> None:
+        # Call the original method to deposit a new record
+        super().store_string(string)
+
+        self._append_count += 1
+        if self._append_count >= self._trim_every:
+            self._trim_history()
+            self._append_count = 0
+
+    def _trim_history(self):
+        if not exists(self.filename):
+            return
+
+        with open(self.filename, "r", encoding="utf-8") as f:
+            lines = f.readlines()
+
+        # By record: each record starts with "# timestamp" followed by a number of "+lines".
+        entries = []
+        current_entry = []
+
+        for line in lines:
+            if line.startswith("# "):
+                if current_entry:
+                    entries.append(current_entry)
+                current_entry = [line]
+            elif line.startswith("+") or line.strip() == "":
+                current_entry.append(line)
+
+        if current_entry:
+            entries.append(current_entry)
+
+        # Keep the most recent max_entries row (the next row is newer)
+        trimmed_entries = entries[-self.max_entries :]
+
+        with open(self.filename, "w", encoding="utf-8") as f:
+            for entry in trimmed_entries:
+                f.writelines(entry)
+
+
 class CLI:
     CONFIG_PATH = Path("~/.config/yaicli/config.ini").expanduser()
 
@@ -105,7 +169,16 @@ class CLI:
         self.verbose = verbose
         self.console = Console()
         self.bindings = KeyBindings()
+        # Disable nonatty warning
+        _origin_stderr = None
+        if not sys.stderr.isatty():
+            _origin_stderr = sys.stderr
+            sys.stderr = open(devnull, "w")
         self.session = PromptSession(key_bindings=self.bindings)
+        # Restore stderr
+        if _origin_stderr:
+            sys.stderr.close()
+            sys.stderr = _origin_stderr
         self.config = {}
         self.history: list[dict[str, str]] = []
         self.max_history_length = 25
@@ -124,7 +197,9 @@ class CLI:
             key_bindings=self.bindings,
             completer=WordCompleter(["/clear", "/exit", "/his"]),
             complete_while_typing=True,
-            history=FileHistory(Path("~/.yaicli_history").expanduser()),
+            history=LimitedFileHistory(
+                Path("~/.yaicli_history").expanduser(), max_entries=int(self.config["MAX_HISTORY"])
+            ),
             enable_history_search=True,
         )
 
@@ -194,7 +269,7 @@ class CLI:
         if current_platform in ("Windows", "nt"):
             is_powershell = len(getenv("PSModulePath", "").split(pathsep)) >= 3
             return "powershell.exe" if is_powershell else "cmd.exe"
-        return basename(getenv("SHELL", "/bin/sh"))
+        return basename(getenv("SHELL", None) or "/bin/sh")
 
     def _filter_command(self, command: str) -> Optional[str]:
         """Filter out unwanted characters from command
@@ -285,8 +360,7 @@ class CLI:
         if not line:
             return None
 
-        if isinstance(line, bytes):
-            line = line.decode("utf-8")
+        line = str(line)
         if not line.startswith("data: "):
             return None
 
@@ -325,12 +399,12 @@ 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 = ""
+        full_content = ""
         in_reasoning = False
         cursor_chars = ["_", " "]
         cursor_index = 0
 
-        with Live() as live:
+        with Live(console=self.console) as live:
             for line in response.iter_lines():
                 json_data = self._parse_stream_line(line)
                 if not json_data:
@@ -340,26 +414,25 @@ class CLI:
                 reason = self.get_reasoning_content(delta)
 
                 if reason is not None:
-                    full_completion, in_reasoning = self._process_reasoning_content(
-                        reason, full_completion, in_reasoning
-                    )
+                    full_content, in_reasoning = self._process_reasoning_content(reason, full_content, in_reasoning)
                 else:
-                    full_completion, in_reasoning = self._process_regular_content(
-                        delta.get("content", "") or "", full_completion, in_reasoning
+                    full_content, in_reasoning = self._process_regular_content(
+                        delta.get("content", "") or "", full_content, in_reasoning
                     )
 
-                live.update(Markdown(markup=full_completion + cursor_chars[cursor_index]), refresh=True)
+                cursor = cursor_chars[cursor_index]
+                live.update(Markdown(markup=full_content + cursor, code_theme=self.config["CODE_THEME"]), 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
+            live.update(Markdown(markup=full_content, code_theme=self.config["CODE_THEME"]), refresh=True)
+        return full_content
 
     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 + "\n"))
-        return full_completion
+        full_content = jmespath.search(self.config.get("ANSWER_PATH", "choices[0].message.content"), response.json())
+        self.console.print(Markdown(full_content + "\n", code_theme=self.config["CODE_THEME"]))
+        return full_content
 
     def get_prompt_tokens(self) -> list[tuple[str, str]]:
         """Return prompt tokens for current mode"""
@@ -447,6 +520,7 @@ class CLI:
    ██    ██   ██ ██ ██      ██      ██
    ██    ██   ██ ██  ██████ ███████ ██
 """)
+        self.console.print("↑/↓: navigate in history")
         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")
@@ -489,6 +563,18 @@ class CLI:
     def run(self, chat: bool, shell: bool, prompt: str) -> None:
         """Run the CLI"""
         self.load_config()
+        if self.verbose:
+            self.console.print(f"CODE_THEME: {self.config['CODE_THEME']}")
+            self.console.print(f"ANSWER_PATH: {self.config['ANSWER_PATH']}")
+            self.console.print(f"COMPLETION_PATH: {self.config['COMPLETION_PATH']}")
+            self.console.print(f"BASE_URL: {self.config['BASE_URL']}")
+            self.console.print(f"MODEL: {self.config['MODEL']}")
+            self.console.print(f"SHELL_NAME: {self.config['SHELL_NAME']}")
+            self.console.print(f"OS_NAME: {self.config['OS_NAME']}")
+            self.console.print(f"STREAM: {self.config['STREAM']}")
+            self.console.print(f"TEMPERATURE: {self.config['TEMPERATURE']}")
+            self.console.print(f"TOP_P: {self.config['TOP_P']}")
+            self.console.print(f"MAX_TOKENS: {self.config['MAX_TOKENS']}")
         if not self.config.get("API_KEY"):
             self.console.print(
                 "[yellow]API key not set. Please set in ~/.config/yaicli/config.ini or AI_API_KEY env[/]"
@@ -518,6 +604,11 @@ def main(
     template: Annotated[bool, typer.Option("--template", help="Show the config template.")] = False,
 ):
     """yaicli - Your AI interface in cli."""
+    # Check for stdin input (from pipe or redirect)
+    if not sys.stdin.isatty():
+        stdin_content = sys.stdin.read()
+        prompt = f"{stdin_content}\n\n{prompt}"
+
     if prompt == "":
         typer.echo("Empty prompt, ignored")
         return

yaicli-0.0.14.dist-info/RECORD

@@ -1,7 +0,0 @@
-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,,