yaicli 0.5.7__py3-none-any.whl → 0.5.8__py3-none-any.whl

pyproject.toml

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

yaicli/chat.py

@@ -10,8 +10,8 @@ from rich.table import Table
 
 from .config import cfg
 from .console import YaiConsole, get_console
-from .schemas import ChatMessage
 from .exceptions import ChatDeleteError, ChatLoadError, ChatSaveError
+from .schemas import ChatMessage
 from .utils import option_callback
 
 console: YaiConsole = get_console()
@@ -68,7 +68,9 @@ class Chat:
                 data = json.load(f)
                 self.title = data.get("title", self.title)
                 self.date = data.get("date", self.date)
-                self.history = [ChatMessage(role=msg["role"], content=msg["content"]) for msg in data.get("history", [])]
+                self.history = [
+                    ChatMessage(role=msg["role"], content=msg["content"]) for msg in data.get("history", [])
+                ]
             return True
         except (json.JSONDecodeError, OSError) as e:
             raise ChatLoadError(f"Error loading chat: {e}") from e

yaicli/cli.py

@@ -17,7 +17,7 @@ from rich.panel import Panel
 from rich.prompt import Prompt
 
 from .chat import Chat, FileChatManager, chat_mgr
-from .client import LitellmClient, ChatMessage
+from .client import ChatMessage, LitellmClient
 from .config import cfg
 from .console import get_console
 from .const import (

yaicli/client.py

@@ -11,7 +11,7 @@ from rich.panel import Panel
 
 from .config import cfg
 from .console import get_console
-from .schemas import LLMResponse, ChatMessage, ToolCall
+from .schemas import ChatMessage, LLMResponse, ToolCall
 from .tools import (
     Function,
     FunctionName,

yaicli/config.py

@@ -1,8 +1,8 @@
 import configparser
+import json
 from dataclasses import dataclass
 from functools import lru_cache
 from os import getenv
-import json
 from typing import Optional
 
 from rich import get_console

yaicli/const.py

@@ -1,4 +1,11 @@
-from enum import StrEnum
+try:
+    from enum import StrEnum
+except ImportError:
+    from enum import Enum
+
+    class StrEnum(str, Enum):
+        """Compatible with python below 3.11"""
+
 from pathlib import Path
 from tempfile import gettempdir
 from typing import Any, Literal, Optional
@@ -63,21 +70,6 @@ DEFAULT_SHOW_FUNCTION_OUTPUT: BOOL_STR = "true"
 DEFAULT_REASONING_EFFORT: Optional[Literal["low", "high", "medium"]] = None
 
 
-class EventTypeEnum(StrEnum):
-    """Enumeration of possible event types from the SSE stream."""
-
-    ERROR = "error"
-    REASONING = "reasoning"
-    REASONING_END = "reasoning_end"
-    CONTENT = "content"
-    FINISH = "finish"
-    TOOL_CALL_START = "tool_call_start"
-    TOOL_CALL_DELTA = "tool_call_delta"
-    TOOL_CALL_END = "tool_call_end"
-    TOOL_RESULT = "tool_result"
-    TOOL_CALLS_FINISH = "tool_calls_finish"
-
-
 SHELL_PROMPT = """You are YAICLI, a shell command generator.
 The context conversation may contain other types of messages, 
 but you should only respond with a single valid {_shell} shell command for {_os}.

yaicli/printer.py

@@ -8,8 +8,8 @@ from rich.live import Live
 from .client import RefreshLive
 from .config import Config, get_config
 from .console import YaiConsole, get_console
-from .schemas import ChatMessage
 from .render import Markdown, plain_formatter
+from .schemas import ChatMessage
 
 if TYPE_CHECKING:
     from .schemas import LLMResponse

{yaicli-0.5.7.dist-info → yaicli-0.5.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: yaicli
-Version: 0.5.7
+Version: 0.5.8
 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

{yaicli-0.5.7.dist-info → yaicli-0.5.8.dist-info}/RECORD

@@ -1,15 +1,15 @@
-pyproject.toml,sha256=PCKvDYSn9V2qnxplRgsrbYZfcPjEzj1PvguJVwEtDo0,1963
+pyproject.toml,sha256=uIesu0u5n1nVFJ4MJ4ZO4G9hr4EGlwcysyKybIG3PyM,1963
 yaicli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-yaicli/chat.py,sha256=DeTmOeBPU-oiOAIaDj2h-auJor0GyVVhrViLYF6zGIM,13638
-yaicli/cli.py,sha256=1Flt0FgrKzbabjJpZJCcvnUjXaaLwyD2paRyMixEQV0,22985
-yaicli/client.py,sha256=PzfoW0lZ2QHxt2IBLGl-l6CGpq98MsxPiag0QxwRKsw,16565
-yaicli/config.py,sha256=65k7gERkOIJ5q4GN-CU7VYZgupShFgsMHT0eWPtCh_U,6535
+yaicli/chat.py,sha256=_emvZEdgMBth2nQGaNWPf0P45oW2k3bpuIwqsxFcM5A,13676
+yaicli/cli.py,sha256=u4rq04CdLiL-IBzQ0IimPyFAh9XZg9YUTuvM4CxE3pE,22985
+yaicli/client.py,sha256=kdEI4HP1zZQHUf-7kNbE9ItpyRN8zcAGyKeBSb1R4Ro,16565
+yaicli/config.py,sha256=HrWYcelLXE61XX719eVcuuo3292xxf1BNQznWdvjQFQ,6535
 yaicli/console.py,sha256=vARPJd-3lafutsQWrGntQVjLrYqaJD3qisN82pmuhjU,1973
-yaicli/const.py,sha256=47I8I5hkJmLtPAhmesMvg7sydDSG_MB94BoRPtdBrUg,8442
+yaicli/const.py,sha256=mt_6m2jo5-tHEmk3DxKCEc5ek9DcyQGHFAn_HEeMP3k,8155
 yaicli/entry.py,sha256=gKzN8Yar3tpBd2Z2a80gD3k0W4Sf3lL7jdyws-2y-H0,8687
 yaicli/exceptions.py,sha256=WBYg8OTJJzaj7lt6HE7ZyBoe5T6A3yZRNCRfWd4iN0c,372
 yaicli/history.py,sha256=s-57X9FMsaQHF7XySq1gGH_jpd_cHHTYafYu2ECuG6M,2472
-yaicli/printer.py,sha256=a409R_4-ppNnen31Pt7KvaaNAFVCzBARYC0T0_EtMbU,8369
+yaicli/printer.py,sha256=J-QPrb0q4Zrx18vwBGIBDQk7RzT4wXJPwdJ9oXLoEVg,8369
 yaicli/render.py,sha256=k8o2P8fI44PJlyQbs7gmMiu2x2prwajdWn5JIt15BIA,505
 yaicli/role.py,sha256=PfwiVJIlzg7EzlvMM-kIy6vBK0d5d_J4M1I_fIZGnWk,7399
 yaicli/schemas.py,sha256=PiuSY7ORZaA4OL_tYm0inwqirHp5M-F3zcCipLwsH9E,571
@@ -17,8 +17,8 @@ yaicli/tools.py,sha256=d-5LXbEB-1Uq5VKSgwlAiNDVOGrHkku2DpmZoorq1zw,3098
 yaicli/utils.py,sha256=bpo3Xhozpxsaci3FtEIKZ32l4ZdyWMsrHjYGX0tB4J4,4541
 yaicli/functions/__init__.py,sha256=_FJooQ9GkijG8xLwuU0cr5GBrGnC9Nc6bnCeUjrsT0k,1271
 yaicli/functions/buildin/execute_shell_command.py,sha256=unl1-F8p6QZajeHdA0u5UpURMJM0WhdWMUWCCCHVRcI,1320
-yaicli-0.5.7.dist-info/METADATA,sha256=VCCVoup7CdV0bKytlc9UtO2tLtAePXbm9faHoYZHRJo,50349
-yaicli-0.5.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-yaicli-0.5.7.dist-info/entry_points.txt,sha256=iYVyQP0PJIm9tQnlQheqT435kK_xdGoi5j9aswGV9hA,66
-yaicli-0.5.7.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-yaicli-0.5.7.dist-info/RECORD,,
+yaicli-0.5.8.dist-info/METADATA,sha256=BRZEKpxpZ6kPABv88FGl4ipt8glgJdAn-9ZxiDLAAeo,50349
+yaicli-0.5.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+yaicli-0.5.8.dist-info/entry_points.txt,sha256=iYVyQP0PJIm9tQnlQheqT435kK_xdGoi5j9aswGV9hA,66
+yaicli-0.5.8.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+yaicli-0.5.8.dist-info/RECORD,,