yaicli 0.3.0__py3-none-any.whl → 0.3.1__py3-none-any.whl

pyproject.toml

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

yaicli/api.py

@@ -93,7 +93,11 @@ class ApiClient:
 
     def get_headers(self) -> Dict[str, str]:
         """Get the request headers."""
-        return {"Authorization": f"Bearer {self.api_key}", "Content-Type": "application/json"}
+        return {
+            "Authorization": f"Bearer {self.api_key}",
+            "Content-Type": "application/json",
+            "X-Title": "Yaicli",
+        }
 
     def _process_completion_response(self, response_json: Dict[str, Any]) -> Tuple[Optional[str], Optional[str]]:
         """Process the JSON response from a non-streamed completion request."""

yaicli/entry.py

@@ -130,7 +130,7 @@ def main(
     code: bool = typer.Option(
         False,
         "--code",
-        help="Generate and optionally execute a code block (non-interactive).",
+        help="Generate code in plaintext (non-interactive).",
         rich_help_panel="Code Options",
     ),
     # ------------------- Chat Options -------------------

{yaicli-0.3.0.dist-info → yaicli-0.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: yaicli
-Version: 0.3.0
+Version: 0.3.1
 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
@@ -465,6 +465,9 @@ ai --chat
 # Generate and execute shell commands
 ai --shell "Create a backup of my Documents folder"
 
+# Generate code snippets, default in Python
+ai --code "Write a Python function to sort a list"
+
 # Analyze code from a file
 cat app.py | ai "Explain what this code does"
 
@@ -475,17 +478,50 @@ ai --verbose "Explain quantum computing"
 ### Command Line Reference
 
 ```
-Usage: ai [OPTIONS] [PROMPT]
+ Usage: ai [OPTIONS] [PROMPT]
+
+ YAICLI: Your AI assistant in the command line.
+ Call with a PROMPT to get a direct answer, use --shell to execute as command, or use --chat for an interactive session.
+
+╭─ Arguments ─────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
+│   prompt      [PROMPT]  The prompt to send to the LLM. Reads from stdin if available. [default: None]                                   │
+╰─────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
+╭─ Options ───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
+│ --install-completion            Install completion for the current shell.                                                               │
+│ --show-completion               Show completion for the current shell, to copy it or customize the installation.                        │
+│ --help                -h        Show this message and exit.                                                                             │
+╰─────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
+╭─ LLM Options ───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
+│ --model        -M      TEXT                       Specify the model to use.                                                             │
+│ --temperature  -T      FLOAT RANGE [0.0<=x<=2.0]  Specify the temperature to use. [default: 0.7]                                        │
+│ --top-p        -P      FLOAT RANGE [0.0<=x<=1.0]  Specify the top-p to use. [default: 1.0]                                              │
+│ --max-tokens   -M      INTEGER RANGE [x>=1]       Specify the max tokens to use. [default: 1024]                                        │
+╰─────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
+╭─ Role Options ──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
+│ --role         -r      TEXT  Specify the assistant role to use. [default: DEFAULT]                                                      │
+│ --create-role          TEXT  Create a new role with the specified name.                                                                 │
+│ --delete-role          TEXT  Delete a role with the specified name.                                                                     │
+│ --list-roles                 List all available roles.                                                                                  │
+│ --show-role            TEXT  Show the role with the specified name.                                                                     │
+╰─────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
+╭─ Chat Options ──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
+│ --chat        -c        Start in interactive chat mode.                                                                                 │
+│ --list-chats            List saved chat sessions.                                                                                       │
+╰─────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
+╭─ Shell Options ─────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
+│ --shell  -s        Generate and optionally execute a shell command (non-interactive).                                                   │
+╰─────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
+╭─ Code Options ──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
+│ --code          Generate code in plaintext (non-interactive).                                                                           │
+╰─────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
+╭─ Other Options ─────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
+│ --verbose         -V                                                           Show verbose output (e.g., loaded config).               │
+│ --template                                                                     Show the default config file template and exit.          │
+│ --show-reasoning      --no-show-reasoning                                      Show reasoning content from the LLM. (default: True)     │
+│ --justify         -j                         [default|left|center|right|full]  Specify the justify to use. [default: default]           │
+╰─────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
 ```
 
-| Option       | Short | Description                         |
-|--------------|-------|-------------------------------------|
-| `--chat`     | `-c`  | Start in interactive chat mode      |
-| `--shell`    | `-s`  | Generate and execute shell commands |
-| `--help`     | `-h`  | Show help message and exit          |
-| `--verbose`  | `-V`  | Show detailed debug information     |
-| `--template` |       | Display the config template         |
-
 ### Interactive Mode Features
 
 <table>
@@ -640,6 +676,32 @@ cat demo.py | ai "Explain this code"
 cat error.log | ai "Why am I getting these errors in my Python app?"
 ```
 
+### Role Management
+
+```bash
+# Create a new role, you need to input the role description
+ai --create-role "Philosopher Master"
+
+# List all roles
+ai --list-roles
+
+# Show a role
+ai --show-role "Philosopher Master"
+
+# Delete a role
+ai --delete-role "Philosopher Master"
+```
+
+Once you create a role, you can use it in the `--role` option.
+
+```bash
+# Use a specific role
+ai --role "Philosopher Master" "What is the meaning of life?"
+
+# Use a role in chat
+ai --chat --role "Philosopher Master"
+```
+
 ### History Management
 
 YAICLI maintains a history of your interactions (default: 500 entries) stored in `~/.yaicli_history`. You can:
@@ -674,6 +736,23 @@ Output:
 109M    ./
 ```
 
+### Code Generation
+
+In code mode, select the language for code generation. If none is specified, Python is the default.
+
+The `--code` mode outputs plain text, making it easy to copy, paste, or redirect to a file, especially when using the standard model.
+
+When using a deep reasoning model, the thinking content is displayed with syntax highlighting. To disable this, use the `--no-show-reasoning` option or set `SHOW_REASONING` to `false` in the configuration.
+
+```bash
+$ ai --code 'Write a fib generator'
+def fib_generator():
+    a, b = 0, 1
+    while True:
+        yield a
+        a, b = b, a + b
+```
+
 ### Chat Mode Example
 
 ```bash
@@ -747,6 +826,14 @@ Output:
 ...
 ```
 
+### Code Mode Example
+
+```bash
+$ ai --code "write a fib generator" --model deepseek-r1
+```
+
+![fib code example](artwork/reasoning_code_example.png)
+
 ## 💻 Technical Details
 
 ### Architecture

{yaicli-0.3.0.dist-info → yaicli-0.3.1.dist-info}/RECORD

@@ -1,20 +1,20 @@
-pyproject.toml,sha256=aRTnC0bBuX6dPMSLTCZOAB0M5AnpW_eVvLfmc0nZZNg,1519
+pyproject.toml,sha256=C4r3nsve6z4u7vKI2eIKvQ4YWoiGwAYL72w_YcjTkcc,1519
 yaicli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-yaicli/api.py,sha256=mRb86Ej1MVR1AbJimX7VWYd0D1pNarqylK4rbAYLSIA,13598
+yaicli/api.py,sha256=9kRozzxBKduQsda3acnxzvOD9wRLL0cH182L4ddHY8E,13666
 yaicli/chat_manager.py,sha256=I7BAMz91FLYT6x69wbomtAGLx0WsoTwS4Wo0MgP6P9I,10644
 yaicli/cli.py,sha256=HnfmD-7mLsaUCJqQYehsfxonuyJ11B5zi3lbKbQxrH0,22684
 yaicli/config.py,sha256=xtzgXApM93zCqSUxmVSBdph0co_NKfEUU3hWtPe8qvM,6236
 yaicli/console.py,sha256=291F4hGksJtxYpg_mehepCIJ-eB2MaDNIyv1JAMgJ1Y,1985
 yaicli/const.py,sha256=iOQNG6M4EBmKgbwZdNqRsHrcQ7Od1nKOyLAqUhfMEBM,7020
-yaicli/entry.py,sha256=z59Ojr_fuj6ykEdgusWzedKO3VA1ytW6OlMFdUuyj4A,7478
+yaicli/entry.py,sha256=Yp0Z--x-7dowrz-h8hJJ4_BoCzuDjS11NcM8YgFzUoY,7460
 yaicli/exceptions.py,sha256=ndedSdE0uaxxHrWN944BkbhMfRMSMxGDfmqmCKCGJco,924
 yaicli/history.py,sha256=s-57X9FMsaQHF7XySq1gGH_jpd_cHHTYafYu2ECuG6M,2472
 yaicli/printer.py,sha256=nXpralD5qZJQga3OTdEPhj22g7UoF-4mJbZeOtWXojo,12430
 yaicli/render.py,sha256=mB1OT9859_PTwI9f-KY802lPaeQXKRw6ls_5jN21jWc,511
 yaicli/roles.py,sha256=bhXpLnGTPRZp3-K1Tt6ppTsuG2v9S0RAXikfMFhDs_U,9144
 yaicli/utils.py,sha256=MLvb-C5n19AD9Z1nW4Z3Z43ZKNH8STxQmNDnL7mq26E,4490
-yaicli-0.3.0.dist-info/METADATA,sha256=4QDEAn4I3sq89qwWQ8e6cQ5ziCu_-HbH03WwS4ssGvE,34744
-yaicli-0.3.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-yaicli-0.3.0.dist-info/entry_points.txt,sha256=iMhGm3btBaqrknQoF6WCg5sdx69ZyNSC73tRpCcbcLw,63
-yaicli-0.3.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-yaicli-0.3.0.dist-info/RECORD,,
+yaicli-0.3.1.dist-info/METADATA,sha256=fAGgZNCTckGuIS8Us6un9AC-60I5ZsEfw6k5RRHK7XU,45320
+yaicli-0.3.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+yaicli-0.3.1.dist-info/entry_points.txt,sha256=iMhGm3btBaqrknQoF6WCg5sdx69ZyNSC73tRpCcbcLw,63
+yaicli-0.3.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+yaicli-0.3.1.dist-info/RECORD,,