PyPI - hitmos - Versions diffs - 0.0.1__tar.gz - Mend

hitmos 0.0.1__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (28) hide show

hitmos-0.0.1/.github/workflows/release.yml +63 -0
hitmos-0.0.1/.gitignore +10 -0
hitmos-0.0.1/.python-version +1 -0
hitmos-0.0.1/CITATION.cff +25 -0
hitmos-0.0.1/PKG-INFO +196 -0
hitmos-0.0.1/README.md +163 -0
hitmos-0.0.1/hitmos/__init__.py +1 -0
hitmos-0.0.1/hitmos/app.py +163 -0
hitmos-0.0.1/hitmos/chat.py +34 -0
hitmos-0.0.1/hitmos/cli.py +46 -0
hitmos-0.0.1/hitmos/client.py +131 -0
hitmos-0.0.1/hitmos/commands.py +37 -0
hitmos-0.0.1/hitmos/config.py +67 -0
hitmos-0.0.1/hitmos/constants.py +41 -0
hitmos-0.0.1/hitmos/context.py +119 -0
hitmos-0.0.1/hitmos/exceptions.py +22 -0
hitmos-0.0.1/hitmos/methods/__init__.py +45 -0
hitmos-0.0.1/hitmos/methods/base.py +33 -0
hitmos-0.0.1/hitmos/methods/edit_file.py +23 -0
hitmos-0.0.1/hitmos/methods/list_directory.py +20 -0
hitmos-0.0.1/hitmos/methods/read_file.py +17 -0
hitmos-0.0.1/hitmos/methods/run_command.py +31 -0
hitmos-0.0.1/hitmos/methods/write_file.py +18 -0
hitmos-0.0.1/hitmos/ui.py +217 -0
hitmos-0.0.1/hitmos/utils.py +10 -0
hitmos-0.0.1/main.py +8 -0
hitmos-0.0.1/pyproject.toml +60 -0
hitmos-0.0.1/uv.lock +425 -0

hitmos-0.0.1/.github/workflows/release.yml ADDED Viewed

@@ -0,0 +1,63 @@
+name: Release
+on:
+  push:
+    branches:
+      - master
+jobs:
+  release:
+    runs-on: ubuntu-latest
+    permissions:
+      contents: write
+    steps:
+      - uses: actions/checkout@v4
+        with:
+          fetch-depth: 0
+      - name: Read version
+        id: ver
+        run: |
+          VERSION=$(python3 -c "import tomllib; print(tomllib.load(open('pyproject.toml','rb'))['project']['version'])")
+          echo "version=$VERSION" >> "$GITHUB_OUTPUT"
+      - name: Check if tag already exists
+        id: tag
+        run: |
+          if git rev-parse "v${{ steps.ver.outputs.version }}" >/dev/null 2>&1; then
+            echo "exists=true" >> "$GITHUB_OUTPUT"
+            echo "Tag v${{ steps.ver.outputs.version }} already exists — skipping release."
+          else
+            echo "exists=false" >> "$GITHUB_OUTPUT"
+          fi
+      - uses: actions/setup-python@v5
+        if: steps.tag.outputs.exists == 'false'
+        with:
+          python-version: "3.13"
+      - name: Build
+        if: steps.tag.outputs.exists == 'false'
+        run: |
+          pip install build
+          python -m build
+      - name: Publish to PyPI
+        if: steps.tag.outputs.exists == 'false'
+        uses: pypa/gh-action-pypi-publish@release/v1
+        with:
+          password: ${{ secrets.PYPI_TOKEN }}
+      - name: Create tag and GitHub release
+        if: steps.tag.outputs.exists == 'false'
+        env:
+          GH_TOKEN: ${{ github.token }}
+        run: |
+          git config user.name "github-actions[bot]"
+          git config user.email "github-actions[bot]@users.noreply.github.com"
+          git tag -a "v${{ steps.ver.outputs.version }}" -m "Release v${{ steps.ver.outputs.version }}"
+          git push origin "v${{ steps.ver.outputs.version }}"
+          gh release create "v${{ steps.ver.outputs.version }}" dist/* \
+            --title "v${{ steps.ver.outputs.version }}" \
+            --generate-notes

hitmos-0.0.1/.gitignore ADDED Viewed

@@ -0,0 +1,10 @@
+# Python-generated files
+__pycache__/
+*.py[oc]
+build/
+dist/
+wheels/
+*.egg-info
+# Virtual environments
+.venv

hitmos-0.0.1/.python-version ADDED Viewed

	@@ -0,0 +1 @@
1	+ 3.13

hitmos-0.0.1/CITATION.cff ADDED Viewed

@@ -0,0 +1,25 @@
+cff-version: 1.2.0
+message: "If you use this software, please cite it as below."
+type: software
+title: "Hitmos"
+abstract: "AI terminal assistant powered by OpenRouter — minimal, fast, developer UX."
+authors:
+  - family-names: "NEFOR"
+    given-names: "White"
+    email: n7for8572@gmail.com
+repository-code: "https://github.com/ndugram/hitmos"
+url: "https://github.com/ndugram/hitmos"
+license: MIT
+version: "0.1.0"
+date-released: "2026-06-06"
+keywords:
+  - ai
+  - cli
+  - assistant
+  - openrouter
+  - llm
+  - terminal
+  - chat
+  - deepseek
+  - developer-tools
+  - python

hitmos-0.0.1/PKG-INFO ADDED Viewed

@@ -0,0 +1,196 @@
+Metadata-Version: 2.4
+Name: hitmos
+Version: 0.0.1
+Summary: AI terminal assistant powered by OpenRouter — minimal, fast, developer UX
+Project-URL: Homepage, https://github.com/ndugram/hitmos
+Project-URL: Documentation, https://github.com/ndugram/hitmos
+Project-URL: Repository, https://github.com/ndugram/hitmos
+Project-URL: Issues, https://github.com/ndugram/hitmos/issues
+Author-email: White NEFOR <n7for8572@gmail.com>
+Maintainer-email: White NEFOR <n7for8572@gmail.com>
+License: MIT
+Keywords: ai,assistant,chat,cli,deepseek,developer-tools,llm,openrouter,python,terminal
+Classifier: Development Status :: 4 - Beta
+Classifier: Environment :: Console
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Classifier: Topic :: Terminals
+Classifier: Topic :: Utilities
+Classifier: Typing :: Typed
+Requires-Python: >=3.13
+Requires-Dist: fasthttp-client>=1.3.13
+Requires-Dist: orjson>=3.9.0
+Requires-Dist: pydantic-settings>=2.0.0
+Requires-Dist: questionary>=2.1.1
+Requires-Dist: rich>=15.0.0
+Requires-Dist: typer>=0.26.7
+Description-Content-Type: text/markdown
+<p align="center">
+  <img src="https://raw.githubusercontent.com/ndugram/hitmos/master/docs/logo.png" style="background:white; padding:12px; border-radius:10px; width:300">
+</p>
+<p align="center">
+    <em>AI terminal assistant powered by OpenRouter — minimal, fast, developer UX.</em>
+</p>
+<p align="center">
+<a href="https://pypi.org/project/hitmos" target="_blank">
+    <img src="https://img.shields.io/pypi/v/hitmos?color=%2300d7af&label=pypi%20package" alt="Package version">
+</a>
+<a href="https://pypi.org/project/hitmos" target="_blank">
+    <img src="https://img.shields.io/pypi/pyversions/hitmos.svg?color=%2300d7af" alt="Supported Python versions">
+</a>
+<a href="https://pypi.org/project/hitmos" target="_blank">
+    <img src="https://img.shields.io/pypi/dm/hitmos?color=%2300d7af&label=downloads" alt="Monthly downloads">
+</a>
+<a href="https://pepy.tech/projects/hitmos" target="_blank">
+    <img src="https://img.shields.io/pepy/dt/hitmos?color=%2300d7af&label=total%20downloads" alt="Total downloads">
+</a>
+<a href="https://github.com/ndugram/hitmos" target="_blank">
+    <img src="https://img.shields.io/github/stars/ndugram/hitmos?style=social" alt="GitHub Stars">
+</a>
+</p>
+---
+**Source Code**: <a href="https://github.com/ndugram/hitmos" target="_blank">https://github.com/ndugram/hitmos</a>
+---
+Hitmos is a modern **AI terminal assistant** for developers, powered by <a href="https://openrouter.ai" target="_blank">OpenRouter</a>. It brings a Claude Code–styled UX to your terminal — live markdown streaming, interactive model switching, project context injection, and Tab completion, all with zero latency overhead.
+Key features:
+- **Streaming** — responses stream token-by-token with live Markdown rendering via <a href="https://github.com/Textualize/rich" target="_blank">rich</a>.
+- **Multi-model** — switch between DeepSeek, Gemini, Claude, GPT-4o, Llama, and more with an interactive arrow-key picker.
+- **Project context** — automatically injects your project files into the system prompt so the model can answer project-specific questions.
+- **Tab completion** — `/model <Tab>` lists all available models; `/` completes all commands.
+- **Minimal UX** — Claude Code–inspired welcome panel, `◆` status lines, clean `>` prompt.
+- **Fast** — built on <a href="https://github.com/ndugram/fasthttp" target="_blank">fasthttp-client</a> with full async SSE streaming.
+- **Configurable** — API key via env (`OPEN_TOKEN`, `OPENROUTER_API_KEY`) or `~/.hitmos/config.toml`.
+## Requirements
+Python 3.13+
+Hitmos depends on:
+- <a href="https://github.com/ndugram/fasthttp" target="_blank"><code>fasthttp-client</code></a> — async HTTP transport with SSE streaming.
+- <a href="https://github.com/Textualize/rich" target="_blank"><code>rich</code></a> — live Markdown rendering and styled console output.
+- <a href="https://typer.tiangolo.com/" target="_blank"><code>typer</code></a> — CLI interface.
+- <a href="https://github.com/ijl/orjson" target="_blank"><code>orjson</code></a> — fast JSON parsing.
+- <a href="https://github.com/prompt-toolkit/python-prompt-toolkit" target="_blank"><code>prompt-toolkit</code></a> — async input with Tab completion (bundled with questionary).
+- <a href="https://github.com/tmbo/questionary" target="_blank"><code>questionary</code></a> — interactive model picker.
+- <a href="https://docs.pydantic.dev/" target="_blank"><code>pydantic-settings</code></a> — config management.
+## Installation
+```console
+$ pip install hitmos
+---> 100%
+```
+## Quickstart
+### Save your API key
+```console
+$ hitmos login
+```
+Or set an environment variable:
+```console
+$ export OPEN_TOKEN=sk-or-...
+```
+### Start chatting
+```console
+$ hitmos
+```
+You will see:
+```
+╭─────────────────────────────────────────────────────╮
+│                                                     │
+│  ✻  Welcome to Hitmos                               │
+│                                                     │
+│  /help for commands                                 │
+│                                                     │
+╰─────────────────────────────────────────────────────╯
+ ◆ ~/my-project
+ ◆ deepseek/deepseek-chat
+ ◆ context 12 KB
+>
+```
+Ask anything about your project:
+```
+> what does this project do?
+> add error handling to client.py
+> explain the streaming logic
+```
+## Commands
+| Command | Description |
+|---|---|
+| `/help` | Show available commands |
+| `/model` | Interactive model picker (↑↓ + Enter) |
+| `/model <name>` | Switch model directly |
+| `/clear` | Clear conversation history |
+| `/reset` | Reset context |
+| `/exit` | Exit Hitmos |
+### Switch model interactively
+Type `/model` and press Enter to open the arrow-key picker:
+```
+? Select model  (↑↓ navigate, Enter confirm, Ctrl+C cancel)
+❯ deepseek/deepseek-chat                   DeepSeek Chat · fast, cheap, great for code
+  deepseek/deepseek-r1                     DeepSeek R1 · reasoning model
+  google/gemini-2.5-flash                  Gemini 2.5 Flash · fast multimodal
+  google/gemini-2.5-pro                    Gemini 2.5 Pro · powerful multimodal
+  anthropic/claude-3.5-haiku               Claude 3.5 Haiku · fast Anthropic
+  anthropic/claude-sonnet-4-5              Claude Sonnet 4.5 · balanced Anthropic
+  openai/gpt-4o-mini                       GPT-4o Mini · OpenAI fast
+  openai/o4-mini                           o4-mini · OpenAI reasoning
+  meta-llama/llama-3.3-70b-instruct        Llama 3.3 70B · open source
+  mistralai/mistral-small-3.2-24b-instruct Mistral Small 3.2 · lightweight
+```
+Or use Tab completion: type `/model deep` then press `Tab`.
+## Configuration
+Hitmos resolves the API key in this order:
+1. `OPEN_TOKEN` environment variable
+2. `OPENROUTER_API_KEY` environment variable
+3. `~/.hitmos/config.toml`
+Config file format:
+```toml
+token = "sk-or-..."
+model = "deepseek/deepseek-chat"
+```
+## Contributing
+Contributions are welcome! Please open an issue or pull request on <a href="https://github.com/ndugram/hitmos" target="_blank">GitHub</a>.
+## License
+This project is licensed under the terms of the <a href="https://github.com/ndugram/hitmos/blob/master/LICENSE" target="_blank">MIT license</a>.

hitmos-0.0.1/README.md ADDED Viewed

@@ -0,0 +1,163 @@
+<p align="center">
+  <img src="https://raw.githubusercontent.com/ndugram/hitmos/master/docs/logo.png" style="background:white; padding:12px; border-radius:10px; width:300">
+</p>
+<p align="center">
+    <em>AI terminal assistant powered by OpenRouter — minimal, fast, developer UX.</em>
+</p>
+<p align="center">
+<a href="https://pypi.org/project/hitmos" target="_blank">
+    <img src="https://img.shields.io/pypi/v/hitmos?color=%2300d7af&label=pypi%20package" alt="Package version">
+</a>
+<a href="https://pypi.org/project/hitmos" target="_blank">
+    <img src="https://img.shields.io/pypi/pyversions/hitmos.svg?color=%2300d7af" alt="Supported Python versions">
+</a>
+<a href="https://pypi.org/project/hitmos" target="_blank">
+    <img src="https://img.shields.io/pypi/dm/hitmos?color=%2300d7af&label=downloads" alt="Monthly downloads">
+</a>
+<a href="https://pepy.tech/projects/hitmos" target="_blank">
+    <img src="https://img.shields.io/pepy/dt/hitmos?color=%2300d7af&label=total%20downloads" alt="Total downloads">
+</a>
+<a href="https://github.com/ndugram/hitmos" target="_blank">
+    <img src="https://img.shields.io/github/stars/ndugram/hitmos?style=social" alt="GitHub Stars">
+</a>
+</p>
+---
+**Source Code**: <a href="https://github.com/ndugram/hitmos" target="_blank">https://github.com/ndugram/hitmos</a>
+---
+Hitmos is a modern **AI terminal assistant** for developers, powered by <a href="https://openrouter.ai" target="_blank">OpenRouter</a>. It brings a Claude Code–styled UX to your terminal — live markdown streaming, interactive model switching, project context injection, and Tab completion, all with zero latency overhead.
+Key features:
+- **Streaming** — responses stream token-by-token with live Markdown rendering via <a href="https://github.com/Textualize/rich" target="_blank">rich</a>.
+- **Multi-model** — switch between DeepSeek, Gemini, Claude, GPT-4o, Llama, and more with an interactive arrow-key picker.
+- **Project context** — automatically injects your project files into the system prompt so the model can answer project-specific questions.
+- **Tab completion** — `/model <Tab>` lists all available models; `/` completes all commands.
+- **Minimal UX** — Claude Code–inspired welcome panel, `◆` status lines, clean `>` prompt.
+- **Fast** — built on <a href="https://github.com/ndugram/fasthttp" target="_blank">fasthttp-client</a> with full async SSE streaming.
+- **Configurable** — API key via env (`OPEN_TOKEN`, `OPENROUTER_API_KEY`) or `~/.hitmos/config.toml`.
+## Requirements
+Python 3.13+
+Hitmos depends on:
+- <a href="https://github.com/ndugram/fasthttp" target="_blank"><code>fasthttp-client</code></a> — async HTTP transport with SSE streaming.
+- <a href="https://github.com/Textualize/rich" target="_blank"><code>rich</code></a> — live Markdown rendering and styled console output.
+- <a href="https://typer.tiangolo.com/" target="_blank"><code>typer</code></a> — CLI interface.
+- <a href="https://github.com/ijl/orjson" target="_blank"><code>orjson</code></a> — fast JSON parsing.
+- <a href="https://github.com/prompt-toolkit/python-prompt-toolkit" target="_blank"><code>prompt-toolkit</code></a> — async input with Tab completion (bundled with questionary).
+- <a href="https://github.com/tmbo/questionary" target="_blank"><code>questionary</code></a> — interactive model picker.
+- <a href="https://docs.pydantic.dev/" target="_blank"><code>pydantic-settings</code></a> — config management.
+## Installation
+```console
+$ pip install hitmos
+---> 100%
+```
+## Quickstart
+### Save your API key
+```console
+$ hitmos login
+```
+Or set an environment variable:
+```console
+$ export OPEN_TOKEN=sk-or-...
+```
+### Start chatting
+```console
+$ hitmos
+```
+You will see:
+```
+╭─────────────────────────────────────────────────────╮
+│                                                     │
+│  ✻  Welcome to Hitmos                               │
+│                                                     │
+│  /help for commands                                 │
+│                                                     │
+╰─────────────────────────────────────────────────────╯
+ ◆ ~/my-project
+ ◆ deepseek/deepseek-chat
+ ◆ context 12 KB
+>
+```
+Ask anything about your project:
+```
+> what does this project do?
+> add error handling to client.py
+> explain the streaming logic
+```
+## Commands
+| Command | Description |
+|---|---|
+| `/help` | Show available commands |
+| `/model` | Interactive model picker (↑↓ + Enter) |
+| `/model <name>` | Switch model directly |
+| `/clear` | Clear conversation history |
+| `/reset` | Reset context |
+| `/exit` | Exit Hitmos |
+### Switch model interactively
+Type `/model` and press Enter to open the arrow-key picker:
+```
+? Select model  (↑↓ navigate, Enter confirm, Ctrl+C cancel)
+❯ deepseek/deepseek-chat                   DeepSeek Chat · fast, cheap, great for code
+  deepseek/deepseek-r1                     DeepSeek R1 · reasoning model
+  google/gemini-2.5-flash                  Gemini 2.5 Flash · fast multimodal
+  google/gemini-2.5-pro                    Gemini 2.5 Pro · powerful multimodal
+  anthropic/claude-3.5-haiku               Claude 3.5 Haiku · fast Anthropic
+  anthropic/claude-sonnet-4-5              Claude Sonnet 4.5 · balanced Anthropic
+  openai/gpt-4o-mini                       GPT-4o Mini · OpenAI fast
+  openai/o4-mini                           o4-mini · OpenAI reasoning
+  meta-llama/llama-3.3-70b-instruct        Llama 3.3 70B · open source
+  mistralai/mistral-small-3.2-24b-instruct Mistral Small 3.2 · lightweight
+```
+Or use Tab completion: type `/model deep` then press `Tab`.
+## Configuration
+Hitmos resolves the API key in this order:
+1. `OPEN_TOKEN` environment variable
+2. `OPENROUTER_API_KEY` environment variable
+3. `~/.hitmos/config.toml`
+Config file format:
+```toml
+token = "sk-or-..."
+model = "deepseek/deepseek-chat"
+```
+## Contributing
+Contributions are welcome! Please open an issue or pull request on <a href="https://github.com/ndugram/hitmos" target="_blank">GitHub</a>.
+## License
+This project is licensed under the terms of the <a href="https://github.com/ndugram/hitmos/blob/master/LICENSE" target="_blank">MIT license</a>.

hitmos-0.0.1/hitmos/__init__.py ADDED Viewed

	@@ -0,0 +1 @@
1	+ __version__ = "0.1.0"

hitmos-0.0.1/hitmos/app.py ADDED Viewed

@@ -0,0 +1,163 @@
+import asyncio
+from collections.abc import AsyncGenerator
+from pathlib import Path
+from .chat import ChatSession
+from .client import OpenRouterClient, ToolCallRequest
+from .commands import CommandHandler, CommandResult, CommandType
+from .config import ConfigManager
+from .constants import SYSTEM_PROMPT
+from .context import ProjectContext
+from .exceptions import AuthError, HitmosError
+from .methods import dispatch
+from .ui import ConsoleUI
+class HitmosApp:
+    def __init__(self) -> None:
+        self._config = ConfigManager()
+        self._ui = ConsoleUI()
+        self._commands = CommandHandler()
+        self._session: ChatSession | None = None
+        self._client: OpenRouterClient | None = None
+    def run(self) -> None:
+        try:
+            token = self._config.resolve_token()
+        except AuthError as e:
+            self._ui.show_error(str(e))
+            raise SystemExit(1)
+        model = self._config.get_model()
+        self._client = OpenRouterClient(token, model)
+        system_prompt, ctx_kb = self._build_system_prompt()
+        self._session = ChatSession(system_prompt=system_prompt)
+        self._ui.show_welcome(model, ctx_kb)
+        try:
+            asyncio.run(self._loop())
+        except KeyboardInterrupt:
+            self._ui.show_exit()
+    async def _loop(self) -> None:
+        assert self._client is not None
+        assert self._session is not None
+        while True:
+            try:
+                text = await self._ui.get_input()
+            except (KeyboardInterrupt, EOFError):
+                self._ui.show_exit()
+                return
+            text = text.strip()
+            if not text:
+                continue
+            result = self._commands.parse(text)
+            if result is not None:
+                should_exit = await self._handle_command(result)
+                if should_exit:
+                    return
+            else:
+                await self._handle_message(text)
+    def _build_system_prompt(self) -> tuple[str, int]:
+        ctx, kb = ProjectContext(Path.cwd()).build()
+        if not ctx:
+            return SYSTEM_PROMPT, 0
+        return (
+            "You are Hitmos, a helpful AI coding assistant. "
+            "Be concise, accurate, and developer-focused.\n\n"
+            "You have access to the user's project files below. "
+            "Use this context to give accurate, project-specific answers.\n\n"
+            + ctx,
+            kb,
+        )
+    async def _handle_command(self, result: CommandResult) -> bool:
+        assert self._client is not None
+        assert self._session is not None
+        match result.type:
+            case CommandType.HELP:
+                self._ui.show_help()
+            case CommandType.CLEAR:
+                self._session.clear()
+                self._ui.show_info("History cleared.")
+            case CommandType.RESET:
+                self._session.reset()
+                self._ui.show_info("Context reset.")
+            case CommandType.MODEL:
+                if result.arg:
+                    self._client.model = result.arg
+                    self._config.save_model(result.arg)
+                    self._ui.show_success(f"Model: {result.arg}")
+                else:
+                    selected = await self._ui.show_model_picker(self._client.model)
+                    if selected:
+                        self._client.model = selected
+                        self._config.save_model(selected)
+                        self._ui.show_success(f"Model: {selected}")
+            case CommandType.EXIT:
+                self._ui.show_exit()
+                return True
+        return False
+    async def _handle_message(self, text: str) -> None:
+        assert self._client is not None
+        assert self._session is not None
+        self._session.add_user(text)
+        try:
+            while True:
+                tool_calls: list[ToolCallRequest] = []
+                async def _text_gen() -> AsyncGenerator[str, None]:
+                    async for item in self._client.stream_chat(  # type: ignore[union-attr]
+                        self._session.messages  # type: ignore[union-attr]
+                    ):
+                        if isinstance(item, list):
+                            tool_calls.extend(item)
+                        else:
+                            yield item
+                full_response = await self._ui.stream_response(_text_gen())
+                if not tool_calls:
+                    if full_response:
+                        self._session.add_assistant(full_response)
+                    break
+                self._session.add_raw({
+                    "role": "assistant",
+                    "content": full_response or None,
+                    "tool_calls": [
+                        {
+                            "id": tc.id,
+                            "type": "function",
+                            "function": {"name": tc.name, "arguments": tc.arguments},
+                        }
+                        for tc in tool_calls
+                    ],
+                })
+                for tc in tool_calls:
+                    self._ui.show_tool_call(tc.name, tc.arguments)
+                    result = dispatch(tc.name, tc.arguments)
+                    self._ui.show_tool_result(result)
+                    self._session.add_raw({
+                        "role": "tool",
+                        "tool_call_id": tc.id,
+                        "content": result,
+                    })
+        except HitmosError as e:
+            self._session.pop_last()
+            self._ui.show_error(str(e))
+        except Exception as e:
+            self._session.pop_last()
+            self._ui.show_error(f"Unexpected error: {e}")

hitmos-0.0.1/hitmos/chat.py ADDED Viewed

@@ -0,0 +1,34 @@
+from .constants import SYSTEM_PROMPT
+class ChatSession:
+    def __init__(self, system_prompt: str | None = None) -> None:
+        self._messages: list[dict] = []
+        self._system = system_prompt or SYSTEM_PROMPT
+    def add_user(self, content: str) -> None:
+        self._messages.append({"role": "user", "content": content})
+    def add_assistant(self, content: str) -> None:
+        self._messages.append({"role": "assistant", "content": content})
+    def add_raw(self, message: dict) -> None:
+        self._messages.append(message)
+    def clear(self) -> None:
+        self._messages.clear()
+    def reset(self) -> None:
+        self._messages.clear()
+    def pop_last(self) -> None:
+        if self._messages:
+            self._messages.pop()
+    @property
+    def messages(self) -> list[dict]:
+        return [{"role": "system", "content": self._system}] + self._messages
+    @property
+    def is_empty(self) -> bool:
+        return len(self._messages) == 0