Compare commits
79 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
1b5c2fbdc1 | ||
|
|
a865a26093 | ||
|
|
1563e83672 | ||
|
|
cf114b37b3 | ||
|
|
b8ef2b978b | ||
|
|
7e37aee1b9 | ||
|
|
53836e556a | ||
|
|
89052bbfd9 | ||
|
|
07eb12d406 | ||
|
|
3272cd6fb2 | ||
|
|
7ab97a39ca | ||
|
|
11c5a9942e | ||
|
|
251dec454c | ||
|
|
1031dbf672 | ||
|
|
581f00b9fb | ||
|
|
e57778d2bd | ||
|
|
2d659964a7 | ||
|
|
661e7263fb | ||
|
|
1705c16762 | ||
|
|
2455117e41 | ||
|
|
3e9c110afc | ||
|
|
d9b5524c97 | ||
|
|
9b169a7d28 | ||
|
|
cb29419d75 | ||
|
|
51ca8c9948 | ||
|
|
12b938d2d1 | ||
|
|
f3d1b32ad7 | ||
|
|
3d3bd3c9f9 | ||
|
|
a25845e89e | ||
|
|
479f54b93d | ||
|
|
90ab6807ac | ||
|
|
5022f79bf5 | ||
|
|
2ba3b5a437 | ||
|
|
2819011b9d | ||
|
|
ede9065f77 | ||
|
|
5e61f5b3a3 | ||
|
|
8633e82f62 | ||
|
|
6aa3d70d57 | ||
|
|
75e29ee3fb | ||
|
|
eab7978b9f | ||
|
|
765e7c17b2 | ||
|
|
748d2b7fd4 | ||
|
|
527759dd02 | ||
|
|
3290255bad | ||
|
|
9b987395b3 | ||
|
|
4852d1fe92 | ||
|
|
8bd313f0d4 | ||
|
|
711e1099d6 | ||
|
|
91c8dd8f7d | ||
|
|
afc5572d6a | ||
|
|
73e13dcf2f | ||
|
|
2bebd109b1 | ||
|
|
7f7c58be1f | ||
|
|
47e5a464a0 | ||
|
|
85f751e514 | ||
|
|
95acad3558 | ||
|
|
304056c59a | ||
|
|
bedc0335f1 | ||
|
|
a8be8c3c1e | ||
|
|
826fa728a9 | ||
|
|
5aef8e8b2f | ||
|
|
f70f20181e | ||
|
|
fcdd4f39ee | ||
|
|
891adfec49 | ||
|
|
35ab79844b | ||
|
|
bbc122fbb1 | ||
|
|
2413c8b88b | ||
|
|
10c3c64469 | ||
|
|
b3307b404b | ||
|
|
7a0d1e830d | ||
|
|
b3bd241823 | ||
|
|
de3d8b054f | ||
|
|
0a55e276a2 | ||
|
|
1f2c65d2e6 | ||
|
|
3b5e4745f2 | ||
|
|
407bb022d9 | ||
|
|
faf92cac09 | ||
|
|
a82e9a1d1f | ||
|
|
8f87f05a08 |
16
.github/workflows/workflow.yml
vendored
16
.github/workflows/workflow.yml
vendored
@@ -16,8 +16,8 @@ jobs:
|
|||||||
name: Unit testing and linting
|
name: Unit testing and linting
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@v6
|
||||||
- uses: dtolnay/rust-toolchain@stable
|
- uses: dtolnay/rust-toolchain@1.93.0
|
||||||
- name: Install SQLite3
|
- name: Install SQLite3
|
||||||
run: sudo apt-get update && sudo apt-get install -y libsqlite3-dev
|
run: sudo apt-get update && sudo apt-get install -y libsqlite3-dev
|
||||||
- run: cargo test --all-features
|
- run: cargo test --all-features
|
||||||
@@ -30,7 +30,7 @@ jobs:
|
|||||||
steps:
|
steps:
|
||||||
- name: Extract metadata (tags, labels) for Docker
|
- name: Extract metadata (tags, labels) for Docker
|
||||||
id: meta
|
id: meta
|
||||||
uses: docker/metadata-action@v5
|
uses: docker/metadata-action@v6
|
||||||
with:
|
with:
|
||||||
images: |
|
images: |
|
||||||
ghcr.io/${{ github.repository }}
|
ghcr.io/${{ github.repository }}
|
||||||
@@ -56,16 +56,16 @@ jobs:
|
|||||||
|
|
||||||
steps:
|
steps:
|
||||||
- name: Checkout
|
- name: Checkout
|
||||||
uses: actions/checkout@v4
|
uses: actions/checkout@v6
|
||||||
- name: Log in to the GitHub Container registry
|
- name: Log in to the GitHub Container registry
|
||||||
uses: docker/login-action@v3
|
uses: docker/login-action@v4
|
||||||
with:
|
with:
|
||||||
registry: ghcr.io
|
registry: ghcr.io
|
||||||
username: ${{ github.actor }}
|
username: ${{ github.actor }}
|
||||||
password: ${{ secrets.GITHUB_TOKEN }}
|
password: ${{ secrets.GITHUB_TOKEN }}
|
||||||
- name: Extract metadata (tags, labels) for Docker
|
- name: Extract metadata (tags, labels) for Docker
|
||||||
id: meta
|
id: meta
|
||||||
uses: docker/metadata-action@v5
|
uses: docker/metadata-action@v6
|
||||||
with:
|
with:
|
||||||
tags: |
|
tags: |
|
||||||
type=raw,value=latest,enable=${{ github.ref_name == 'main' }}
|
type=raw,value=latest,enable=${{ github.ref_name == 'main' }}
|
||||||
@@ -77,7 +77,7 @@ jobs:
|
|||||||
ghcr.io/${{ github.repository }}
|
ghcr.io/${{ github.repository }}
|
||||||
|
|
||||||
- name: Build and push Docker images
|
- name: Build and push Docker images
|
||||||
uses: docker/build-push-action@v6
|
uses: docker/build-push-action@v7
|
||||||
with:
|
with:
|
||||||
push: true
|
push: true
|
||||||
tags: ${{ steps.meta.outputs.tags }}
|
tags: ${{ steps.meta.outputs.tags }}
|
||||||
@@ -95,7 +95,7 @@ jobs:
|
|||||||
|
|
||||||
steps:
|
steps:
|
||||||
- name: Log in to the GitHub Container registry
|
- name: Log in to the GitHub Container registry
|
||||||
uses: docker/login-action@v3
|
uses: docker/login-action@v4
|
||||||
with:
|
with:
|
||||||
registry: ghcr.io
|
registry: ghcr.io
|
||||||
username: ${{ github.actor }}
|
username: ${{ github.actor }}
|
||||||
|
|||||||
36
.pre-commit-config.yaml
Normal file
36
.pre-commit-config.yaml
Normal file
@@ -0,0 +1,36 @@
|
|||||||
|
repos:
|
||||||
|
# Fast built-in hooks (Rust-native, no dependencies)
|
||||||
|
- repo: builtin
|
||||||
|
hooks:
|
||||||
|
- id: trailing-whitespace
|
||||||
|
- id: end-of-file-fixer
|
||||||
|
- id: check-yaml
|
||||||
|
- id: check-merge-conflict
|
||||||
|
- id: check-added-large-files
|
||||||
|
args: ['--maxkb=1024']
|
||||||
|
|
||||||
|
# Local hooks that run project-specific tools
|
||||||
|
- repo: local
|
||||||
|
hooks:
|
||||||
|
- id: cargo-fmt-check
|
||||||
|
name: Cargo Format Check
|
||||||
|
entry: cargo fmt --all -- --check
|
||||||
|
language: system
|
||||||
|
files: '\.rs$'
|
||||||
|
pass_filenames: false
|
||||||
|
|
||||||
|
- id: cargo-clippy
|
||||||
|
name: Cargo Clippy
|
||||||
|
entry: cargo clippy -- -D warnings
|
||||||
|
language: system
|
||||||
|
files: '\.rs$'
|
||||||
|
pass_filenames: false
|
||||||
|
priority: 100
|
||||||
|
|
||||||
|
- id: test-unit
|
||||||
|
name: Unit Tests
|
||||||
|
entry: just test
|
||||||
|
language: system
|
||||||
|
files: '\.rs$'
|
||||||
|
pass_filenames: false
|
||||||
|
priority: 100
|
||||||
65
CHANGELOG.md
65
CHANGELOG.md
@@ -1,3 +1,68 @@
|
|||||||
|
# (2026-04-11) Version 1.18.0
|
||||||
|
|
||||||
|
- (**Bugfix**) Fix the bot not sending a welcome message when joining a room on homeservers (like [Continuwuity](https://continuwuity.org/)) that place the join membership event in the sync response's `state` block rather than the `timeline` block, via [mxlink](https://crates.io/crates/mxlink) 1.13.1
|
||||||
|
|
||||||
|
- (**Improvement**) Update [tiktoken-rs](https://crates.io/crates/tiktoken-rs) to 0.11, adding tokenization support for newer GPT models (gpt-5.x, codex, etc.) and fixing context sizes for o1-mini/chatgpt-4o/gpt-4.5
|
||||||
|
|
||||||
|
- (**Internal Improvement**) Dependency updates
|
||||||
|
|
||||||
|
|
||||||
|
# (2026-03-25) Version 1.17.0
|
||||||
|
|
||||||
|
- (**Feature**) Add `text-generation sender-context-mode` for attaching sender metadata to conversation messages. See the [💬 Text Generation](./docs/configuration/text-generation.md#-sender-context-mode) documentation for details. Thanks to [kschwank](https://github.com/kschwank) for the contribution in [#104](https://github.com/etkecc/baibot/pull/104)!
|
||||||
|
|
||||||
|
|
||||||
|
# (2026-03-24) Version 1.16.1
|
||||||
|
|
||||||
|
- (**Bugfix**) Fix compatibility with [async-openai](https://crates.io/crates/async-openai) 0.34.0 by populating the new `phase` field required for OpenAI Responses API message inputs. baibot does not currently distinguish between assistant `commentary` and `final_answer` turns, so using `None` preserves the previous behavior while remaining compatible with the updated crate.
|
||||||
|
|
||||||
|
- (**Internal Improvement**) Dependency updates.
|
||||||
|
|
||||||
|
|
||||||
|
# (2026-03-20) Version 1.16.0
|
||||||
|
|
||||||
|
- (**Feature**) Add support for file attachments (`m.file` Matrix messages) in conversations. Files like PDFs, text documents, spreadsheets, code files, etc. are now downloaded and forwarded to the LLM alongside the conversation context, similar to how images (`m.image`) are already handled. See the [💬 Text Generation](./docs/features.md#-text-generation) documentation for details and known limitations.
|
||||||
|
|
||||||
|
- (**Improvement**) Use the [mime_guess](https://crates.io/crates/mime_guess) crate for MIME type detection from file extensions, replacing a hand-maintained mapping. This covers hundreds of file extensions out of the box.
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
# (2026-03-07) Version 1.15.0
|
||||||
|
|
||||||
|
- (**Feature**) Add support for authentication via access tokens (for [Matrix Authentication Service](https://github.com/element-hq/matrix-authentication-service)/OIDC-enabled homeservers) as an alternative to password authentication. See [🔐 Authentication](./docs/configuration/authentication.md) for setup details. Thanks to [Taylor Southwick](https://github.com/twsouthwick) for the contribution in [#83](https://github.com/etkecc/baibot/pull/83)!
|
||||||
|
|
||||||
|
- (**Internal Improvement**) Pin the Rust toolchain to `1.93.0` in both CI and local development to avoid `matrix-sdk` build failures on newer stable toolchains.
|
||||||
|
|
||||||
|
- (**Internal Improvement**) Documentation updates.
|
||||||
|
|
||||||
|
- (**Internal Improvement**) Dependency updates.
|
||||||
|
|
||||||
|
|
||||||
|
# (2026-02-18) Version 1.14.3
|
||||||
|
|
||||||
|
- (**Internal Improvement**) Add [Renovate](https://docs.renovatebot.com/) configuration for automated dependency updates
|
||||||
|
|
||||||
|
- (**Internal Improvement**) Dependency updates
|
||||||
|
|
||||||
|
|
||||||
|
# (2026-02-18) Version 1.14.2
|
||||||
|
|
||||||
|
- (**Internal Improvement**) Dependency updates
|
||||||
|
|
||||||
|
- (**Internal Improvement**) Reorganize the development environment to support [Continuwuity](https://continuwuity.org/) as a homeserver choice (in addition to [Synapse](https://github.com/element-hq/synapse)). Continuwuity is now the default for its lighter footprint (no external database required). See [development docs](./docs/development.md) for details.
|
||||||
|
|
||||||
|
|
||||||
|
# (2026-02-10) Version 1.14.1
|
||||||
|
|
||||||
|
- (**Security**) Dependency updates to fix security vulnerabilities ([time](https://crates.io/crates/time) stack exhaustion DoS, [bytes](https://crates.io/crates/bytes) integer overflow), via [mxlink](https://crates.io/crates/mxlink) 1.12.0
|
||||||
|
|
||||||
|
- (**Internal Improvement**) Switch from deprecated [serde_yaml](https://crates.io/crates/serde_yaml) to its maintained fork [serde_yaml_ng](https://crates.io/crates/serde_yaml_ng)
|
||||||
|
|
||||||
|
- (**Internal Improvement**) Add [prek](https://github.com/nicholasgasior/prek) pre-commit hooks via [mise](https://mise.jdx.dev/) for automated code quality checks (formatting, clippy, tests)
|
||||||
|
|
||||||
|
- (**Internal Improvement**) Fix clippy warnings and formatting issues
|
||||||
|
|
||||||
|
|
||||||
# (2026-02-04) Version 1.14.0
|
# (2026-02-04) Version 1.14.0
|
||||||
|
|
||||||
- (**Feature**) The `openai` provider now uses OpenAI's [Responses API](https://platform.openai.com/docs/api-reference/responses) (instead of the older Chat Completions API), adding support for [🛠️ built-in tools](./docs/features.md#️-built-in-tools-openai-only) (`web_search` and `code_interpreter`). These tools are **disabled by default** and can be enabled via the `text_generation.tools` configuration (see the [sample configuration](https://github.com/etkecc/baibot/blob/c70387b0c38d8d0f30bba2179a2a21a3710dbeaf/docs/sample-provider-configs/openai.yml#L12-L15)). To enable tools on an existing agent, you need to [update the agent](./docs/agents.md#updating-agents) to re-create it with the `text_generation.tools` section added and enable the tools you need. Thanks to [Layla Manley](https://github.com/yeslayla) for the contribution in [#62](https://github.com/etkecc/baibot/pull/62)!
|
- (**Feature**) The `openai` provider now uses OpenAI's [Responses API](https://platform.openai.com/docs/api-reference/responses) (instead of the older Chat Completions API), adding support for [🛠️ built-in tools](./docs/features.md#️-built-in-tools-openai-only) (`web_search` and `code_interpreter`). These tools are **disabled by default** and can be enabled via the `text_generation.tools` configuration (see the [sample configuration](https://github.com/etkecc/baibot/blob/c70387b0c38d8d0f30bba2179a2a21a3710dbeaf/docs/sample-provider-configs/openai.yml#L12-L15)). To enable tools on an existing agent, you need to [update the agent](./docs/agents.md#updating-agents) to re-create it with the `text_generation.tools` section added and enable the tools you need. Thanks to [Layla Manley](https://github.com/yeslayla) for the contribution in [#62](https://github.com/etkecc/baibot/pull/62)!
|
||||||
|
|||||||
856
Cargo.lock
generated
856
Cargo.lock
generated
File diff suppressed because it is too large
Load Diff
14
Cargo.toml
14
Cargo.toml
@@ -7,7 +7,7 @@ license = "AGPL-3.0-or-later"
|
|||||||
readme = "README.md"
|
readme = "README.md"
|
||||||
keywords = ["matrix", "chat", "bot", "AI", "LLM"]
|
keywords = ["matrix", "chat", "bot", "AI", "LLM"]
|
||||||
include = ["/etc/assets/baibot-torso-768.png", "/src", "/README.md", "/CHANGELOG.md", "/LICENSE"]
|
include = ["/etc/assets/baibot-torso-768.png", "/src", "/README.md", "/CHANGELOG.md", "/LICENSE"]
|
||||||
version = "1.14.0"
|
version = "1.18.0"
|
||||||
edition = "2024"
|
edition = "2024"
|
||||||
|
|
||||||
[lib]
|
[lib]
|
||||||
@@ -17,7 +17,7 @@ path = "src/lib.rs"
|
|||||||
[dependencies]
|
[dependencies]
|
||||||
anthropic = { git = "https://github.com/etkecc/anthropic-rs.git", branch = "fix-content-block-image" }
|
anthropic = { git = "https://github.com/etkecc/anthropic-rs.git", branch = "fix-content-block-image" }
|
||||||
anyhow = "1.0.*"
|
anyhow = "1.0.*"
|
||||||
async-openai = { version = "0.32.3", features = ["audio", "chat-completion", "image", "responses"] }
|
async-openai = { version = "0.34.0", features = ["audio", "chat-completion", "image", "responses"] }
|
||||||
base64 = "0.22.*"
|
base64 = "0.22.*"
|
||||||
chrono = { version = "0.4.*", default-features = false, features = ["std", "now"] }
|
chrono = { version = "0.4.*", default-features = false, features = ["std", "now"] }
|
||||||
# We'd rather not depend on this, but we cannot use the ruma-events EventContent macro without it.
|
# We'd rather not depend on this, but we cannot use the ruma-events EventContent macro without it.
|
||||||
@@ -25,16 +25,16 @@ chrono = { version = "0.4.*", default-features = false, features = ["std", "now"
|
|||||||
matrix-sdk = { version = "0.16.0", default-features = false, features = ["native-tls"] }
|
matrix-sdk = { version = "0.16.0", default-features = false, features = ["native-tls"] }
|
||||||
mime_guess = "2.0.*"
|
mime_guess = "2.0.*"
|
||||||
mxidwc = "1.0.*"
|
mxidwc = "1.0.*"
|
||||||
mxlink = ">=1.11.0"
|
mxlink = ">=1.13.0"
|
||||||
etke_openai_api_rust = "0.1.*"
|
etke_openai_api_rust = "0.1.*"
|
||||||
quick_cache = "0.6.*"
|
quick_cache = "0.6.*"
|
||||||
regex = "1.12.*"
|
regex = "1.12.*"
|
||||||
serde = { version = "1.0.*", features = ["derive"], default-features = false }
|
serde = { version = "1.0.*", features = ["derive"], default-features = false }
|
||||||
serde_json = "1.0.*"
|
serde_json = "1.0.*"
|
||||||
serde_yaml = "0.9.*"
|
serde_yaml_ng = "0.10.*"
|
||||||
tempfile = "3.24.*"
|
tempfile = "3.27.*"
|
||||||
tiktoken-rs = { version = "0.9.*", default-features = false }
|
tiktoken-rs = { version = "0.11.*", default-features = false }
|
||||||
tokio = { version = "1.49.*", features = ["rt", "rt-multi-thread", "macros"] }
|
tokio = { version = "1.51.*", features = ["rt", "rt-multi-thread", "macros"] }
|
||||||
tracing = "0.1.*"
|
tracing = "0.1.*"
|
||||||
tracing-subscriber = { version = "0.3.*", features = ["env-filter"] }
|
tracing-subscriber = { version = "0.3.*", features = ["env-filter"] }
|
||||||
url = "2.5.*"
|
url = "2.5.*"
|
||||||
|
|||||||
@@ -4,7 +4,7 @@
|
|||||||
# #
|
# #
|
||||||
#######################################
|
#######################################
|
||||||
|
|
||||||
FROM docker.io/rust:1.93.0-slim-trixie AS build
|
FROM docker.io/rust:1.93.1-slim-trixie AS build
|
||||||
|
|
||||||
RUN apt-get update && apt-get install -y build-essential pkg-config libssl-dev libsqlite3-dev
|
RUN apt-get update && apt-get install -y build-essential pkg-config libssl-dev libsqlite3-dev
|
||||||
|
|
||||||
|
|||||||
@@ -4,7 +4,7 @@
|
|||||||
# #
|
# #
|
||||||
#######################################
|
#######################################
|
||||||
|
|
||||||
FROM docker.io/rust:1.93.0-slim-trixie AS build
|
FROM docker.io/rust:1.93.1-slim-trixie AS build
|
||||||
|
|
||||||
RUN apt-get update && apt-get install -y build-essential pkg-config libssl-dev libsqlite3-dev
|
RUN apt-get update && apt-get install -y build-essential pkg-config libssl-dev libsqlite3-dev
|
||||||
|
|
||||||
|
|||||||
@@ -22,6 +22,8 @@ You can see the list of supported environment variables in the [🦀 src/entity/
|
|||||||
> [!WARNING]
|
> [!WARNING]
|
||||||
> The static configuration contains an `initial_global_config` key, which is used to populate the bot's global configuration (stored as [dynamic configuration](#dynamic-configuration)) the first time the bot starts. Modifying this subsequently will not have any effect. After initial global configuration creation, it's expected to be managed dynamically via chat commands.
|
> The static configuration contains an `initial_global_config` key, which is used to populate the bot's global configuration (stored as [dynamic configuration](#dynamic-configuration)) the first time the bot starts. Modifying this subsequently will not have any effect. After initial global configuration creation, it's expected to be managed dynamically via chat commands.
|
||||||
|
|
||||||
|
For Matrix-account authentication setup, see [🔐 Authentication](./authentication.md).
|
||||||
|
|
||||||
|
|
||||||
### Dynamic configuration
|
### Dynamic configuration
|
||||||
|
|
||||||
|
|||||||
23
docs/configuration/authentication.md
Normal file
23
docs/configuration/authentication.md
Normal file
@@ -0,0 +1,23 @@
|
|||||||
|
## 🔐 Authentication
|
||||||
|
|
||||||
|
baibot supports 2 authentication modes for the Matrix account (`user.*` keys in config).
|
||||||
|
|
||||||
|
Set **exactly one** mode. If both are set (or neither is set), startup validation fails.
|
||||||
|
|
||||||
|
### Password authentication
|
||||||
|
|
||||||
|
- Config key: `user.password`
|
||||||
|
- Environment variable: `BAIBOT_USER_PASSWORD`
|
||||||
|
|
||||||
|
### Access token authentication
|
||||||
|
|
||||||
|
- Config keys: `user.access_token` + `user.device_id`
|
||||||
|
- Environment variables: `BAIBOT_USER_ACCESS_TOKEN` + `BAIBOT_USER_DEVICE_ID`
|
||||||
|
|
||||||
|
Access-token authentication is useful for OIDC-enabled homeservers (e.g. those using [Matrix Authentication Service](https://github.com/element-hq/matrix-authentication-service)).
|
||||||
|
|
||||||
|
Example token-generation command:
|
||||||
|
|
||||||
|
```sh
|
||||||
|
mas-cli manage issue-compatibility-token <username> [device_id]
|
||||||
|
```
|
||||||
@@ -8,7 +8,7 @@ You can also use **different models within the same room** (e.g. [💬 text-gene
|
|||||||
|
|
||||||
The bot supports the following use-purposes:
|
The bot supports the following use-purposes:
|
||||||
|
|
||||||
- [💬 text-generation](../features.md#-text-generation): communicating with you via text (though certain models may "see" images as well)
|
- [💬 text-generation](../features.md#-text-generation): communicating with you via text (though certain models may also process images and files)
|
||||||
- [🦻 speech-to-text](../features.md#-speech-to-text): turning your voice messages into text
|
- [🦻 speech-to-text](../features.md#-speech-to-text): turning your voice messages into text
|
||||||
- [🗣️ text-to-speech](../features.md#️-text-to-speech): turning bot or users text messages into voice messages
|
- [🗣️ text-to-speech](../features.md#️-text-to-speech): turning bot or users text messages into voice messages
|
||||||
- [🖌️ image-generation](../features.md#image-generation): generating images based on instructions
|
- [🖌️ image-generation](../features.md#image-generation): generating images based on instructions
|
||||||
|
|||||||
@@ -57,6 +57,25 @@ This feature relies on [tokenization](https://en.wikipedia.org/wiki/Large_langua
|
|||||||
This setting is **disabled by default**, but can be enabled via `!bai config room text-generation set-context-management-enabled true` (this can also be set globally, see [🛠️ Room Settings](./README.md#room-settings)).
|
This setting is **disabled by default**, but can be enabled via `!bai config room text-generation set-context-management-enabled true` (this can also be set globally, see [🛠️ Room Settings](./README.md#room-settings)).
|
||||||
|
|
||||||
|
|
||||||
|
### 👤 Sender Context Mode
|
||||||
|
|
||||||
|
In multi-user rooms, it may be useful for the model to know which participant sent each message in the conversation context.
|
||||||
|
|
||||||
|
To support this, the bot has a `text-generation sender-context-mode` setting, which can be set to:
|
||||||
|
|
||||||
|
- (default) `disabled`: do not attach sender metadata to messages before sending them to the model
|
||||||
|
|
||||||
|
- `matrix_user_id`: prefix text messages with the sender's Matrix user ID, for example: `[sender=@alice:example.com] Hello bot`
|
||||||
|
|
||||||
|
- `matrix_user_id_and_timestamp`: prefix text messages with the sender's Matrix user ID and the message timestamp, for example: `[sender=@alice:example.com sent_at=2026-03-23T14:30:00Z] Hello bot`
|
||||||
|
|
||||||
|
This sender metadata is attached to conversation messages before they are sent to the model provider. It applies to user and assistant text messages, but not to system prompts or non-text content.
|
||||||
|
|
||||||
|
⚠️ Enabling this sends Matrix user IDs, and optionally timestamps, to the model provider.
|
||||||
|
|
||||||
|
Example: `!bai config room text-generation set-sender-context-mode matrix_user_id` (this can also be set globally, see [🛠️ Room Settings](./README.md#room-settings))
|
||||||
|
|
||||||
|
|
||||||
### ⌨️ Prompt Override
|
### ⌨️ Prompt Override
|
||||||
|
|
||||||
You can override the [system prompt](https://huggingface.co/docs/transformers/en/tasks/prompting) configured at the [🤖 agent](../agents.md) level.
|
You can override the [system prompt](https://huggingface.co/docs/transformers/en/tasks/prompting) configured at the [🤖 agent](../agents.md) level.
|
||||||
|
|||||||
@@ -18,6 +18,27 @@ For local development, we run all dependency services in [🐋 Docker](https://w
|
|||||||
- (Optional) an API key for some Large Language Model [☁️ provider](./providers.md) (e.g. [OpenAI](./providers.md#openai)), though we recommend using [LocalAI](#localai) or [Ollama](#ollama) for local development
|
- (Optional) an API key for some Large Language Model [☁️ provider](./providers.md) (e.g. [OpenAI](./providers.md#openai)), though we recommend using [LocalAI](#localai) or [Ollama](#ollama) for local development
|
||||||
|
|
||||||
|
|
||||||
|
### Choosing a homeserver
|
||||||
|
|
||||||
|
The development environment supports two homeserver implementations:
|
||||||
|
|
||||||
|
- **[Continuwuity](https://continuwuity.org/)** (default) — lightweight, no external database required. Good for most development needs.
|
||||||
|
- **[Synapse](https://github.com/element-hq/synapse)** — the reference implementation, bundled with Postgres. Use this if you need Synapse-specific behavior.
|
||||||
|
|
||||||
|
To choose a homeserver (optional — defaults to Continuwuity if skipped):
|
||||||
|
|
||||||
|
```sh
|
||||||
|
just homeserver-init continuwuity # or: just homeserver-init synapse
|
||||||
|
```
|
||||||
|
|
||||||
|
The choice is stored in `var/homeserver` and affects all subsequent commands.
|
||||||
|
|
||||||
|
> **Note:** If you switch homeservers after initial setup, you will need to:
|
||||||
|
> - Delete `var/app/local/` and/or `var/app/container/` (app config and data)
|
||||||
|
> - Delete `var/services/element-web/` (to regenerate its config)
|
||||||
|
> - Re-run the prepare and user registration steps
|
||||||
|
|
||||||
|
|
||||||
### Getting started guide
|
### Getting started guide
|
||||||
|
|
||||||
Developing [locally](#running-locally) is possible, but requires a [Rust](https://www.rust-lang.org/) toolchain.
|
Developing [locally](#running-locally) is possible, but requires a [Rust](https://www.rust-lang.org/) toolchain.
|
||||||
@@ -28,11 +49,12 @@ In any case, you will need [🐋 Docker](https://www.docker.com/) as [dependency
|
|||||||
|
|
||||||
#### Running locally
|
#### Running locally
|
||||||
|
|
||||||
1. Start the core dependency services (Postgres, Synapse, Element Web): `just services-start`
|
1. (Optional) Choose a homeserver: `just homeserver-init continuwuity` (or `synapse`). Default is `continuwuity`.
|
||||||
2. (Only the first time around) Prepare initial app configuration in `var/app/local/config.yml`: `just app-local-prepare`
|
2. Start the homeserver and Element Web: `just services-start`
|
||||||
3. (Only the first time around) [Prepare your configuration file](#prepare-your-configuration-file)
|
3. (Only the first time around) Prepare initial app configuration in `var/app/local/config.yml`: `just app-local-prepare`
|
||||||
4. (Only the first time around) Prepare initial default Matrix user accounts (`admin` and `baibot`): `just users-prepare`
|
4. (Only the first time around) [Prepare your configuration file](#prepare-your-configuration-file)
|
||||||
5. (Optional) Start additional services depending on which [agent provider you've chosen](#choosing-an-agent-provider):
|
5. (Only the first time around) Prepare initial default Matrix user accounts (`admin` and `baibot`): `just users-prepare`
|
||||||
|
6. (Optional) Start additional services depending on which [agent provider you've chosen](#choosing-an-agent-provider):
|
||||||
- for [LocalAI](#localai):
|
- for [LocalAI](#localai):
|
||||||
- Start services: `just localai-start`
|
- Start services: `just localai-start`
|
||||||
- Wait a while for LocalAI to start up. It has a lot of models to download. Monitor progress using `just localai-tail-logs`
|
- Wait a while for LocalAI to start up. It has a lot of models to download. Monitor progress using `just localai-tail-logs`
|
||||||
@@ -40,12 +62,12 @@ In any case, you will need [🐋 Docker](https://www.docker.com/) as [dependency
|
|||||||
- for [Ollama](#ollama):
|
- for [Ollama](#ollama):
|
||||||
- Start services: `just ollama-start`
|
- Start services: `just ollama-start`
|
||||||
- (Only the first time around) Pull the model configured in `agents.static_definitions` in the configuration file: `just ollama-pull-model gemma2:2b`
|
- (Only the first time around) Pull the model configured in `agents.static_definitions` in the configuration file: `just ollama-pull-model gemma2:2b`
|
||||||
6. Start the bot: `just run-locally`
|
7. Start the bot: `just run-locally`
|
||||||
7. Go to http://element.127.0.0.1.nip.io:42025/ and login with `admin` / `admin`
|
8. Go to http://element.127.0.0.1.nip.io:42025/ and login with `admin` / `admin`
|
||||||
8. Create a new room and invite `@baibot:synapse.127.0.0.1.nip.io`
|
9. Create a new room and invite `@baibot:continuwuity.127.0.0.1.nip.io` (or `@baibot:synapse.127.0.0.1.nip.io` if using Synapse)
|
||||||
9. When done, stop the bot (`Ctrl` + `C`)
|
10. When done, stop the bot (`Ctrl` + `C`)
|
||||||
10. Stop the core dependency services: `just services-stop`
|
11. Stop the services: `just services-stop`
|
||||||
11. (Optional) Stop additional services:
|
12. (Optional) Stop additional services:
|
||||||
- for [LocalAI](#localai): `just localai-stop`
|
- for [LocalAI](#localai): `just localai-stop`
|
||||||
- for [Ollama](#ollama): `just ollama-stop`
|
- for [Ollama](#ollama): `just ollama-stop`
|
||||||
|
|
||||||
@@ -54,11 +76,12 @@ In any case, you will need [🐋 Docker](https://www.docker.com/) as [dependency
|
|||||||
|
|
||||||
You can avoid having a [Rust](https://www.rust-lang.org/) toolchain installed locally and build/run this in a container.
|
You can avoid having a [Rust](https://www.rust-lang.org/) toolchain installed locally and build/run this in a container.
|
||||||
|
|
||||||
1. Start the core dependency services (Postgres, Synapse, Element Web): `just services-start`
|
1. (Optional) Choose a homeserver: `just homeserver-init continuwuity` (or `synapse`). Default is `continuwuity`.
|
||||||
2. (Only the first time around) Prepare initial app configuration in `var/app/container/config.yml`: `just app-container-prepare`
|
2. Start the homeserver and Element Web: `just services-start`
|
||||||
3. (Only the first time around) [Prepare your configuration file](#prepare-your-configuration-file)
|
3. (Only the first time around) Prepare initial app configuration in `var/app/container/config.yml`: `just app-container-prepare`
|
||||||
4. (Only the first time around) Prepare initial default Matrix user accounts (`admin` and `baibot`): `just users-prepare`
|
4. (Only the first time around) [Prepare your configuration file](#prepare-your-configuration-file)
|
||||||
5. (Optional) Start additional services depending on which [agent provider you've chosen](#choosing-an-agent-provider):
|
5. (Only the first time around) Prepare initial default Matrix user accounts (`admin` and `baibot`): `just users-prepare`
|
||||||
|
6. (Optional) Start additional services depending on which [agent provider you've chosen](#choosing-an-agent-provider):
|
||||||
- for [LocalAI](#localai):
|
- for [LocalAI](#localai):
|
||||||
- Start services: `just localai-start`
|
- Start services: `just localai-start`
|
||||||
- Wait a while for LocalAI to start up. It has a lot of models to download. Monitor progress using `just localai-tail-logs`
|
- Wait a while for LocalAI to start up. It has a lot of models to download. Monitor progress using `just localai-tail-logs`
|
||||||
@@ -66,12 +89,12 @@ You can avoid having a [Rust](https://www.rust-lang.org/) toolchain installed lo
|
|||||||
- for [Ollama](#ollama):
|
- for [Ollama](#ollama):
|
||||||
- Start services: `just ollama-start`
|
- Start services: `just ollama-start`
|
||||||
- (Only the first time around) Pull the model configured in `agents.static_definitions` in the configuration file: `just ollama-pull-model gemma2:2b`
|
- (Only the first time around) Pull the model configured in `agents.static_definitions` in the configuration file: `just ollama-pull-model gemma2:2b`
|
||||||
6. Start the bot: `just run-in-container`
|
7. Start the bot: `just run-in-container`
|
||||||
7. Go to http://element.127.0.0.1.nip.io:42025/ and login with `admin` / `admin`
|
8. Go to http://element.127.0.0.1.nip.io:42025/ and login with `admin` / `admin`
|
||||||
8. Create a new room and invite `@baibot:synapse.127.0.0.1.nip.io`
|
9. Create a new room and invite `@baibot:continuwuity.127.0.0.1.nip.io` (or `@baibot:synapse.127.0.0.1.nip.io` if using Synapse)
|
||||||
9. When done, stop the bot (`Ctrl` + `C`)
|
10. When done, stop the bot (`Ctrl` + `C`)
|
||||||
10. Stop the dependency services: `just services-stop`
|
11. Stop the services: `just services-stop`
|
||||||
11. (Optional) Stop additional services:
|
12. (Optional) Stop additional services:
|
||||||
- for [LocalAI](#localai): `just localai-stop`
|
- for [LocalAI](#localai): `just localai-stop`
|
||||||
- for [Ollama](#ollama): `just ollama-stop`
|
- for [Ollama](#ollama): `just ollama-stop`
|
||||||
|
|
||||||
|
|||||||
@@ -8,7 +8,7 @@ You can also use **different models within the same room** (e.g. [💬 text-gene
|
|||||||
|
|
||||||
The bot supports the following use-purposes:
|
The bot supports the following use-purposes:
|
||||||
|
|
||||||
- [💬 text-generation](#-text-generation): communicating with you via text (though certain models may "see" images as well)
|
- [💬 text-generation](#-text-generation): communicating with you via text (though certain models may also process images and files)
|
||||||
- [🦻 speech-to-text](#-speech-to-text): turning your voice messages into text
|
- [🦻 speech-to-text](#-speech-to-text): turning your voice messages into text
|
||||||
- [🗣️ text-to-speech](#%EF%B8%8F-text-to-speech): turning bot or users text messages into voice messages
|
- [🗣️ text-to-speech](#%EF%B8%8F-text-to-speech): turning bot or users text messages into voice messages
|
||||||
- [🖌️ image-generation](#%EF%B8%8F-image-generation): generating images based on instructions
|
- [🖌️ image-generation](#%EF%B8%8F-image-generation): generating images based on instructions
|
||||||
@@ -26,12 +26,14 @@ Text Generation is the bot's ability to **respond to users' messages with text**
|
|||||||
|
|
||||||

|

|
||||||
|
|
||||||
Some models also support vision, so you may be able to mix text and images in the same conversation.
|
Some models also support vision and document understanding, so you may be able to mix text, images, and files (PDFs, text documents, etc.) in the same conversation. Note that certain providers may not support all file types or may have issues with specific files (e.g. scanned/image-based PDFs). If a file is rejected by the provider, the conversation thread may become unusable — start a new thread to work around this.
|
||||||
|
|
||||||
In multi-user (group) rooms, to avoid disturbing the normal conversation between people, the bot is auto-configured to only respond to messages starting with the command prefix (`!bai`) or direct mentions via the [💬 Text Generation / 🗟 Prefix Requirement Type](./configuration/text-generation.md#-prefix-requirement-type) setting.
|
In multi-user (group) rooms, to avoid disturbing the normal conversation between people, the bot is auto-configured to only respond to messages starting with the command prefix (`!bai`) or direct mentions via the [💬 Text Generation / 🗟 Prefix Requirement Type](./configuration/text-generation.md#-prefix-requirement-type) setting.
|
||||||
|
|
||||||
Normally, the bot only responds to allowed [👥 Users](./access.md#-users). In certain cases, it's useful for an allowed user to provoke the bot to respond even in foreign threads or reply chains. You can learn more about this feature in the [On-demand involvement](./features.md#on-demand-involvement) section below.
|
Normally, the bot only responds to allowed [👥 Users](./access.md#-users). In certain cases, it's useful for an allowed user to provoke the bot to respond even in foreign threads or reply chains. You can learn more about this feature in the [On-demand involvement](./features.md#on-demand-involvement) section below.
|
||||||
|
|
||||||
|
If needed, the bot can also attach sender metadata to conversation messages before sending them to the model, which can help the model distinguish between participants in multi-user rooms. See [🛠️ Configuration / 💬 Text Generation / 👤 Sender Context Mode](./configuration/text-generation.md#-sender-context-mode).
|
||||||
|
|
||||||
A few other features (like [🗣️ Text-to-Speech](#️-text-to-speech) and [🦻 Speech-to-Text](#-speech-to-text)) combine well with Text Generation, so you **don't necessarily need to communicate with the bot via text** (with [Seamless voice interaction](#seamless-voice-interaction), you can communicate only with voice).
|
A few other features (like [🗣️ Text-to-Speech](#️-text-to-speech) and [🦻 Speech-to-Text](#-speech-to-text)) combine well with Text Generation, so you **don't necessarily need to communicate with the bot via text** (with [Seamless voice interaction](#seamless-voice-interaction), you can communicate only with voice).
|
||||||
|
|
||||||
You may also wish to see:
|
You may also wish to see:
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
base_url: https://api.openai.com/v1
|
base_url: https://api.openai.com/v1
|
||||||
api_key: YOUR_API_KEY_HERE
|
api_key: YOUR_API_KEY_HERE
|
||||||
text_generation:
|
text_generation:
|
||||||
model_id: gpt-5.2
|
model_id: gpt-5.4
|
||||||
prompt: "You are a brief, but helpful bot called {{ baibot_name }} powered by the {{ baibot_model_id }} model. The date/time of this conversation's start is: {{ baibot_conversation_start_time_utc }}."
|
prompt: "You are a brief, but helpful bot called {{ baibot_name }} powered by the {{ baibot_model_id }} model. The date/time of this conversation's start is: {{ baibot_conversation_start_time_utc }}."
|
||||||
temperature: 1.0
|
temperature: 1.0
|
||||||
# Reasoning models need to use `max_completion_tokens` instead of `max_response_tokens`.
|
# Reasoning models need to use `max_completion_tokens` instead of `max_response_tokens`.
|
||||||
|
|||||||
@@ -11,7 +11,7 @@ This is related to the [💬 Text Generation](./features.md#-text-generation) fe
|
|||||||
|
|
||||||
If there's a text-generation handler agent configured, the bot **may** respond to messages sent in the room.
|
If there's a text-generation handler agent configured, the bot **may** respond to messages sent in the room.
|
||||||
|
|
||||||
Some models also support vision, so you may be able to mix text and images in the same conversation.
|
Some models also support vision and document understanding, so you may be able to mix text, images, and files (PDFs, text documents, etc.) in the same conversation.
|
||||||
|
|
||||||
See screenshots of:
|
See screenshots of:
|
||||||
|
|
||||||
|
|||||||
@@ -1,12 +1,21 @@
|
|||||||
homeserver:
|
homeserver:
|
||||||
# The canonical homeserver domain name
|
# The canonical homeserver domain name
|
||||||
server_name: synapse.127.0.0.1.nip.io
|
server_name: __HOMESERVER_SERVER_NAME__
|
||||||
url: http://synapse.127.0.0.1.nip.io:42020
|
url: __HOMESERVER_URL__
|
||||||
|
|
||||||
user:
|
user:
|
||||||
mxid_localpart: baibot
|
mxid_localpart: baibot
|
||||||
|
|
||||||
|
# Authentication: set EITHER password OR access_token + device_id.
|
||||||
|
#
|
||||||
|
# Password-based login (traditional homeservers):
|
||||||
password: baibot
|
password: baibot
|
||||||
|
|
||||||
|
# Access token login (for Matrix Authentication Service/OIDC-enabled homeservers):
|
||||||
|
# Generate a token via: mas-cli manage issue-compatibility-token <username> [device_id]
|
||||||
|
# access_token: null
|
||||||
|
# device_id: null
|
||||||
|
|
||||||
# The name the bot uses as a display name and when it refers to itself.
|
# The name the bot uses as a display name and when it refers to itself.
|
||||||
# Leave empty to use the default (baibot).
|
# Leave empty to use the default (baibot).
|
||||||
name: baibot
|
name: baibot
|
||||||
@@ -45,7 +54,7 @@ room:
|
|||||||
access:
|
access:
|
||||||
# Space-separated list of MXID patterns which specify who is an admin.
|
# Space-separated list of MXID patterns which specify who is an admin.
|
||||||
admin_patterns:
|
admin_patterns:
|
||||||
- "@admin:synapse.127.0.0.1.nip.io"
|
- "@admin:__HOMESERVER_SERVER_NAME__"
|
||||||
|
|
||||||
persistence:
|
persistence:
|
||||||
# This is unset here, because we expect the configuration to come from an environment variable (BAIBOT_PERSISTENCE_DATA_DIR_PATH).
|
# This is unset here, because we expect the configuration to come from an environment variable (BAIBOT_PERSISTENCE_DATA_DIR_PATH).
|
||||||
@@ -82,7 +91,7 @@ agents:
|
|||||||
# base_url: https://api.openai.com/v1
|
# base_url: https://api.openai.com/v1
|
||||||
# api_key: ""
|
# api_key: ""
|
||||||
# text_generation:
|
# text_generation:
|
||||||
# model_id: gpt-5.2
|
# model_id: gpt-5.4
|
||||||
# prompt: "You are a brief, but helpful bot called {{ baibot_name }} powered by the {{ baibot_model_id }} model. The date/time of this conversation's start is: {{ baibot_conversation_start_time_utc }}."
|
# prompt: "You are a brief, but helpful bot called {{ baibot_name }} powered by the {{ baibot_model_id }} model. The date/time of this conversation's start is: {{ baibot_conversation_start_time_utc }}."
|
||||||
# temperature: 1.0
|
# temperature: 1.0
|
||||||
# # Reasoning models need to use `max_completion_tokens` instead of `max_response_tokens`.
|
# # Reasoning models need to use `max_completion_tokens` instead of `max_response_tokens`.
|
||||||
@@ -157,7 +166,7 @@ initial_global_config:
|
|||||||
# Space-separated list of MXID patterns which specify who can use the bot.
|
# Space-separated list of MXID patterns which specify who can use the bot.
|
||||||
# By default, we let anyone on the homeserver use the bot.
|
# By default, we let anyone on the homeserver use the bot.
|
||||||
user_patterns:
|
user_patterns:
|
||||||
- "@*:synapse.127.0.0.1.nip.io"
|
- "@*:__HOMESERVER_SERVER_NAME__"
|
||||||
|
|
||||||
# Controls logging.
|
# Controls logging.
|
||||||
#
|
#
|
||||||
|
|||||||
23
etc/services/continuwuity/compose.yml
Normal file
23
etc/services/continuwuity/compose.yml
Normal file
@@ -0,0 +1,23 @@
|
|||||||
|
services:
|
||||||
|
continuwuity:
|
||||||
|
image: forgejo.ellis.link/continuwuation/continuwuity:v0.5.6
|
||||||
|
user: "${UID}:${GID}"
|
||||||
|
restart: unless-stopped
|
||||||
|
cap_drop:
|
||||||
|
- ALL
|
||||||
|
read_only: true
|
||||||
|
environment:
|
||||||
|
CONDUWUIT_CONFIG: /etc/continuwuity/continuwuity.toml
|
||||||
|
CONDUWUIT_DATABASE_PATH: /var/lib/continuwuity
|
||||||
|
ports:
|
||||||
|
- "${SERVICE_CONTINUWUITY_BIND_PORT_CLIENT_API}:6167"
|
||||||
|
volumes:
|
||||||
|
- ../../etc/services/continuwuity/config:/etc/continuwuity:ro
|
||||||
|
- ./continuwuity/data:/var/lib/continuwuity
|
||||||
|
tmpfs:
|
||||||
|
- /tmp:rw,noexec,nosuid,size=500m
|
||||||
|
|
||||||
|
networks:
|
||||||
|
default:
|
||||||
|
name: ${NETWORK_NAME}
|
||||||
|
external: true
|
||||||
19
etc/services/continuwuity/config/continuwuity.toml
Normal file
19
etc/services/continuwuity/config/continuwuity.toml
Normal file
@@ -0,0 +1,19 @@
|
|||||||
|
[global]
|
||||||
|
server_name = "continuwuity.127.0.0.1.nip.io"
|
||||||
|
|
||||||
|
address = "0.0.0.0"
|
||||||
|
port = 6167
|
||||||
|
|
||||||
|
database_path = "/var/lib/continuwuity"
|
||||||
|
|
||||||
|
allow_registration = true
|
||||||
|
yes_i_am_very_very_sure_i_want_an_open_registration_server_prone_to_abuse = true
|
||||||
|
|
||||||
|
new_user_displayname_suffix = ""
|
||||||
|
|
||||||
|
max_request_size = 20_000_000
|
||||||
|
|
||||||
|
allow_federation = false
|
||||||
|
trusted_servers = ["matrix.org"]
|
||||||
|
|
||||||
|
log = "info,state_res=warn,rocket=off,_=off,sled=off"
|
||||||
48
etc/services/continuwuity/register-user.sh
Executable file
48
etc/services/continuwuity/register-user.sh
Executable file
@@ -0,0 +1,48 @@
|
|||||||
|
#!/bin/sh
|
||||||
|
set -eu
|
||||||
|
|
||||||
|
if [ $# -ne 3 ]; then
|
||||||
|
echo "Usage: $0 <env-file> <username> <password>"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
ENV_FILE="$1"
|
||||||
|
USERNAME="$2"
|
||||||
|
PASSWORD="$3"
|
||||||
|
|
||||||
|
SERVER="http://$(grep '^SERVICE_CONTINUWUITY_BIND_PORT_CLIENT_API=' "${ENV_FILE}" | cut -d= -f2)"
|
||||||
|
REGISTER_URL="${SERVER}/_matrix/client/v3/register"
|
||||||
|
|
||||||
|
echo "Registering user '${USERNAME}' on ${SERVER}..."
|
||||||
|
|
||||||
|
SESSION_RESPONSE=$(curl -s -X POST "${REGISTER_URL}" \
|
||||||
|
-H 'Content-Type: application/json' \
|
||||||
|
-d "{\"username\": \"${USERNAME}\", \"password\": \"${PASSWORD}\"}")
|
||||||
|
|
||||||
|
SESSION_ID=$(echo "${SESSION_RESPONSE}" | grep -o '"session":"[^"]*"' | head -1 | cut -d'"' -f4)
|
||||||
|
if [ -z "${SESSION_ID}" ]; then
|
||||||
|
echo "Error: Could not get session ID. Response: ${SESSION_RESPONSE}"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Determine the required auth flow from the server response.
|
||||||
|
# The first user requires m.login.registration_token (bootstrap token from logs).
|
||||||
|
# Subsequent users use m.login.dummy (open registration).
|
||||||
|
if echo "${SESSION_RESPONSE}" | grep -q 'm.login.registration_token'; then
|
||||||
|
CONTAINER_ID=$(docker ps -q --filter name=baibot-continuwuity-continuwuity)
|
||||||
|
REG_TOKEN=$(docker logs "${CONTAINER_ID}" 2>&1 | sed 's/\x1b\[[0-9;]*m//g' | grep 'using the registration token' | grep -oP 'registration token \K[A-Za-z0-9]+' | head -1)
|
||||||
|
AUTH_BODY="{\"type\": \"m.login.registration_token\", \"token\": \"${REG_TOKEN}\", \"session\": \"${SESSION_ID}\"}"
|
||||||
|
else
|
||||||
|
AUTH_BODY="{\"type\": \"m.login.dummy\", \"session\": \"${SESSION_ID}\"}"
|
||||||
|
fi
|
||||||
|
|
||||||
|
RESULT=$(curl -s -X POST "${REGISTER_URL}" \
|
||||||
|
-H 'Content-Type: application/json' \
|
||||||
|
-d "{\"username\": \"${USERNAME}\", \"password\": \"${PASSWORD}\", \"auth\": ${AUTH_BODY}}")
|
||||||
|
|
||||||
|
if echo "${RESULT}" | grep -q '"user_id"'; then
|
||||||
|
echo "Successfully registered user: $(echo "${RESULT}" | grep -o '"user_id":"[^"]*"' | cut -d'"' -f4)"
|
||||||
|
else
|
||||||
|
echo "Registration failed. Response: ${RESULT}"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
21
etc/services/element-web/compose.yml
Normal file
21
etc/services/element-web/compose.yml
Normal file
@@ -0,0 +1,21 @@
|
|||||||
|
services:
|
||||||
|
element-web:
|
||||||
|
image: ghcr.io/element-hq/element-web:v1.12.15
|
||||||
|
user: "${UID}:${GID}"
|
||||||
|
restart: unless-stopped
|
||||||
|
environment:
|
||||||
|
ELEMENT_WEB_PORT: 8080
|
||||||
|
ports:
|
||||||
|
- "${SERVICE_ELEMENT_WEB_BIND_PORT_HTTP}:8080"
|
||||||
|
volumes:
|
||||||
|
- ./element-web/config.json:/app/config.json:ro
|
||||||
|
tmpfs:
|
||||||
|
- /var/cache/nginx:rw,mode=777
|
||||||
|
- /var/run:rw,mode=777
|
||||||
|
- /tmp/element-web-config:rw,mode=777
|
||||||
|
- /etc/nginx/conf.d:rw,mode=777
|
||||||
|
|
||||||
|
networks:
|
||||||
|
default:
|
||||||
|
name: ${NETWORK_NAME}
|
||||||
|
external: true
|
||||||
@@ -1,5 +1,5 @@
|
|||||||
{
|
{
|
||||||
"default_hs_url": "http://synapse.127.0.0.1.nip.io:42020",
|
"default_hs_url": "__HOMESERVER_CLIENT_URL__",
|
||||||
"default_is_url": "https://vector.im",
|
"default_is_url": "https://vector.im",
|
||||||
"integrations_ui_url": "https://scalar.vector.im/",
|
"integrations_ui_url": "https://scalar.vector.im/",
|
||||||
"integrations_rest_url": "https://scalar.vector.im/api",
|
"integrations_rest_url": "https://scalar.vector.im/api",
|
||||||
@@ -3,6 +3,8 @@ SERVICE_SYNAPSE_BIND_PORT_FEDERATION_API=127.0.0.1:42028
|
|||||||
|
|
||||||
SERVICE_ELEMENT_WEB_BIND_PORT_HTTP=127.0.0.1:42025
|
SERVICE_ELEMENT_WEB_BIND_PORT_HTTP=127.0.0.1:42025
|
||||||
|
|
||||||
|
SERVICE_CONTINUWUITY_BIND_PORT_CLIENT_API=127.0.0.1:42030
|
||||||
|
|
||||||
SERVICE_OLLAMA_BIND_PORT_HTTP=127.0.0.1:42026
|
SERVICE_OLLAMA_BIND_PORT_HTTP=127.0.0.1:42026
|
||||||
|
|
||||||
# See https://localai.io/basics/container/#all-in-one-images for the list of available images
|
# See https://localai.io/basics/container/#all-in-one-images for the list of available images
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
services:
|
services:
|
||||||
ollama:
|
ollama:
|
||||||
image: docker.io/ollama/ollama:0.15.4
|
image: docker.io/ollama/ollama:0.20.5
|
||||||
restart: unless-stopped
|
restart: unless-stopped
|
||||||
ports:
|
ports:
|
||||||
- "${SERVICE_OLLAMA_BIND_PORT_HTTP}:11434"
|
- "${SERVICE_OLLAMA_BIND_PORT_HTTP}:11434"
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
services:
|
services:
|
||||||
postgres:
|
postgres:
|
||||||
image: docker.io/postgres:18.1-alpine
|
image: docker.io/postgres:18.3-alpine
|
||||||
user: ${UID}:${GID}
|
user: ${UID}:${GID}
|
||||||
restart: unless-stopped
|
restart: unless-stopped
|
||||||
environment:
|
environment:
|
||||||
@@ -14,7 +14,7 @@ services:
|
|||||||
- /etc/passwd:/etc/passwd:ro
|
- /etc/passwd:/etc/passwd:ro
|
||||||
|
|
||||||
synapse:
|
synapse:
|
||||||
image: ghcr.io/element-hq/synapse:v1.146.0
|
image: ghcr.io/element-hq/synapse:v1.151.0
|
||||||
user: "${UID}:${GID}"
|
user: "${UID}:${GID}"
|
||||||
restart: unless-stopped
|
restart: unless-stopped
|
||||||
entrypoint: python
|
entrypoint: python
|
||||||
@@ -23,25 +23,9 @@ services:
|
|||||||
- "${SERVICE_SYNAPSE_BIND_PORT_CLIENT_API}:8008"
|
- "${SERVICE_SYNAPSE_BIND_PORT_CLIENT_API}:8008"
|
||||||
- "${SERVICE_SYNAPSE_BIND_PORT_FEDERATION_API}:8008"
|
- "${SERVICE_SYNAPSE_BIND_PORT_FEDERATION_API}:8008"
|
||||||
volumes:
|
volumes:
|
||||||
- ../../etc/services/core/synapse/config:/config:ro
|
- ../../etc/services/synapse/config:/config:ro
|
||||||
- ./synapse/media-store:/media-store
|
- ./synapse/media-store:/media-store
|
||||||
|
|
||||||
element-web:
|
|
||||||
image: ghcr.io/element-hq/element-web:v1.12.9
|
|
||||||
user: "${UID}:${GID}"
|
|
||||||
restart: unless-stopped
|
|
||||||
environment:
|
|
||||||
ELEMENT_WEB_PORT: 8080
|
|
||||||
ports:
|
|
||||||
- "${SERVICE_ELEMENT_WEB_BIND_PORT_HTTP}:8080"
|
|
||||||
volumes:
|
|
||||||
- ../../etc/services/core/element-web/config.json:/app/config.json:ro
|
|
||||||
tmpfs:
|
|
||||||
- /var/cache/nginx:rw,mode=777
|
|
||||||
- /var/run:rw,mode=777
|
|
||||||
- /tmp/element-web-config:rw,mode=777
|
|
||||||
- /etc/nginx/conf.d:rw,mode=777
|
|
||||||
|
|
||||||
networks:
|
networks:
|
||||||
default:
|
default:
|
||||||
name: ${NETWORK_NAME}
|
name: ${NETWORK_NAME}
|
||||||
214
justfile
214
justfile
@@ -2,10 +2,33 @@ project_name := "baibot"
|
|||||||
container_image_name := "localhost/baibot"
|
container_image_name := "localhost/baibot"
|
||||||
project_container_network := "baibot"
|
project_container_network := "baibot"
|
||||||
|
|
||||||
|
admin_username := "admin"
|
||||||
|
admin_password := "admin"
|
||||||
|
bot_username := "baibot"
|
||||||
|
bot_password := "baibot"
|
||||||
|
|
||||||
|
homeserver := `cat var/homeserver 2>/dev/null || echo continuwuity`
|
||||||
|
|
||||||
|
mise_data_dir := env("MISE_DATA_DIR", justfile_directory() / "var/mise")
|
||||||
|
mise_trusted_config_paths := justfile_directory() / "mise.toml"
|
||||||
|
|
||||||
# Show help by default
|
# Show help by default
|
||||||
default:
|
default:
|
||||||
@just --list --justfile {{ justfile() }}
|
@just --list --justfile {{ justfile() }}
|
||||||
|
|
||||||
|
# Selects which homeserver implementation to use (continuwuity or synapse)
|
||||||
|
homeserver-init value:
|
||||||
|
#!/bin/sh
|
||||||
|
mkdir -p {{ justfile_directory() }}/var
|
||||||
|
echo {{ value }} > {{ justfile_directory() }}/var/homeserver
|
||||||
|
echo ""
|
||||||
|
echo "⚠️ If you had already prepared your app configuration (var/app/local/config.yml or var/app/container/config.yml),"
|
||||||
|
echo " you will need to update it manually or delete it and re-run the prepare step."
|
||||||
|
echo " You should also delete var/app/local/data and/or var/app/container/data,"
|
||||||
|
echo " as old application state is not compatible across homeserver implementations."
|
||||||
|
echo ""
|
||||||
|
echo "⚠️ If Element Web was already prepared, delete var/services/element-web/ to regenerate its config."
|
||||||
|
|
||||||
# Builds and runs a development binary
|
# Builds and runs a development binary
|
||||||
run-locally *extra_args: app-local-prepare
|
run-locally *extra_args: app-local-prepare
|
||||||
RUST_BACKTRACE=1 \
|
RUST_BACKTRACE=1 \
|
||||||
@@ -65,9 +88,13 @@ docker-compose services_type *extra_args:
|
|||||||
-p {{ project_name }}-{{ services_type }} \
|
-p {{ project_name }}-{{ services_type }} \
|
||||||
{{ extra_args }}
|
{{ extra_args }}
|
||||||
|
|
||||||
# Runs a docker-compose command against the core services
|
# Runs a docker-compose command against the synapse services
|
||||||
docker-compose-core *extra_args:
|
docker-compose-synapse *extra_args:
|
||||||
just docker-compose core {{ extra_args }}
|
just docker-compose synapse {{ extra_args }}
|
||||||
|
|
||||||
|
# Runs a docker-compose command against the element-web services
|
||||||
|
docker-compose-element-web *extra_args:
|
||||||
|
just docker-compose element-web {{ extra_args }}
|
||||||
|
|
||||||
# Runs a docker-compose command against the localai services
|
# Runs a docker-compose command against the localai services
|
||||||
docker-compose-localai *extra_args:
|
docker-compose-localai *extra_args:
|
||||||
@@ -77,17 +104,52 @@ docker-compose-localai *extra_args:
|
|||||||
docker-compose-ollama *extra_args:
|
docker-compose-ollama *extra_args:
|
||||||
just docker-compose ollama {{ extra_args }}
|
just docker-compose ollama {{ extra_args }}
|
||||||
|
|
||||||
# Runs all core dependency components (in the background)
|
# Runs a docker-compose command against the continuwuity services
|
||||||
services-start: services-prepare (docker-compose-core "up" "-d")
|
docker-compose-continuwuity *extra_args:
|
||||||
|
just docker-compose continuwuity {{ extra_args }}
|
||||||
|
|
||||||
# Stops all core dependency components
|
# Runs the homeserver and Element Web (in the background)
|
||||||
services-stop: (docker-compose-core "down")
|
services-start: services-prepare
|
||||||
|
just -f {{ justfile_directory() }}/justfile {{ homeserver }}-start
|
||||||
|
just -f {{ justfile_directory() }}/justfile element-web-start
|
||||||
|
|
||||||
# Tails the logs for all running core services
|
# Stops Element Web and the homeserver
|
||||||
services-tail-logs: (docker-compose-core "logs" "-f")
|
services-stop:
|
||||||
|
just -f {{ justfile_directory() }}/justfile element-web-stop
|
||||||
|
just -f {{ justfile_directory() }}/justfile {{ homeserver }}-stop
|
||||||
|
|
||||||
# Prepares the core services for running
|
# Tails the logs for the homeserver and Element Web
|
||||||
services-prepare: _prepare-var-services-env _prepare-var-services-postgres _prepare-var-services-synapse _prepare-container-network
|
services-tail-logs:
|
||||||
|
just -f {{ justfile_directory() }}/justfile {{ homeserver }}-tail-logs
|
||||||
|
|
||||||
|
# Prepares the homeserver and Element Web for running
|
||||||
|
services-prepare:
|
||||||
|
just -f {{ justfile_directory() }}/justfile {{ homeserver }}-prepare
|
||||||
|
just -f {{ justfile_directory() }}/justfile element-web-prepare
|
||||||
|
|
||||||
|
# Runs Synapse (in the background)
|
||||||
|
synapse-start: synapse-prepare (docker-compose-synapse "up" "-d")
|
||||||
|
|
||||||
|
# Stops Synapse
|
||||||
|
synapse-stop: (docker-compose-synapse "down")
|
||||||
|
|
||||||
|
# Tails the logs for Synapse
|
||||||
|
synapse-tail-logs: (docker-compose-synapse "logs" "-f")
|
||||||
|
|
||||||
|
# Prepares Synapse for running
|
||||||
|
synapse-prepare: _prepare-var-services-env _prepare-var-services-postgres _prepare-var-services-synapse _prepare-container-network
|
||||||
|
|
||||||
|
# Runs Element Web (in the background)
|
||||||
|
element-web-start: element-web-prepare (docker-compose-element-web "up" "-d")
|
||||||
|
|
||||||
|
# Stops Element Web
|
||||||
|
element-web-stop: (docker-compose-element-web "down")
|
||||||
|
|
||||||
|
# Tails the logs for Element Web
|
||||||
|
element-web-tail-logs: (docker-compose-element-web "logs" "-f")
|
||||||
|
|
||||||
|
# Prepares Element Web for running
|
||||||
|
element-web-prepare: _prepare-var-services-env _prepare-var-services-element-web _prepare-container-network
|
||||||
|
|
||||||
# Runs LocalAI (in the background)
|
# Runs LocalAI (in the background)
|
||||||
localai-start: localai-prepare (docker-compose-localai "up" "-d")
|
localai-start: localai-prepare (docker-compose-localai "up" "-d")
|
||||||
@@ -113,6 +175,27 @@ ollama-tail-logs: (docker-compose-ollama "logs" "-f")
|
|||||||
# Prepares Ollama for running
|
# Prepares Ollama for running
|
||||||
ollama-prepare: _prepare-var-services-env _prepare-var-services-ollama _prepare-container-network
|
ollama-prepare: _prepare-var-services-env _prepare-var-services-ollama _prepare-container-network
|
||||||
|
|
||||||
|
# Runs Continuwuity (in the background)
|
||||||
|
continuwuity-start: continuwuity-prepare (docker-compose-continuwuity "up" "-d")
|
||||||
|
|
||||||
|
# Stops Continuwuity
|
||||||
|
continuwuity-stop: (docker-compose-continuwuity "down")
|
||||||
|
|
||||||
|
# Tails the logs for Continuwuity
|
||||||
|
continuwuity-tail-logs: (docker-compose-continuwuity "logs" "-f")
|
||||||
|
|
||||||
|
# Prepares Continuwuity for running
|
||||||
|
continuwuity-prepare: _prepare-var-services-env _prepare-var-services-continuwuity _prepare-container-network
|
||||||
|
|
||||||
|
# Registers a user on Continuwuity via the Matrix Client-Server API
|
||||||
|
continuwuity-register-user username password:
|
||||||
|
{{ justfile_directory() }}/etc/services/continuwuity/register-user.sh {{ justfile_directory() }}/var/services/env {{ username }} {{ password }}
|
||||||
|
|
||||||
|
# Prepares the Continuwuity user accounts
|
||||||
|
continuwuity-users-prepare: continuwuity-prepare
|
||||||
|
just -f {{ justfile_directory() }}/justfile continuwuity-register-user "{{ admin_username }}" "{{ admin_password }}"
|
||||||
|
just -f {{ justfile_directory() }}/justfile continuwuity-register-user "{{ bot_username }}" "{{ bot_password }}"
|
||||||
|
|
||||||
# Pulls an Ollama model
|
# Pulls an Ollama model
|
||||||
ollama-pull-model model_id:
|
ollama-pull-model model_id:
|
||||||
just -f {{ justfile_directory() }}/justfile docker-compose-ollama \
|
just -f {{ justfile_directory() }}/justfile docker-compose-ollama \
|
||||||
@@ -126,16 +209,20 @@ app-local-prepare: _prepare-var-app-local-config_yml _prepare-var-app-local-data
|
|||||||
app-container-prepare: _prepare-var-app-container-config_yml _prepare-var-app-container-data
|
app-container-prepare: _prepare-var-app-container-config_yml _prepare-var-app-container-data
|
||||||
|
|
||||||
# Prepares the user accounts
|
# Prepares the user accounts
|
||||||
users-prepare: services-prepare
|
users-prepare:
|
||||||
just -f {{ justfile_directory() }}/justfile synapse-register-admin-user "admin" "admin"
|
just -f {{ justfile_directory() }}/justfile {{ homeserver }}-users-prepare
|
||||||
just -f {{ justfile_directory() }}/justfile synapse-register-regular-user "baibot" "baibot"
|
|
||||||
|
# Prepares the Synapse user accounts
|
||||||
|
synapse-users-prepare: synapse-prepare
|
||||||
|
just -f {{ justfile_directory() }}/justfile synapse-register-admin-user "{{ admin_username }}" "{{ admin_password }}"
|
||||||
|
just -f {{ justfile_directory() }}/justfile synapse-register-regular-user "{{ bot_username }}" "{{ bot_password }}"
|
||||||
|
|
||||||
# Starts a Postgres CLI (psql)
|
# Starts a Postgres CLI (psql)
|
||||||
postgres-cli: services-prepare (docker-compose-core "exec" "postgres" "/bin/sh" "-c" "'PGUSER=synapse PGPASSWORD=synapse-password PGDATABASE=homeserver psql -h postgres'")
|
postgres-cli: synapse-prepare (docker-compose-synapse "exec" "postgres" "/bin/sh" "-c" "'PGUSER=synapse PGPASSWORD=synapse-password PGDATABASE=homeserver psql -h postgres'")
|
||||||
|
|
||||||
# Creates an administrator user
|
# Creates an administrator user on Synapse
|
||||||
synapse-register-admin-user username password: services-prepare
|
synapse-register-admin-user username password: synapse-prepare
|
||||||
just -f {{ justfile_directory() }}/justfile docker-compose-core \
|
just -f {{ justfile_directory() }}/justfile docker-compose-synapse \
|
||||||
exec synapse \
|
exec synapse \
|
||||||
register_new_matrix_user \
|
register_new_matrix_user \
|
||||||
--admin \
|
--admin \
|
||||||
@@ -144,9 +231,9 @@ synapse-register-admin-user username password: services-prepare
|
|||||||
-c /config/homeserver.yaml \
|
-c /config/homeserver.yaml \
|
||||||
http://localhost:8008
|
http://localhost:8008
|
||||||
|
|
||||||
# Create a regular user
|
# Creates a regular user on Synapse
|
||||||
synapse-register-regular-user username password: services-prepare
|
synapse-register-regular-user username password: synapse-prepare
|
||||||
just -f {{ justfile_directory() }}/justfile docker-compose-core \
|
just -f {{ justfile_directory() }}/justfile docker-compose-synapse \
|
||||||
exec synapse \
|
exec synapse \
|
||||||
register_new_matrix_user \
|
register_new_matrix_user \
|
||||||
--no-admin \
|
--no-admin \
|
||||||
@@ -159,6 +246,44 @@ synapse-register-regular-user username password: services-prepare
|
|||||||
clippy *extra_args:
|
clippy *extra_args:
|
||||||
cargo clippy {{ extra_args }}
|
cargo clippy {{ extra_args }}
|
||||||
|
|
||||||
|
# Checks that the code compiles without building
|
||||||
|
check:
|
||||||
|
cargo check
|
||||||
|
|
||||||
|
# Invokes mise with the project-local data directory
|
||||||
|
mise *args: _ensure_mise_data_directory
|
||||||
|
#!/bin/sh
|
||||||
|
export MISE_DATA_DIR="{{ mise_data_dir }}"
|
||||||
|
export MISE_TRUSTED_CONFIG_PATHS="{{ mise_trusted_config_paths }}"
|
||||||
|
mise {{ args }}
|
||||||
|
|
||||||
|
# Runs prek (pre-commit hooks manager) with the given arguments
|
||||||
|
prek *args: _ensure_mise_tools_installed
|
||||||
|
@just --justfile {{ justfile() }} mise exec -- prek {{ args }}
|
||||||
|
|
||||||
|
# Runs pre-commit hooks on staged files
|
||||||
|
prek-run-on-staged *args: _ensure_mise_tools_installed
|
||||||
|
@just --justfile {{ justfile() }} mise exec -- prek run {{ args }}
|
||||||
|
|
||||||
|
# Runs pre-commit hooks on all files
|
||||||
|
prek-run-on-all *args: _ensure_mise_tools_installed
|
||||||
|
@just --justfile {{ justfile() }} mise exec -- prek run --all-files {{ args }}
|
||||||
|
|
||||||
|
# Installs the git pre-commit hook (runs prek automatically before each commit)
|
||||||
|
prek-install-git-pre-commit-hook: _ensure_mise_tools_installed
|
||||||
|
@just --justfile {{ justfile() }} mise exec -- prek install
|
||||||
|
|
||||||
|
# Internal - ensures var/mise directory exists
|
||||||
|
_ensure_mise_data_directory:
|
||||||
|
#!/bin/sh
|
||||||
|
if [ ! -d "{{ mise_data_dir }}" ]; then
|
||||||
|
mkdir -p "{{ mise_data_dir }}"
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Internal - ensures mise tools are installed
|
||||||
|
_ensure_mise_tools_installed: _ensure_mise_data_directory
|
||||||
|
@just --justfile {{ justfile() }} mise install --quiet
|
||||||
|
|
||||||
_prepare-var-services-env:
|
_prepare-var-services-env:
|
||||||
#!/bin/sh
|
#!/bin/sh
|
||||||
cd {{ justfile_directory() }};
|
cd {{ justfile_directory() }};
|
||||||
@@ -188,6 +313,22 @@ _prepare-var-services-synapse:
|
|||||||
mkdir -p var/services/synapse/media-store
|
mkdir -p var/services/synapse/media-store
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
_prepare-var-services-element-web:
|
||||||
|
#!/bin/sh
|
||||||
|
cd {{ justfile_directory() }};
|
||||||
|
|
||||||
|
if [ ! -f var/services/element-web/config.json ]; then
|
||||||
|
mkdir -p var/services/element-web
|
||||||
|
cp {{ justfile_directory() }}/etc/services/element-web/config.json.dist var/services/element-web/config.json
|
||||||
|
|
||||||
|
homeserver="{{ homeserver }}"
|
||||||
|
if [ "$homeserver" = "continuwuity" ]; then
|
||||||
|
sed --in-place 's|__HOMESERVER_CLIENT_URL__|http://continuwuity.127.0.0.1.nip.io:42030|g' var/services/element-web/config.json
|
||||||
|
elif [ "$homeserver" = "synapse" ]; then
|
||||||
|
sed --in-place 's|__HOMESERVER_CLIENT_URL__|http://synapse.127.0.0.1.nip.io:42020|g' var/services/element-web/config.json
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
|
||||||
_prepare-var-services-ollama:
|
_prepare-var-services-ollama:
|
||||||
#!/bin/sh
|
#!/bin/sh
|
||||||
cd {{ justfile_directory() }};
|
cd {{ justfile_directory() }};
|
||||||
@@ -196,6 +337,14 @@ _prepare-var-services-ollama:
|
|||||||
mkdir -p var/services/ollama
|
mkdir -p var/services/ollama
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
_prepare-var-services-continuwuity:
|
||||||
|
#!/bin/sh
|
||||||
|
cd {{ justfile_directory() }};
|
||||||
|
|
||||||
|
if [ ! -f var/services/continuwuity ]; then
|
||||||
|
mkdir -p var/services/continuwuity/data
|
||||||
|
fi
|
||||||
|
|
||||||
_prepare-var-services-localai:
|
_prepare-var-services-localai:
|
||||||
#!/bin/sh
|
#!/bin/sh
|
||||||
cd {{ justfile_directory() }};
|
cd {{ justfile_directory() }};
|
||||||
@@ -219,6 +368,15 @@ _prepare-var-app-local-config_yml:
|
|||||||
if [ ! -f var/app/local/config.yml ]; then
|
if [ ! -f var/app/local/config.yml ]; then
|
||||||
mkdir -p var/app/local
|
mkdir -p var/app/local
|
||||||
cp {{ justfile_directory() }}/etc/app/config.yml.dist var/app/local/config.yml
|
cp {{ justfile_directory() }}/etc/app/config.yml.dist var/app/local/config.yml
|
||||||
|
|
||||||
|
homeserver="{{ homeserver }}"
|
||||||
|
if [ "$homeserver" = "continuwuity" ]; then
|
||||||
|
sed --in-place 's/__HOMESERVER_SERVER_NAME__/continuwuity.127.0.0.1.nip.io/g' var/app/local/config.yml
|
||||||
|
sed --in-place 's|__HOMESERVER_URL__|http://continuwuity.127.0.0.1.nip.io:42030|g' var/app/local/config.yml
|
||||||
|
elif [ "$homeserver" = "synapse" ]; then
|
||||||
|
sed --in-place 's/__HOMESERVER_SERVER_NAME__/synapse.127.0.0.1.nip.io/g' var/app/local/config.yml
|
||||||
|
sed --in-place 's|__HOMESERVER_URL__|http://synapse.127.0.0.1.nip.io:42020|g' var/app/local/config.yml
|
||||||
|
fi
|
||||||
fi
|
fi
|
||||||
|
|
||||||
_prepare-var-app-local-data:
|
_prepare-var-app-local-data:
|
||||||
@@ -236,7 +394,18 @@ _prepare-var-app-container-config_yml:
|
|||||||
if [ ! -f var/app/container/config.yml ]; then
|
if [ ! -f var/app/container/config.yml ]; then
|
||||||
mkdir -p var/app/container
|
mkdir -p var/app/container
|
||||||
cp {{ justfile_directory() }}/etc/app/config.yml.dist var/app/container/config.yml
|
cp {{ justfile_directory() }}/etc/app/config.yml.dist var/app/container/config.yml
|
||||||
sed --in-place 's/synapse.127.0.0.1.nip.io:42020/synapse:8008/g' var/app/container/config.yml
|
|
||||||
|
homeserver="{{ homeserver }}"
|
||||||
|
if [ "$homeserver" = "continuwuity" ]; then
|
||||||
|
sed --in-place 's/__HOMESERVER_SERVER_NAME__/continuwuity.127.0.0.1.nip.io/g' var/app/container/config.yml
|
||||||
|
sed --in-place 's|__HOMESERVER_URL__|http://continuwuity.127.0.0.1.nip.io:42030|g' var/app/container/config.yml
|
||||||
|
sed --in-place 's/continuwuity.127.0.0.1.nip.io:42030/continuwuity:6167/g' var/app/container/config.yml
|
||||||
|
elif [ "$homeserver" = "synapse" ]; then
|
||||||
|
sed --in-place 's/__HOMESERVER_SERVER_NAME__/synapse.127.0.0.1.nip.io/g' var/app/container/config.yml
|
||||||
|
sed --in-place 's|__HOMESERVER_URL__|http://synapse.127.0.0.1.nip.io:42020|g' var/app/container/config.yml
|
||||||
|
sed --in-place 's/synapse.127.0.0.1.nip.io:42020/synapse:8008/g' var/app/container/config.yml
|
||||||
|
fi
|
||||||
|
|
||||||
sed --in-place 's/127.0.0.1:42026/ollama:11434/g' var/app/container/config.yml
|
sed --in-place 's/127.0.0.1:42026/ollama:11434/g' var/app/container/config.yml
|
||||||
sed --in-place 's/127.0.0.1:42027/localai:8080/g' var/app/container/config.yml
|
sed --in-place 's/127.0.0.1:42027/localai:8080/g' var/app/container/config.yml
|
||||||
fi
|
fi
|
||||||
@@ -248,4 +417,3 @@ _prepare-var-app-container-data:
|
|||||||
if [ ! -f var/app/container/data ]; then
|
if [ ! -f var/app/container/data ]; then
|
||||||
mkdir -p var/app/container/data
|
mkdir -p var/app/container/data
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
|||||||
6
mise.toml
Normal file
6
mise.toml
Normal file
@@ -0,0 +1,6 @@
|
|||||||
|
[tools]
|
||||||
|
prek = "0.3.2"
|
||||||
|
|
||||||
|
[settings]
|
||||||
|
# Disable automatic trust prompts - we trust this config
|
||||||
|
yes = true
|
||||||
9
renovate.json
Normal file
9
renovate.json
Normal file
@@ -0,0 +1,9 @@
|
|||||||
|
{
|
||||||
|
"$schema": "https://docs.renovatebot.com/renovate-schema.json",
|
||||||
|
"extends": [
|
||||||
|
"config:recommended"
|
||||||
|
],
|
||||||
|
"labels": [
|
||||||
|
"dependencies"
|
||||||
|
]
|
||||||
|
}
|
||||||
4
rust-toolchain.toml
Normal file
4
rust-toolchain.toml
Normal file
@@ -0,0 +1,4 @@
|
|||||||
|
[toolchain]
|
||||||
|
channel = "1.93.0"
|
||||||
|
components = ["rustfmt", "clippy"]
|
||||||
|
profile = "default"
|
||||||
@@ -33,11 +33,11 @@ pub struct AgentDefinition {
|
|||||||
)]
|
)]
|
||||||
pub provider: AgentProvider,
|
pub provider: AgentProvider,
|
||||||
|
|
||||||
pub config: serde_yaml::Value,
|
pub config: serde_yaml_ng::Value,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl AgentDefinition {
|
impl AgentDefinition {
|
||||||
pub fn new(id: String, provider: AgentProvider, config: serde_yaml::Value) -> Self {
|
pub fn new(id: String, provider: AgentProvider, config: serde_yaml_ng::Value) -> Self {
|
||||||
Self {
|
Self {
|
||||||
id,
|
id,
|
||||||
provider,
|
provider,
|
||||||
|
|||||||
@@ -15,7 +15,7 @@ pub enum Error {
|
|||||||
// Contains the error from the constructor function
|
// Contains the error from the constructor function
|
||||||
ConstructionFailed(anyhow::Error),
|
ConstructionFailed(anyhow::Error),
|
||||||
// Contains the error from the YAML deserialization function
|
// Contains the error from the YAML deserialization function
|
||||||
Yaml(serde_yaml::Error),
|
Yaml(serde_yaml_ng::Error),
|
||||||
}
|
}
|
||||||
|
|
||||||
pub type Result<T> = std::result::Result<T, Error>;
|
pub type Result<T> = std::result::Result<T, Error>;
|
||||||
@@ -69,7 +69,7 @@ pub(super) fn create(
|
|||||||
pub fn create_from_provider_and_yaml_value_config(
|
pub fn create_from_provider_and_yaml_value_config(
|
||||||
provider: &AgentProvider,
|
provider: &AgentProvider,
|
||||||
identifier: &PublicIdentifier,
|
identifier: &PublicIdentifier,
|
||||||
config: serde_yaml::Value,
|
config: serde_yaml_ng::Value,
|
||||||
) -> Result<AgentInstance> {
|
) -> Result<AgentInstance> {
|
||||||
let definition = AgentDefinition::new(identifier.prefixless(), provider.to_owned(), config);
|
let definition = AgentDefinition::new(identifier.prefixless(), provider.to_owned(), config);
|
||||||
|
|
||||||
@@ -79,7 +79,7 @@ pub fn create_from_provider_and_yaml_value_config(
|
|||||||
fn create_controller_from_provider_and_json_value_config(
|
fn create_controller_from_provider_and_json_value_config(
|
||||||
agent_id: &str,
|
agent_id: &str,
|
||||||
provider: &AgentProvider,
|
provider: &AgentProvider,
|
||||||
config: serde_yaml::Value,
|
config: serde_yaml_ng::Value,
|
||||||
) -> Result<ControllerType> {
|
) -> Result<ControllerType> {
|
||||||
match provider {
|
match provider {
|
||||||
AgentProvider::Anthropic => {
|
AgentProvider::Anthropic => {
|
||||||
@@ -112,43 +112,43 @@ fn create_controller_from_provider_and_json_value_config(
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn default_config_for_provider(provider: &AgentProvider) -> serde_yaml::Value {
|
pub fn default_config_for_provider(provider: &AgentProvider) -> serde_yaml_ng::Value {
|
||||||
match provider {
|
match provider {
|
||||||
AgentProvider::Anthropic => {
|
AgentProvider::Anthropic => {
|
||||||
let config = super::provider::anthropic::default_config();
|
let config = super::provider::anthropic::default_config();
|
||||||
serde_yaml::to_value(config).expect("Failed to serialize config")
|
serde_yaml_ng::to_value(config).expect("Failed to serialize config")
|
||||||
}
|
}
|
||||||
AgentProvider::Groq => {
|
AgentProvider::Groq => {
|
||||||
let config = super::provider::groq::default_config();
|
let config = super::provider::groq::default_config();
|
||||||
serde_yaml::to_value(config).expect("Failed to serialize config")
|
serde_yaml_ng::to_value(config).expect("Failed to serialize config")
|
||||||
}
|
}
|
||||||
AgentProvider::LocalAI => {
|
AgentProvider::LocalAI => {
|
||||||
let config = super::provider::localai::default_config();
|
let config = super::provider::localai::default_config();
|
||||||
serde_yaml::to_value(config).expect("Failed to serialize config")
|
serde_yaml_ng::to_value(config).expect("Failed to serialize config")
|
||||||
}
|
}
|
||||||
AgentProvider::Mistral => {
|
AgentProvider::Mistral => {
|
||||||
let config = super::provider::mistral::default_config();
|
let config = super::provider::mistral::default_config();
|
||||||
serde_yaml::to_value(config).expect("Failed to serialize config")
|
serde_yaml_ng::to_value(config).expect("Failed to serialize config")
|
||||||
}
|
}
|
||||||
AgentProvider::Ollama => {
|
AgentProvider::Ollama => {
|
||||||
let config = super::provider::ollama::default_config();
|
let config = super::provider::ollama::default_config();
|
||||||
serde_yaml::to_value(config).expect("Failed to serialize config")
|
serde_yaml_ng::to_value(config).expect("Failed to serialize config")
|
||||||
}
|
}
|
||||||
AgentProvider::OpenAI => {
|
AgentProvider::OpenAI => {
|
||||||
let config = super::provider::openai::default_config();
|
let config = super::provider::openai::default_config();
|
||||||
serde_yaml::to_value(config).expect("Failed to serialize config")
|
serde_yaml_ng::to_value(config).expect("Failed to serialize config")
|
||||||
}
|
}
|
||||||
AgentProvider::OpenAICompat => {
|
AgentProvider::OpenAICompat => {
|
||||||
let config = super::provider::openai_compat::default_config();
|
let config = super::provider::openai_compat::default_config();
|
||||||
serde_yaml::to_value(config).expect("Failed to serialize config")
|
serde_yaml_ng::to_value(config).expect("Failed to serialize config")
|
||||||
}
|
}
|
||||||
AgentProvider::OpenRouter => {
|
AgentProvider::OpenRouter => {
|
||||||
let config = super::provider::openrouter::default_config();
|
let config = super::provider::openrouter::default_config();
|
||||||
serde_yaml::to_value(config).expect("Failed to serialize config")
|
serde_yaml_ng::to_value(config).expect("Failed to serialize config")
|
||||||
}
|
}
|
||||||
AgentProvider::TogetherAI => {
|
AgentProvider::TogetherAI => {
|
||||||
let config = super::provider::togetherai::default_config();
|
let config = super::provider::togetherai::default_config();
|
||||||
serde_yaml::to_value(config).expect("Failed to serialize config")
|
serde_yaml_ng::to_value(config).expect("Failed to serialize config")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -71,6 +71,7 @@ impl ControllerTrait for Controller {
|
|||||||
|
|
||||||
let messages = vec![LLMMessage {
|
let messages = vec![LLMMessage {
|
||||||
author: LLMAuthor::User,
|
author: LLMAuthor::User,
|
||||||
|
sender_id: None,
|
||||||
content: LLMMessageContent::Text("Hello!".to_string()),
|
content: LLMMessageContent::Text("Hello!".to_string()),
|
||||||
timestamp: chrono::Utc::now(),
|
timestamp: chrono::Utc::now(),
|
||||||
}];
|
}];
|
||||||
@@ -108,6 +109,7 @@ impl ControllerTrait for Controller {
|
|||||||
} else {
|
} else {
|
||||||
Some(LLMMessage {
|
Some(LLMMessage {
|
||||||
author: LLMAuthor::Prompt,
|
author: LLMAuthor::Prompt,
|
||||||
|
sender_id: None,
|
||||||
content: LLMMessageContent::Text(prompt_text),
|
content: LLMMessageContent::Text(prompt_text),
|
||||||
timestamp: chrono::Utc::now(),
|
timestamp: chrono::Utc::now(),
|
||||||
})
|
})
|
||||||
@@ -146,10 +148,10 @@ impl ControllerTrait for Controller {
|
|||||||
.temperature_override
|
.temperature_override
|
||||||
.unwrap_or(text_generation_config.temperature);
|
.unwrap_or(text_generation_config.temperature);
|
||||||
|
|
||||||
if let Some(prompt_message) = prompt_message {
|
if let Some(prompt_message) = prompt_message
|
||||||
if let LLMMessageContent::Text(text) = &prompt_message.content {
|
&& let LLMMessageContent::Text(text) = &prompt_message.content
|
||||||
request.system = text.clone();
|
{
|
||||||
}
|
request.system = text.clone();
|
||||||
}
|
}
|
||||||
|
|
||||||
request.model = text_generation_config.model_id.clone();
|
request.model = text_generation_config.model_id.clone();
|
||||||
|
|||||||
@@ -12,12 +12,12 @@ use super::controller::ControllerType;
|
|||||||
|
|
||||||
pub fn create_controller_from_yaml_value_config(
|
pub fn create_controller_from_yaml_value_config(
|
||||||
agent_id: &str,
|
agent_id: &str,
|
||||||
config: serde_yaml::Value,
|
config: serde_yaml_ng::Value,
|
||||||
) -> AgentInstantiationResult<ControllerType> {
|
) -> AgentInstantiationResult<ControllerType> {
|
||||||
let config = match &config {
|
let config = match &config {
|
||||||
serde_yaml::Value::Mapping(_) => {
|
serde_yaml_ng::Value::Mapping(_) => {
|
||||||
let config: Config =
|
let config: Config =
|
||||||
serde_yaml::from_value(config).map_err(AgentInstantiationError::Yaml)?;
|
serde_yaml_ng::from_value(config).map_err(AgentInstantiationError::Yaml)?;
|
||||||
|
|
||||||
config
|
config
|
||||||
.validate()
|
.validate()
|
||||||
|
|||||||
@@ -28,6 +28,13 @@ pub(super) fn create_anthropic_message_request(llm_messages: Vec<LLMMessage>) ->
|
|||||||
},
|
},
|
||||||
}]
|
}]
|
||||||
}
|
}
|
||||||
|
LLMMessageContent::File(file_details) => {
|
||||||
|
tracing::warn!(
|
||||||
|
"The Anthropic provider's library does not support file/document content. This file message ({}) will be skipped.",
|
||||||
|
file_details.filename(),
|
||||||
|
);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
let message = Message { role, content };
|
let message = Message { role, content };
|
||||||
|
|||||||
@@ -84,7 +84,7 @@ impl Default for TextGenerationConfig {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn default_text_model_id() -> String {
|
fn default_text_model_id() -> String {
|
||||||
"gpt-5.2".to_owned()
|
"gpt-5.4".to_owned()
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone, Serialize, Deserialize, Default)]
|
#[derive(Debug, Clone, Serialize, Deserialize, Default)]
|
||||||
|
|||||||
@@ -5,14 +5,14 @@ use async_openai::{
|
|||||||
config::OpenAIConfig,
|
config::OpenAIConfig,
|
||||||
types::{
|
types::{
|
||||||
audio::{AudioInput, CreateSpeechRequestArgs, CreateTranscriptionRequestArgs},
|
audio::{AudioInput, CreateSpeechRequestArgs, CreateTranscriptionRequestArgs},
|
||||||
|
images::{
|
||||||
|
CreateImageEditRequestArgs, CreateImageRequestArgs, Image, ImageInput, ImageModel,
|
||||||
|
ImageResponseFormat,
|
||||||
|
},
|
||||||
responses::{
|
responses::{
|
||||||
CodeInterpreterContainerAuto, CodeInterpreterTool, CodeInterpreterToolContainer,
|
CodeInterpreterContainerAuto, CodeInterpreterTool, CodeInterpreterToolContainer,
|
||||||
CreateResponseArgs, OutputItem, OutputMessageContent, Tool, WebSearchTool,
|
CreateResponseArgs, OutputItem, OutputMessageContent, Tool, WebSearchTool,
|
||||||
},
|
},
|
||||||
images::{
|
|
||||||
CreateImageEditRequestArgs, CreateImageRequestArgs,
|
|
||||||
Image, ImageInput, ImageModel, ImageResponseFormat,
|
|
||||||
},
|
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -32,8 +32,8 @@ use crate::{
|
|||||||
agent::{
|
agent::{
|
||||||
AgentPurpose,
|
AgentPurpose,
|
||||||
provider::entity::{
|
provider::entity::{
|
||||||
ImageEditResult, ImageGenerationResult, ImageSource, PingResult,
|
ImageEditResult, ImageGenerationResult, ImageSource, PingResult, TextToSpeechParams,
|
||||||
TextToSpeechParams, TextToSpeechResult,
|
TextToSpeechResult,
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
strings,
|
strings,
|
||||||
@@ -67,6 +67,7 @@ impl ControllerTrait for Controller {
|
|||||||
|
|
||||||
let messages = vec![LLMMessage {
|
let messages = vec![LLMMessage {
|
||||||
author: LLMAuthor::User,
|
author: LLMAuthor::User,
|
||||||
|
sender_id: None,
|
||||||
content: LLMMessageContent::Text("Hello!".to_string()),
|
content: LLMMessageContent::Text("Hello!".to_string()),
|
||||||
timestamp: chrono::Utc::now(),
|
timestamp: chrono::Utc::now(),
|
||||||
}];
|
}];
|
||||||
@@ -104,6 +105,7 @@ impl ControllerTrait for Controller {
|
|||||||
} else {
|
} else {
|
||||||
Some(LLMMessage {
|
Some(LLMMessage {
|
||||||
author: LLMAuthor::Prompt,
|
author: LLMAuthor::Prompt,
|
||||||
|
sender_id: None,
|
||||||
content: LLMMessageContent::Text(prompt_text),
|
content: LLMMessageContent::Text(prompt_text),
|
||||||
timestamp: chrono::Utc::now(),
|
timestamp: chrono::Utc::now(),
|
||||||
})
|
})
|
||||||
@@ -129,7 +131,8 @@ impl ControllerTrait for Controller {
|
|||||||
conversation_messages.insert(0, prompt_message);
|
conversation_messages.insert(0, prompt_message);
|
||||||
}
|
}
|
||||||
|
|
||||||
let input = super::utils::convert_llm_messages_to_openai_response_input(conversation_messages);
|
let input =
|
||||||
|
super::utils::convert_llm_messages_to_openai_response_input(conversation_messages);
|
||||||
|
|
||||||
let messages_count = match &input {
|
let messages_count = match &input {
|
||||||
async_openai::types::responses::InputParam::Items(items) => items.len(),
|
async_openai::types::responses::InputParam::Items(items) => items.len(),
|
||||||
@@ -182,10 +185,7 @@ impl ControllerTrait for Controller {
|
|||||||
|
|
||||||
let response = self.client.responses().create(request).await?;
|
let response = self.client.responses().create(request).await?;
|
||||||
|
|
||||||
tracing::trace!(
|
tracing::trace!(?response, "Got response from the OpenAI response API");
|
||||||
?response,
|
|
||||||
"Got response from the OpenAI response API"
|
|
||||||
);
|
|
||||||
|
|
||||||
for item in response.output {
|
for item in response.output {
|
||||||
if let OutputItem::Message(message) = item {
|
if let OutputItem::Message(message) = item {
|
||||||
@@ -271,9 +271,7 @@ impl ControllerTrait for Controller {
|
|||||||
ImageModel::GptImage1 => ImageModel::GptImage1Mini,
|
ImageModel::GptImage1 => ImageModel::GptImage1Mini,
|
||||||
ImageModel::GptImage1dot5 => ImageModel::GptImage1Mini,
|
ImageModel::GptImage1dot5 => ImageModel::GptImage1Mini,
|
||||||
ImageModel::GptImage1Mini => ImageModel::GptImage1Mini,
|
ImageModel::GptImage1Mini => ImageModel::GptImage1Mini,
|
||||||
ImageModel::Other(_) => {
|
ImageModel::Other(_) => ImageModel::DallE2,
|
||||||
ImageModel::DallE2
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
original_model
|
original_model
|
||||||
@@ -408,9 +406,15 @@ impl ControllerTrait for Controller {
|
|||||||
}
|
}
|
||||||
|
|
||||||
let dalle2_size = match image_generation_config.size {
|
let dalle2_size = match image_generation_config.size {
|
||||||
Some(async_openai::types::images::ImageSize::S256x256) => Some(async_openai::types::images::ImageSize::S256x256),
|
Some(async_openai::types::images::ImageSize::S256x256) => {
|
||||||
Some(async_openai::types::images::ImageSize::S512x512) => Some(async_openai::types::images::ImageSize::S512x512),
|
Some(async_openai::types::images::ImageSize::S256x256)
|
||||||
Some(async_openai::types::images::ImageSize::S1024x1024) => Some(async_openai::types::images::ImageSize::S1024x1024),
|
}
|
||||||
|
Some(async_openai::types::images::ImageSize::S512x512) => {
|
||||||
|
Some(async_openai::types::images::ImageSize::S512x512)
|
||||||
|
}
|
||||||
|
Some(async_openai::types::images::ImageSize::S1024x1024) => {
|
||||||
|
Some(async_openai::types::images::ImageSize::S1024x1024)
|
||||||
|
}
|
||||||
_ => None,
|
_ => None,
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -419,12 +423,8 @@ impl ControllerTrait for Controller {
|
|||||||
.map_err(|err| anyhow::anyhow!(err))?;
|
.map_err(|err| anyhow::anyhow!(err))?;
|
||||||
|
|
||||||
let response_format = match model.clone() {
|
let response_format = match model.clone() {
|
||||||
ImageModel::DallE2 => {
|
ImageModel::DallE2 => Some(ImageResponseFormat::B64Json),
|
||||||
Some(ImageResponseFormat::B64Json)
|
ImageModel::DallE3 => Some(ImageResponseFormat::B64Json),
|
||||||
}
|
|
||||||
ImageModel::DallE3 => {
|
|
||||||
Some(ImageResponseFormat::B64Json)
|
|
||||||
}
|
|
||||||
// gpt-image-1 only outputs base64 and we don't need to specify the response format.
|
// gpt-image-1 only outputs base64 and we don't need to specify the response format.
|
||||||
// In fact, specifying the response format results in an error.
|
// In fact, specifying the response format results in an error.
|
||||||
ImageModel::GptImage1 => None,
|
ImageModel::GptImage1 => None,
|
||||||
|
|||||||
@@ -20,12 +20,12 @@ pub const OPENAI_IMAGE_MODEL_GPT_IMAGE_1_DOT_5: &str = "gpt-image-1.5";
|
|||||||
|
|
||||||
pub fn create_controller_from_yaml_value_config(
|
pub fn create_controller_from_yaml_value_config(
|
||||||
agent_id: &str,
|
agent_id: &str,
|
||||||
config: serde_yaml::Value,
|
config: serde_yaml_ng::Value,
|
||||||
) -> AgentInstantiationResult<ControllerType> {
|
) -> AgentInstantiationResult<ControllerType> {
|
||||||
let config = match &config {
|
let config = match &config {
|
||||||
serde_yaml::Value::Mapping(_) => {
|
serde_yaml_ng::Value::Mapping(_) => {
|
||||||
let config: Config =
|
let config: Config =
|
||||||
serde_yaml::from_value(config).map_err(AgentInstantiationError::Yaml)?;
|
serde_yaml_ng::from_value(config).map_err(AgentInstantiationError::Yaml)?;
|
||||||
|
|
||||||
config
|
config
|
||||||
.validate()
|
.validate()
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
use async_openai::types::responses::{
|
use async_openai::types::responses::{
|
||||||
EasyInputContent, EasyInputMessage, ImageDetail, InputContent, InputImageContent, InputItem,
|
EasyInputContent, EasyInputMessage, ImageDetail, InputContent, InputFileArgs,
|
||||||
InputParam, MessageType, Role,
|
InputImageContent, InputItem, InputParam, MessageType, Role,
|
||||||
};
|
};
|
||||||
|
|
||||||
use crate::conversation::llm::{
|
use crate::conversation::llm::{
|
||||||
@@ -35,12 +35,28 @@ pub fn convert_llm_messages_to_openai_response_input(
|
|||||||
file_id: None,
|
file_id: None,
|
||||||
})])
|
})])
|
||||||
}
|
}
|
||||||
|
LLMMessageContent::File(file_details) => {
|
||||||
|
let file_data = format!(
|
||||||
|
"data:{};base64,{}",
|
||||||
|
file_details.mime,
|
||||||
|
base64_encode(&file_details.data)
|
||||||
|
);
|
||||||
|
|
||||||
|
let file_content = InputFileArgs::default()
|
||||||
|
.file_data(file_data)
|
||||||
|
.filename(file_details.filename())
|
||||||
|
.build()
|
||||||
|
.expect("Failed to build InputFileContent");
|
||||||
|
|
||||||
|
EasyInputContent::ContentList(vec![InputContent::InputFile(file_content)])
|
||||||
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
items.push(InputItem::EasyMessage(EasyInputMessage {
|
items.push(InputItem::EasyMessage(EasyInputMessage {
|
||||||
r#type: MessageType::Message,
|
r#type: MessageType::Message,
|
||||||
role,
|
role,
|
||||||
content,
|
content,
|
||||||
|
phase: None,
|
||||||
}));
|
}));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -162,13 +162,14 @@ impl TryInto<OpenAITextToSpeechConfig> for TextToSpeechConfig {
|
|||||||
type Error = String;
|
type Error = String;
|
||||||
|
|
||||||
fn try_into(self) -> Result<OpenAITextToSpeechConfig, Self::Error> {
|
fn try_into(self) -> Result<OpenAITextToSpeechConfig, Self::Error> {
|
||||||
let model_id = convert_string_to_enum::<async_openai::types::audio::SpeechModel>(&self.model_id)?;
|
let model_id =
|
||||||
|
convert_string_to_enum::<async_openai::types::audio::SpeechModel>(&self.model_id)?;
|
||||||
|
|
||||||
let voice = convert_string_to_enum::<async_openai::types::audio::Voice>(&self.voice)?;
|
let voice = convert_string_to_enum::<async_openai::types::audio::Voice>(&self.voice)?;
|
||||||
|
|
||||||
let response_format = convert_string_to_enum::<async_openai::types::audio::SpeechResponseFormat>(
|
let response_format = convert_string_to_enum::<
|
||||||
&self.response_format,
|
async_openai::types::audio::SpeechResponseFormat,
|
||||||
)?;
|
>(&self.response_format)?;
|
||||||
|
|
||||||
Ok(OpenAITextToSpeechConfig {
|
Ok(OpenAITextToSpeechConfig {
|
||||||
model_id,
|
model_id,
|
||||||
@@ -225,25 +226,25 @@ impl TryInto<OpenAIImageGenerationConfig> for ImageGenerationConfig {
|
|||||||
|
|
||||||
fn try_into(self) -> Result<OpenAIImageGenerationConfig, Self::Error> {
|
fn try_into(self) -> Result<OpenAIImageGenerationConfig, Self::Error> {
|
||||||
let size = if let Some(size) = &self.size {
|
let size = if let Some(size) = &self.size {
|
||||||
Some(convert_string_to_enum::<async_openai::types::images::ImageSize>(
|
Some(convert_string_to_enum::<
|
||||||
size,
|
async_openai::types::images::ImageSize,
|
||||||
)?)
|
>(size)?)
|
||||||
} else {
|
} else {
|
||||||
None
|
None
|
||||||
};
|
};
|
||||||
|
|
||||||
let style = if let Some(style) = &self.style {
|
let style = if let Some(style) = &self.style {
|
||||||
Some(convert_string_to_enum::<async_openai::types::images::ImageStyle>(
|
Some(convert_string_to_enum::<
|
||||||
style,
|
async_openai::types::images::ImageStyle,
|
||||||
)?)
|
>(style)?)
|
||||||
} else {
|
} else {
|
||||||
None
|
None
|
||||||
};
|
};
|
||||||
|
|
||||||
let quality = if let Some(quality) = &self.quality {
|
let quality = if let Some(quality) = &self.quality {
|
||||||
Some(convert_string_to_enum::<async_openai::types::images::ImageQuality>(
|
Some(convert_string_to_enum::<
|
||||||
quality,
|
async_openai::types::images::ImageQuality,
|
||||||
)?)
|
>(quality)?)
|
||||||
} else {
|
} else {
|
||||||
None
|
None
|
||||||
};
|
};
|
||||||
|
|||||||
@@ -64,6 +64,7 @@ impl ControllerTrait for Controller {
|
|||||||
|
|
||||||
let messages = vec![LLMMessage {
|
let messages = vec![LLMMessage {
|
||||||
author: LLMAuthor::User,
|
author: LLMAuthor::User,
|
||||||
|
sender_id: None,
|
||||||
content: LLMMessageContent::Text("Hello!".to_string()),
|
content: LLMMessageContent::Text("Hello!".to_string()),
|
||||||
timestamp: chrono::Utc::now(),
|
timestamp: chrono::Utc::now(),
|
||||||
}];
|
}];
|
||||||
@@ -101,6 +102,7 @@ impl ControllerTrait for Controller {
|
|||||||
} else {
|
} else {
|
||||||
Some(LLMMessage {
|
Some(LLMMessage {
|
||||||
author: LLMAuthor::Prompt,
|
author: LLMAuthor::Prompt,
|
||||||
|
sender_id: None,
|
||||||
content: LLMMessageContent::Text(prompt_text),
|
content: LLMMessageContent::Text(prompt_text),
|
||||||
timestamp: chrono::Utc::now(),
|
timestamp: chrono::Utc::now(),
|
||||||
})
|
})
|
||||||
|
|||||||
@@ -26,12 +26,12 @@ use super::controller::ControllerType;
|
|||||||
|
|
||||||
pub fn create_controller_from_yaml_value_config(
|
pub fn create_controller_from_yaml_value_config(
|
||||||
agent_id: &str,
|
agent_id: &str,
|
||||||
config: serde_yaml::Value,
|
config: serde_yaml_ng::Value,
|
||||||
) -> AgentInstantiationResult<ControllerType> {
|
) -> AgentInstantiationResult<ControllerType> {
|
||||||
let config = match &config {
|
let config = match &config {
|
||||||
serde_yaml::Value::Mapping(_) => {
|
serde_yaml_ng::Value::Mapping(_) => {
|
||||||
let config: Config =
|
let config: Config =
|
||||||
serde_yaml::from_value(config).map_err(AgentInstantiationError::Yaml)?;
|
serde_yaml_ng::from_value(config).map_err(AgentInstantiationError::Yaml)?;
|
||||||
|
|
||||||
config
|
config
|
||||||
.validate()
|
.validate()
|
||||||
|
|||||||
@@ -40,6 +40,12 @@ fn convert_llm_message_to_openai_message(llm_message: LLMMessage) -> Option<Mess
|
|||||||
);
|
);
|
||||||
None
|
None
|
||||||
}
|
}
|
||||||
|
LLMMessageContent::File(_file_details) => {
|
||||||
|
tracing::warn!(
|
||||||
|
"The OpenAI-compat provider's library does not support file content. This file message will be skipped."
|
||||||
|
);
|
||||||
|
None
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -4,10 +4,10 @@ use std::{future::Future, pin::Pin};
|
|||||||
|
|
||||||
use mxlink::matrix_sdk::Room;
|
use mxlink::matrix_sdk::Room;
|
||||||
use mxlink::matrix_sdk::media::{MediaFormat, MediaRequestParameters};
|
use mxlink::matrix_sdk::media::{MediaFormat, MediaRequestParameters};
|
||||||
|
use mxlink::matrix_sdk::ruma::api::client::profile::{AvatarUrl, DisplayName};
|
||||||
use mxlink::matrix_sdk::ruma::{
|
use mxlink::matrix_sdk::ruma::{
|
||||||
MilliSecondsSinceUnixEpoch, OwnedUserId, events::room::MediaSource,
|
MilliSecondsSinceUnixEpoch, OwnedUserId, events::room::MediaSource,
|
||||||
};
|
};
|
||||||
use mxlink::matrix_sdk::ruma::api::client::profile::{AvatarUrl, DisplayName};
|
|
||||||
|
|
||||||
use mxlink::{
|
use mxlink::{
|
||||||
InitConfig, LoginConfig, LoginCredentials, LoginEncryption, MatrixLink, PersistenceConfig,
|
InitConfig, LoginConfig, LoginCredentials, LoginEncryption, MatrixLink, PersistenceConfig,
|
||||||
@@ -25,7 +25,7 @@ use crate::agent::Manager as AgentManager;
|
|||||||
use crate::entity::catch_up_marker::{
|
use crate::entity::catch_up_marker::{
|
||||||
CatchUpMarker, CatchUpMarkerManager, DelayedCatchUpMarkerManager,
|
CatchUpMarker, CatchUpMarkerManager, DelayedCatchUpMarkerManager,
|
||||||
};
|
};
|
||||||
use crate::entity::cfg::{Avatar, Config};
|
use crate::entity::cfg::{Avatar, Config, ConfigUserAuth};
|
||||||
use crate::entity::globalconfig::{GlobalConfig, GlobalConfigurationManager};
|
use crate::entity::globalconfig::{GlobalConfig, GlobalConfigurationManager};
|
||||||
use crate::entity::roomconfig::{RoomConfig, RoomConfigurationManager};
|
use crate::entity::roomconfig::{RoomConfig, RoomConfigurationManager};
|
||||||
|
|
||||||
@@ -395,10 +395,22 @@ async fn create_matrix_link(config: &Config) -> anyhow::Result<MatrixLink> {
|
|||||||
let session_encryption_key = config.persistence.session_encryption_key()?;
|
let session_encryption_key = config.persistence.session_encryption_key()?;
|
||||||
let db_dir_path: std::path::PathBuf = config.persistence.db_dir_path()?;
|
let db_dir_path: std::path::PathBuf = config.persistence.db_dir_path()?;
|
||||||
|
|
||||||
let login_creds = LoginCredentials::UserPassword(
|
let user_auth = config.user.auth_config(&config.homeserver.server_name)?;
|
||||||
config.user.mxid_localpart.to_owned(),
|
|
||||||
config.user.password.to_owned(),
|
let login_creds = match user_auth {
|
||||||
);
|
ConfigUserAuth::UserPassword { username, password } => {
|
||||||
|
LoginCredentials::UserPassword(username, password)
|
||||||
|
}
|
||||||
|
ConfigUserAuth::AccessToken {
|
||||||
|
user_id,
|
||||||
|
device_id,
|
||||||
|
access_token,
|
||||||
|
} => LoginCredentials::AccessToken {
|
||||||
|
user_id,
|
||||||
|
device_id,
|
||||||
|
access_token,
|
||||||
|
},
|
||||||
|
};
|
||||||
|
|
||||||
let login_encryption = LoginEncryption::new(
|
let login_encryption = LoginEncryption::new(
|
||||||
config.user.encryption.recovery_passphrase.clone(),
|
config.user.encryption.recovery_passphrase.clone(),
|
||||||
|
|||||||
@@ -21,7 +21,7 @@ pub fn load() -> anyhow::Result<Config> {
|
|||||||
}
|
}
|
||||||
|
|
||||||
let config_str = std::fs::read_to_string(config_file_path)?;
|
let config_str = std::fs::read_to_string(config_file_path)?;
|
||||||
let mut config: Config = serde_yaml::from_str(&config_str)?;
|
let mut config: Config = serde_yaml_ng::from_str(&config_str)?;
|
||||||
|
|
||||||
// Allow environment variables to override some configuration keys
|
// Allow environment variables to override some configuration keys
|
||||||
for (key, value) in env::vars() {
|
for (key, value) in env::vars() {
|
||||||
@@ -29,7 +29,15 @@ pub fn load() -> anyhow::Result<Config> {
|
|||||||
cfg_env::BAIBOT_HOMESERVER_SERVER_NAME => config.homeserver.server_name = value,
|
cfg_env::BAIBOT_HOMESERVER_SERVER_NAME => config.homeserver.server_name = value,
|
||||||
cfg_env::BAIBOT_HOMESERVER_URL => config.homeserver.url = value,
|
cfg_env::BAIBOT_HOMESERVER_URL => config.homeserver.url = value,
|
||||||
cfg_env::BAIBOT_USER_MXID_LOCALPART => config.user.mxid_localpart = value,
|
cfg_env::BAIBOT_USER_MXID_LOCALPART => config.user.mxid_localpart = value,
|
||||||
cfg_env::BAIBOT_USER_PASSWORD => config.user.password = value,
|
cfg_env::BAIBOT_USER_PASSWORD => {
|
||||||
|
config.user.password = optional_non_empty(value);
|
||||||
|
}
|
||||||
|
cfg_env::BAIBOT_USER_ACCESS_TOKEN => {
|
||||||
|
config.user.access_token = optional_non_empty(value);
|
||||||
|
}
|
||||||
|
cfg_env::BAIBOT_USER_DEVICE_ID => {
|
||||||
|
config.user.device_id = optional_non_empty(value);
|
||||||
|
}
|
||||||
cfg_env::BAIBOT_USER_ENCRYPTION_RECOVERY_PASSPHRASE => {
|
cfg_env::BAIBOT_USER_ENCRYPTION_RECOVERY_PASSPHRASE => {
|
||||||
config.user.encryption.recovery_passphrase = Some(value);
|
config.user.encryption.recovery_passphrase = Some(value);
|
||||||
}
|
}
|
||||||
@@ -120,3 +128,7 @@ pub fn load() -> anyhow::Result<Config> {
|
|||||||
|
|
||||||
Ok(config)
|
Ok(config)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn optional_non_empty(value: String) -> Option<String> {
|
||||||
|
if value.is_empty() { None } else { Some(value) }
|
||||||
|
}
|
||||||
|
|||||||
@@ -28,18 +28,18 @@ pub async fn handle_set(
|
|||||||
message_context: &MessageContext,
|
message_context: &MessageContext,
|
||||||
patterns: &Option<Vec<String>>,
|
patterns: &Option<Vec<String>>,
|
||||||
) -> anyhow::Result<()> {
|
) -> anyhow::Result<()> {
|
||||||
if let Some(patterns) = patterns {
|
if let Some(patterns) = patterns
|
||||||
if let Err(err) = mxidwc::parse_patterns_vector(patterns) {
|
&& let Err(err) = mxidwc::parse_patterns_vector(patterns)
|
||||||
bot.messaging()
|
{
|
||||||
.send_error_markdown_no_fail(
|
bot.messaging()
|
||||||
message_context.room(),
|
.send_error_markdown_no_fail(
|
||||||
&strings::access::failed_to_parse_patterns(&err.to_string()),
|
message_context.room(),
|
||||||
MessageResponseType::Reply(message_context.thread_info().root_event_id.clone()),
|
&strings::access::failed_to_parse_patterns(&err.to_string()),
|
||||||
)
|
MessageResponseType::Reply(message_context.thread_info().root_event_id.clone()),
|
||||||
.await;
|
)
|
||||||
|
.await;
|
||||||
|
|
||||||
return Ok(());
|
return Ok(());
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
let mut global_config_manager_guard = bot.global_config_manager().lock().await;
|
let mut global_config_manager_guard = bot.global_config_manager().lock().await;
|
||||||
|
|||||||
@@ -24,18 +24,18 @@ pub async fn handle_set(
|
|||||||
message_context: &MessageContext,
|
message_context: &MessageContext,
|
||||||
patterns: &Option<Vec<String>>,
|
patterns: &Option<Vec<String>>,
|
||||||
) -> anyhow::Result<()> {
|
) -> anyhow::Result<()> {
|
||||||
if let Some(patterns) = patterns {
|
if let Some(patterns) = patterns
|
||||||
if let Err(err) = mxidwc::parse_patterns_vector(patterns) {
|
&& let Err(err) = mxidwc::parse_patterns_vector(patterns)
|
||||||
bot.messaging()
|
{
|
||||||
.send_error_markdown_no_fail(
|
bot.messaging()
|
||||||
message_context.room(),
|
.send_error_markdown_no_fail(
|
||||||
&strings::access::failed_to_parse_patterns(&err.to_string()),
|
message_context.room(),
|
||||||
MessageResponseType::Reply(message_context.thread_info().root_event_id.clone()),
|
&strings::access::failed_to_parse_patterns(&err.to_string()),
|
||||||
)
|
MessageResponseType::Reply(message_context.thread_info().root_event_id.clone()),
|
||||||
.await;
|
)
|
||||||
|
.await;
|
||||||
|
|
||||||
return Ok(());
|
return Ok(());
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
let mut global_config_manager_guard = bot.global_config_manager().lock().await;
|
let mut global_config_manager_guard = bot.global_config_manager().lock().await;
|
||||||
|
|||||||
@@ -15,7 +15,7 @@ use crate::{Bot, entity::MessageContext};
|
|||||||
|
|
||||||
struct ParsedAgentConfig {
|
struct ParsedAgentConfig {
|
||||||
agent: AgentInstance,
|
agent: AgentInstance,
|
||||||
config: serde_yaml::Value,
|
config: serde_yaml_ng::Value,
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn handle_room_local(
|
pub async fn handle_room_local(
|
||||||
@@ -250,7 +250,7 @@ async fn send_guide(
|
|||||||
provider: &AgentProvider,
|
provider: &AgentProvider,
|
||||||
) -> anyhow::Result<()> {
|
) -> anyhow::Result<()> {
|
||||||
let sample_config = crate::agent::default_config_for_provider(provider);
|
let sample_config = crate::agent::default_config_for_provider(provider);
|
||||||
let sample_config_pretty_yaml = serde_yaml::to_string(&sample_config)?;
|
let sample_config_pretty_yaml = serde_yaml_ng::to_string(&sample_config)?;
|
||||||
|
|
||||||
bot.messaging()
|
bot.messaging()
|
||||||
.send_text_markdown_no_fail(
|
.send_text_markdown_no_fail(
|
||||||
@@ -263,7 +263,7 @@ async fn send_guide(
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
fn parse_from_message_to_yaml_value(text: &str) -> Result<serde_yaml::Value, String> {
|
fn parse_from_message_to_yaml_value(text: &str) -> Result<serde_yaml_ng::Value, String> {
|
||||||
let mut text = text.trim();
|
let mut text = text.trim();
|
||||||
|
|
||||||
if text.starts_with("```") {
|
if text.starts_with("```") {
|
||||||
@@ -274,10 +274,10 @@ fn parse_from_message_to_yaml_value(text: &str) -> Result<serde_yaml::Value, Str
|
|||||||
text = text.trim_end_matches("```");
|
text = text.trim_end_matches("```");
|
||||||
}
|
}
|
||||||
|
|
||||||
let config: serde_yaml::Value = serde_yaml::from_str(text).map_err(|e| e.to_string())?;
|
let config: serde_yaml_ng::Value = serde_yaml_ng::from_str(text).map_err(|e| e.to_string())?;
|
||||||
|
|
||||||
match config {
|
match config {
|
||||||
serde_yaml::Value::Mapping(_) => {}
|
serde_yaml_ng::Value::Mapping(_) => {}
|
||||||
_ => {
|
_ => {
|
||||||
return Err("Not a valid YAML hashmap".to_owned());
|
return Err("Not a valid YAML hashmap".to_owned());
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -2,12 +2,12 @@
|
|||||||
fn agent_config_parsing_works() {
|
fn agent_config_parsing_works() {
|
||||||
struct TestCase {
|
struct TestCase {
|
||||||
input: String,
|
input: String,
|
||||||
expected: Option<serde_yaml::Value>,
|
expected: Option<serde_yaml_ng::Value>,
|
||||||
}
|
}
|
||||||
|
|
||||||
let provider = crate::agent::AgentProvider::OpenAI;
|
let provider = crate::agent::AgentProvider::OpenAI;
|
||||||
let sample_config = crate::agent::default_config_for_provider(&provider);
|
let sample_config = crate::agent::default_config_for_provider(&provider);
|
||||||
let sample_config_pretty_yaml = serde_yaml::to_string(&sample_config).unwrap();
|
let sample_config_pretty_yaml = serde_yaml_ng::to_string(&sample_config).unwrap();
|
||||||
|
|
||||||
let test_cases = vec![
|
let test_cases = vec![
|
||||||
// Invalid input
|
// Invalid input
|
||||||
|
|||||||
@@ -64,7 +64,7 @@ pub async fn handle(
|
|||||||
PublicIdentifier::Static(_) => {}
|
PublicIdentifier::Static(_) => {}
|
||||||
};
|
};
|
||||||
|
|
||||||
let config_yaml_pretty = serde_yaml::to_string(&agent.definition().config)?;
|
let config_yaml_pretty = serde_yaml_ng::to_string(&agent.definition().config)?;
|
||||||
|
|
||||||
bot.messaging()
|
bot.messaging()
|
||||||
.send_text_markdown_no_fail(
|
.send_text_markdown_no_fail(
|
||||||
|
|||||||
@@ -3,7 +3,8 @@ use crate::{
|
|||||||
entity::roomconfig::{
|
entity::roomconfig::{
|
||||||
SpeechToTextFlowType, SpeechToTextMessageTypeForNonThreadedOnlyTranscribedMessages,
|
SpeechToTextFlowType, SpeechToTextMessageTypeForNonThreadedOnlyTranscribedMessages,
|
||||||
TextGenerationAutoUsage, TextGenerationPrefixRequirementType,
|
TextGenerationAutoUsage, TextGenerationPrefixRequirementType,
|
||||||
TextToSpeechBotMessagesFlowType, TextToSpeechUserMessagesFlowType,
|
TextGenerationSenderContextMode, TextToSpeechBotMessagesFlowType,
|
||||||
|
TextToSpeechUserMessagesFlowType,
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -48,6 +49,9 @@ pub enum ConfigTextGenerationSettingRelatedControllerType {
|
|||||||
|
|
||||||
GetTemperatureOverride,
|
GetTemperatureOverride,
|
||||||
SetTemperatureOverride(Option<f32>),
|
SetTemperatureOverride(Option<f32>),
|
||||||
|
|
||||||
|
GetSenderContextMode,
|
||||||
|
SetSenderContextMode(Option<TextGenerationSenderContextMode>),
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, PartialEq)]
|
#[derive(Debug, PartialEq)]
|
||||||
|
|||||||
@@ -163,6 +163,26 @@ fn determine_controller() {
|
|||||||
),
|
),
|
||||||
)),
|
)),
|
||||||
},
|
},
|
||||||
|
TestCase {
|
||||||
|
name: "per-room text-generation/sender-context-mode getter",
|
||||||
|
input: "room text-generation sender-context-mode",
|
||||||
|
expected: super::ControllerType::Config(controller_type::ConfigControllerType::SettingsRelated(
|
||||||
|
controller_type::SettingsStorageSource::Room,
|
||||||
|
controller_type::ConfigSettingRelatedControllerType::TextGeneration(
|
||||||
|
controller_type::ConfigTextGenerationSettingRelatedControllerType::GetSenderContextMode,
|
||||||
|
),
|
||||||
|
)),
|
||||||
|
},
|
||||||
|
TestCase {
|
||||||
|
name: "global text-generation/sender-context-mode getter",
|
||||||
|
input: "global text-generation sender-context-mode",
|
||||||
|
expected: super::ControllerType::Config(controller_type::ConfigControllerType::SettingsRelated(
|
||||||
|
controller_type::SettingsStorageSource::Global,
|
||||||
|
controller_type::ConfigSettingRelatedControllerType::TextGeneration(
|
||||||
|
controller_type::ConfigTextGenerationSettingRelatedControllerType::GetSenderContextMode,
|
||||||
|
),
|
||||||
|
)),
|
||||||
|
},
|
||||||
TestCase {
|
TestCase {
|
||||||
name: "per-room text-to-speech/speed-override getter",
|
name: "per-room text-to-speech/speed-override getter",
|
||||||
input: "room text-to-speech speed-override",
|
input: "room text-to-speech speed-override",
|
||||||
|
|||||||
@@ -3,7 +3,10 @@ mod tests;
|
|||||||
|
|
||||||
use crate::{
|
use crate::{
|
||||||
controller::ControllerType,
|
controller::ControllerType,
|
||||||
entity::roomconfig::{TextGenerationAutoUsage, TextGenerationPrefixRequirementType},
|
entity::roomconfig::{
|
||||||
|
TextGenerationAutoUsage, TextGenerationPrefixRequirementType,
|
||||||
|
TextGenerationSenderContextMode,
|
||||||
|
},
|
||||||
strings,
|
strings,
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -197,5 +200,43 @@ pub(super) fn determine(
|
|||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if let Some(remaining_text) = text.strip_prefix("sender-context-mode") {
|
||||||
|
let remaining_text = remaining_text.trim();
|
||||||
|
|
||||||
|
if !remaining_text.is_empty() {
|
||||||
|
return Err(ControllerType::Error(
|
||||||
|
strings::cfg::configuration_getter_used_with_extra_text(
|
||||||
|
"sender-context-mode",
|
||||||
|
remaining_text,
|
||||||
|
)
|
||||||
|
.to_owned(),
|
||||||
|
));
|
||||||
|
}
|
||||||
|
|
||||||
|
return Ok(ConfigTextGenerationSettingRelatedControllerType::GetSenderContextMode);
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(value_string) = text.strip_prefix("set-sender-context-mode") {
|
||||||
|
let value_string = value_string.trim().to_owned();
|
||||||
|
let value_choice = if value_string.is_empty() {
|
||||||
|
None
|
||||||
|
} else {
|
||||||
|
let value_choice =
|
||||||
|
TextGenerationSenderContextMode::from_str(&value_string.to_lowercase());
|
||||||
|
|
||||||
|
if value_choice.is_none() {
|
||||||
|
return Err(ControllerType::Error(
|
||||||
|
strings::cfg::configuration_value_unrecognized(&value_string).to_owned(),
|
||||||
|
));
|
||||||
|
}
|
||||||
|
|
||||||
|
value_choice
|
||||||
|
};
|
||||||
|
|
||||||
|
return Ok(
|
||||||
|
ConfigTextGenerationSettingRelatedControllerType::SetSenderContextMode(value_choice),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
Err(ControllerType::Unknown)
|
Err(ControllerType::Unknown)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -90,6 +90,74 @@ fn determine_controller_context_management() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn determine_controller_sender_context() {
|
||||||
|
use super::ConfigTextGenerationSettingRelatedControllerType;
|
||||||
|
use super::ControllerType;
|
||||||
|
use crate::entity::roomconfig::TextGenerationSenderContextMode;
|
||||||
|
|
||||||
|
struct TestCase {
|
||||||
|
name: &'static str,
|
||||||
|
input: &'static str,
|
||||||
|
expected: Result<ConfigTextGenerationSettingRelatedControllerType, ControllerType>,
|
||||||
|
}
|
||||||
|
|
||||||
|
let test_cases = vec![
|
||||||
|
TestCase {
|
||||||
|
name: "sender-context-mode getter ok",
|
||||||
|
input: "sender-context-mode",
|
||||||
|
expected: Ok(ConfigTextGenerationSettingRelatedControllerType::GetSenderContextMode),
|
||||||
|
},
|
||||||
|
TestCase {
|
||||||
|
name: "sender-context-mode getter extra args",
|
||||||
|
input: "sender-context-mode some values here",
|
||||||
|
expected: Err(ControllerType::Error(
|
||||||
|
crate::strings::cfg::configuration_getter_used_with_extra_text(
|
||||||
|
"sender-context-mode",
|
||||||
|
"some values here",
|
||||||
|
),
|
||||||
|
)),
|
||||||
|
},
|
||||||
|
TestCase {
|
||||||
|
name: "sender-context-mode setter matrix_user_id",
|
||||||
|
input: "set-sender-context-mode matrix_user_id",
|
||||||
|
expected: Ok(
|
||||||
|
ConfigTextGenerationSettingRelatedControllerType::SetSenderContextMode(Some(
|
||||||
|
TextGenerationSenderContextMode::MatrixUserId,
|
||||||
|
)),
|
||||||
|
),
|
||||||
|
},
|
||||||
|
TestCase {
|
||||||
|
name: "sender-context-mode setter uppercase",
|
||||||
|
input: "set-sender-context-mode MATRIX_USER_ID_AND_TIMESTAMP",
|
||||||
|
expected: Ok(
|
||||||
|
ConfigTextGenerationSettingRelatedControllerType::SetSenderContextMode(Some(
|
||||||
|
TextGenerationSenderContextMode::MatrixUserIdAndTimestamp,
|
||||||
|
)),
|
||||||
|
),
|
||||||
|
},
|
||||||
|
TestCase {
|
||||||
|
name: "sender-context-mode setter invalid",
|
||||||
|
input: "set-sender-context-mode non-Enum-Value",
|
||||||
|
expected: Err(ControllerType::Error(
|
||||||
|
crate::strings::cfg::configuration_value_unrecognized("non-Enum-Value"),
|
||||||
|
)),
|
||||||
|
},
|
||||||
|
TestCase {
|
||||||
|
name: "sender-context-mode unsetter",
|
||||||
|
input: "set-sender-context-mode",
|
||||||
|
expected: Ok(
|
||||||
|
ConfigTextGenerationSettingRelatedControllerType::SetSenderContextMode(None),
|
||||||
|
),
|
||||||
|
},
|
||||||
|
];
|
||||||
|
|
||||||
|
for test_case in test_cases {
|
||||||
|
let result = super::determine(test_case.input);
|
||||||
|
assert_eq!(result, test_case.expected, "Test case: {}", test_case.name);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn determine_controller_prefix_requirement_type() {
|
fn determine_controller_prefix_requirement_type() {
|
||||||
use super::ConfigTextGenerationSettingRelatedControllerType;
|
use super::ConfigTextGenerationSettingRelatedControllerType;
|
||||||
|
|||||||
@@ -39,18 +39,18 @@ async fn dispatch_config_related_handler(
|
|||||||
message_context: &MessageContext,
|
message_context: &MessageContext,
|
||||||
bot: &Bot,
|
bot: &Bot,
|
||||||
) -> anyhow::Result<()> {
|
) -> anyhow::Result<()> {
|
||||||
if let SettingsStorageSource::Global = config_type {
|
if let SettingsStorageSource::Global = config_type
|
||||||
if !message_context.sender_can_manage_global_config() {
|
&& !message_context.sender_can_manage_global_config()
|
||||||
bot.messaging()
|
{
|
||||||
.send_error_markdown_no_fail(
|
bot.messaging()
|
||||||
message_context.room(),
|
.send_error_markdown_no_fail(
|
||||||
strings::global_config::no_permissions_to_administrate(),
|
message_context.room(),
|
||||||
MessageResponseType::Reply(message_context.thread_info().root_event_id.clone()),
|
strings::global_config::no_permissions_to_administrate(),
|
||||||
)
|
MessageResponseType::Reply(message_context.thread_info().root_event_id.clone()),
|
||||||
.await;
|
)
|
||||||
return Ok(());
|
.await;
|
||||||
}
|
return Ok(());
|
||||||
};
|
}
|
||||||
|
|
||||||
let room_settings = match config_type {
|
let room_settings = match config_type {
|
||||||
SettingsStorageSource::Room => &message_context.room_config().settings,
|
SettingsStorageSource::Room => &message_context.room_config().settings,
|
||||||
|
|||||||
@@ -1,5 +1,6 @@
|
|||||||
use crate::entity::roomconfig::{
|
use crate::entity::roomconfig::{
|
||||||
RoomSettings, TextGenerationAutoUsage, TextGenerationPrefixRequirementType,
|
RoomSettings, TextGenerationAutoUsage, TextGenerationPrefixRequirementType,
|
||||||
|
TextGenerationSenderContextMode,
|
||||||
};
|
};
|
||||||
use crate::{Bot, entity::MessageContext};
|
use crate::{Bot, entity::MessageContext};
|
||||||
|
|
||||||
@@ -151,5 +152,38 @@ pub(super) async fn dispatch(
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
ConfigTextGenerationSettingRelatedControllerType::GetSenderContextMode => {
|
||||||
|
let value = &room_settings.text_generation.sender_context_mode;
|
||||||
|
setting_get::<TextGenerationSenderContextMode>(bot, message_context, value).await
|
||||||
|
}
|
||||||
|
ConfigTextGenerationSettingRelatedControllerType::SetSenderContextMode(value) => {
|
||||||
|
let value = value.to_owned();
|
||||||
|
|
||||||
|
let setter_callback = Box::new(move |room_settings: &mut RoomSettings| {
|
||||||
|
room_settings.text_generation.sender_context_mode = value;
|
||||||
|
});
|
||||||
|
|
||||||
|
match config_type {
|
||||||
|
SettingsStorageSource::Room => {
|
||||||
|
room_setting_set::<TextGenerationSenderContextMode>(
|
||||||
|
bot,
|
||||||
|
message_context,
|
||||||
|
&value,
|
||||||
|
setter_callback,
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
}
|
||||||
|
SettingsStorageSource::Global => {
|
||||||
|
global_setting_set::<TextGenerationSenderContextMode>(
|
||||||
|
bot,
|
||||||
|
message_context,
|
||||||
|
&value,
|
||||||
|
setter_callback,
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -7,7 +7,8 @@ use crate::{
|
|||||||
roomconfig::{
|
roomconfig::{
|
||||||
SpeechToTextFlowType, SpeechToTextMessageTypeForNonThreadedOnlyTranscribedMessages,
|
SpeechToTextFlowType, SpeechToTextMessageTypeForNonThreadedOnlyTranscribedMessages,
|
||||||
TextGenerationAutoUsage, TextGenerationPrefixRequirementType,
|
TextGenerationAutoUsage, TextGenerationPrefixRequirementType,
|
||||||
TextToSpeechBotMessagesFlowType, TextToSpeechUserMessagesFlowType,
|
TextGenerationSenderContextMode, TextToSpeechBotMessagesFlowType,
|
||||||
|
TextToSpeechUserMessagesFlowType,
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
strings,
|
strings,
|
||||||
@@ -233,6 +234,46 @@ fn build_section_text_generation(command_prefix: &str, bot_username: &str) -> St
|
|||||||
));
|
));
|
||||||
message.push_str("\n\n");
|
message.push_str("\n\n");
|
||||||
|
|
||||||
|
// Sender Context
|
||||||
|
|
||||||
|
message.push_str(&format!(
|
||||||
|
"#### {}",
|
||||||
|
strings::help::cfg::text_generation_sender_context_heading()
|
||||||
|
));
|
||||||
|
message.push_str("\n\n");
|
||||||
|
message.push_str(&strings::help::cfg::text_generation_sender_context_intro());
|
||||||
|
message.push('\n');
|
||||||
|
message.push_str(
|
||||||
|
&strings::help::cfg::the_following_configuration_values_are_recognized(
|
||||||
|
TextGenerationSenderContextMode::choices(),
|
||||||
|
),
|
||||||
|
);
|
||||||
|
message.push_str("\n\n");
|
||||||
|
message.push_str(&format!(
|
||||||
|
"- {}",
|
||||||
|
&strings::help::cfg::current_setting_show(
|
||||||
|
command_prefix,
|
||||||
|
"text-generation sender-context-mode"
|
||||||
|
)
|
||||||
|
));
|
||||||
|
message.push('\n');
|
||||||
|
message.push_str(&format!(
|
||||||
|
"- {}",
|
||||||
|
&strings::help::cfg::current_setting_set(
|
||||||
|
command_prefix,
|
||||||
|
"text-generation set-sender-context-mode VALUE"
|
||||||
|
)
|
||||||
|
));
|
||||||
|
message.push('\n');
|
||||||
|
message.push_str(&format!(
|
||||||
|
"- {}",
|
||||||
|
&strings::help::cfg::current_setting_unset(
|
||||||
|
command_prefix,
|
||||||
|
"text-generation set-sender-context-mode"
|
||||||
|
)
|
||||||
|
));
|
||||||
|
message.push_str("\n\n");
|
||||||
|
|
||||||
// Prompt override
|
// Prompt override
|
||||||
|
|
||||||
message.push_str(&format!(
|
message.push_str(&format!(
|
||||||
|
|||||||
@@ -359,6 +359,33 @@ async fn generate_text_generation_section(
|
|||||||
),
|
),
|
||||||
);
|
);
|
||||||
|
|
||||||
|
// Sender Context
|
||||||
|
|
||||||
|
let effective_sender_context = room_config_context.text_generation_sender_context_mode();
|
||||||
|
let room_config_sender_context = room_config_context
|
||||||
|
.room_config
|
||||||
|
.settings
|
||||||
|
.text_generation
|
||||||
|
.sender_context_mode;
|
||||||
|
let global_config_sender_context = room_config_context
|
||||||
|
.global_config
|
||||||
|
.fallback_room_settings
|
||||||
|
.text_generation
|
||||||
|
.sender_context_mode;
|
||||||
|
|
||||||
|
let sender_context_set_where = if room_config_sender_context.is_some() {
|
||||||
|
strings::cfg::status_badge_set_in_room_config()
|
||||||
|
} else if global_config_sender_context.is_some() {
|
||||||
|
strings::cfg::status_badge_set_in_global_config()
|
||||||
|
} else {
|
||||||
|
strings::cfg::status_badge_using_hardcoded_default()
|
||||||
|
};
|
||||||
|
|
||||||
|
message.push_str(&strings::cfg::status_text_generation_entry_sender_context(
|
||||||
|
effective_sender_context,
|
||||||
|
sender_context_set_where,
|
||||||
|
));
|
||||||
|
|
||||||
// Prompt override
|
// Prompt override
|
||||||
|
|
||||||
let text_agent_prompt = if let Some(text_generation_agent) = &text_generation_agent {
|
let text_agent_prompt = if let Some(text_generation_agent) = &text_generation_agent {
|
||||||
|
|||||||
@@ -15,7 +15,8 @@ use crate::conversation::matrix::MatrixMessageProcessingParams;
|
|||||||
use crate::entity::MessagePayload;
|
use crate::entity::MessagePayload;
|
||||||
use crate::entity::roomconfig::{
|
use crate::entity::roomconfig::{
|
||||||
SpeechToTextFlowType, SpeechToTextMessageTypeForNonThreadedOnlyTranscribedMessages,
|
SpeechToTextFlowType, SpeechToTextMessageTypeForNonThreadedOnlyTranscribedMessages,
|
||||||
TextToSpeechBotMessagesFlowType, TextToSpeechUserMessagesFlowType,
|
TextGenerationSenderContextMode, TextToSpeechBotMessagesFlowType,
|
||||||
|
TextToSpeechUserMessagesFlowType,
|
||||||
};
|
};
|
||||||
use crate::strings;
|
use crate::strings;
|
||||||
use crate::utils::text_to_speech::create_transcribed_message_text;
|
use crate::utils::text_to_speech::create_transcribed_message_text;
|
||||||
@@ -23,6 +24,7 @@ use crate::{
|
|||||||
Bot,
|
Bot,
|
||||||
conversation::{
|
conversation::{
|
||||||
create_llm_conversation_for_matrix_reply_chain, create_llm_conversation_for_matrix_thread,
|
create_llm_conversation_for_matrix_reply_chain, create_llm_conversation_for_matrix_thread,
|
||||||
|
llm::{Author, Conversation, MessageContent},
|
||||||
matrix::create_list_of_bot_user_prefixes_to_strip,
|
matrix::create_list_of_bot_user_prefixes_to_strip,
|
||||||
},
|
},
|
||||||
entity::MessageContext,
|
entity::MessageContext,
|
||||||
@@ -41,6 +43,8 @@ pub enum ChatCompletionControllerType {
|
|||||||
|
|
||||||
Image,
|
Image,
|
||||||
|
|
||||||
|
File,
|
||||||
|
|
||||||
ThreadMention,
|
ThreadMention,
|
||||||
ReplyMention,
|
ReplyMention,
|
||||||
}
|
}
|
||||||
@@ -419,7 +423,8 @@ async fn handle_stage_text_generation(
|
|||||||
| ChatCompletionControllerType::TextMention
|
| ChatCompletionControllerType::TextMention
|
||||||
| ChatCompletionControllerType::TextDirect
|
| ChatCompletionControllerType::TextDirect
|
||||||
| ChatCompletionControllerType::Audio
|
| ChatCompletionControllerType::Audio
|
||||||
| ChatCompletionControllerType::Image => {
|
| ChatCompletionControllerType::Image
|
||||||
|
| ChatCompletionControllerType::File => {
|
||||||
Some(message_context.combined_admin_and_user_regexes())
|
Some(message_context.combined_admin_and_user_regexes())
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -483,6 +488,13 @@ async fn handle_stage_text_generation(
|
|||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
let conversation = inject_sender_context(
|
||||||
|
conversation,
|
||||||
|
message_context
|
||||||
|
.room_config_context()
|
||||||
|
.text_generation_sender_context_mode(),
|
||||||
|
);
|
||||||
|
|
||||||
tracing::debug!(
|
tracing::debug!(
|
||||||
agent_id = agent.identifier().as_string(),
|
agent_id = agent.identifier().as_string(),
|
||||||
provider = format!("{}", agent.definition().provider.clone()),
|
provider = format!("{}", agent.definition().provider.clone()),
|
||||||
@@ -758,3 +770,238 @@ async fn generate_and_send_tts_for_message(
|
|||||||
)
|
)
|
||||||
.await
|
.await
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn inject_sender_context(
|
||||||
|
conversation: Conversation,
|
||||||
|
sender_context_mode: TextGenerationSenderContextMode,
|
||||||
|
) -> Conversation {
|
||||||
|
if sender_context_mode == TextGenerationSenderContextMode::Disabled {
|
||||||
|
return conversation;
|
||||||
|
}
|
||||||
|
|
||||||
|
let include_timestamp =
|
||||||
|
sender_context_mode == TextGenerationSenderContextMode::MatrixUserIdAndTimestamp;
|
||||||
|
|
||||||
|
let messages = conversation
|
||||||
|
.messages
|
||||||
|
.into_iter()
|
||||||
|
.map(|mut message| {
|
||||||
|
if message.author == Author::Prompt {
|
||||||
|
return message;
|
||||||
|
}
|
||||||
|
|
||||||
|
let Some(sender_id) = &message.sender_id else {
|
||||||
|
return message;
|
||||||
|
};
|
||||||
|
|
||||||
|
if let MessageContent::Text(ref mut text) = message.content {
|
||||||
|
*text = if include_timestamp {
|
||||||
|
let timestamp = message.timestamp.format("%Y-%m-%dT%H:%M:%SZ");
|
||||||
|
format!("[sender={} sent_at={}] {}", sender_id, timestamp, text)
|
||||||
|
} else {
|
||||||
|
format!("[sender={}] {}", sender_id, text)
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
message
|
||||||
|
})
|
||||||
|
.collect();
|
||||||
|
|
||||||
|
Conversation { messages }
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod sender_context_tests {
|
||||||
|
use super::inject_sender_context;
|
||||||
|
use crate::conversation::llm::{Author, Conversation, ImageDetails, Message, MessageContent};
|
||||||
|
use crate::entity::roomconfig::TextGenerationSenderContextMode;
|
||||||
|
use chrono::{TimeZone, Utc};
|
||||||
|
use mxlink::matrix_sdk::ruma::events::room::message::ImageMessageEventContent;
|
||||||
|
use mxlink::matrix_sdk::ruma::{OwnedMxcUri, OwnedUserId};
|
||||||
|
use mxlink::mime;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_inject_sender_context_prefixes_text_messages() {
|
||||||
|
let timestamp = Utc.with_ymd_and_hms(2026, 3, 23, 14, 30, 0).unwrap();
|
||||||
|
let user_id = OwnedUserId::try_from("@alice:example.com").unwrap();
|
||||||
|
|
||||||
|
let conversation = Conversation {
|
||||||
|
messages: vec![Message {
|
||||||
|
author: Author::User,
|
||||||
|
sender_id: Some(user_id),
|
||||||
|
timestamp,
|
||||||
|
content: MessageContent::Text("Hello bot".to_string()),
|
||||||
|
}],
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = inject_sender_context(
|
||||||
|
conversation,
|
||||||
|
TextGenerationSenderContextMode::MatrixUserIdAndTimestamp,
|
||||||
|
);
|
||||||
|
|
||||||
|
assert_eq!(result.messages.len(), 1);
|
||||||
|
assert_eq!(
|
||||||
|
result.messages[0].content,
|
||||||
|
MessageContent::Text(
|
||||||
|
"[sender=@alice:example.com sent_at=2026-03-23T14:30:00Z] Hello bot".to_string()
|
||||||
|
)
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_inject_sender_context_can_prefix_without_timestamp() {
|
||||||
|
let timestamp = Utc.with_ymd_and_hms(2026, 3, 23, 14, 30, 0).unwrap();
|
||||||
|
let user_id = OwnedUserId::try_from("@alice:example.com").unwrap();
|
||||||
|
|
||||||
|
let conversation = Conversation {
|
||||||
|
messages: vec![Message {
|
||||||
|
author: Author::User,
|
||||||
|
sender_id: Some(user_id),
|
||||||
|
timestamp,
|
||||||
|
content: MessageContent::Text("Hello bot".to_string()),
|
||||||
|
}],
|
||||||
|
};
|
||||||
|
|
||||||
|
let result =
|
||||||
|
inject_sender_context(conversation, TextGenerationSenderContextMode::MatrixUserId);
|
||||||
|
|
||||||
|
assert_eq!(result.messages.len(), 1);
|
||||||
|
assert_eq!(
|
||||||
|
result.messages[0].content,
|
||||||
|
MessageContent::Text("[sender=@alice:example.com] Hello bot".to_string())
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_inject_sender_context_prefixes_assistant_messages() {
|
||||||
|
let timestamp = Utc.with_ymd_and_hms(2026, 3, 23, 14, 30, 0).unwrap();
|
||||||
|
let user_id = OwnedUserId::try_from("@baibot:example.com").unwrap();
|
||||||
|
|
||||||
|
let conversation = Conversation {
|
||||||
|
messages: vec![Message {
|
||||||
|
author: Author::Assistant,
|
||||||
|
sender_id: Some(user_id),
|
||||||
|
timestamp,
|
||||||
|
content: MessageContent::Text("Hello human".to_string()),
|
||||||
|
}],
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = inject_sender_context(
|
||||||
|
conversation,
|
||||||
|
TextGenerationSenderContextMode::MatrixUserIdAndTimestamp,
|
||||||
|
);
|
||||||
|
|
||||||
|
assert_eq!(result.messages.len(), 1);
|
||||||
|
assert_eq!(
|
||||||
|
result.messages[0].content,
|
||||||
|
MessageContent::Text(
|
||||||
|
"[sender=@baibot:example.com sent_at=2026-03-23T14:30:00Z] Hello human".to_string()
|
||||||
|
)
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_inject_sender_context_skips_prompt_messages() {
|
||||||
|
let timestamp = Utc.with_ymd_and_hms(2026, 3, 23, 14, 30, 0).unwrap();
|
||||||
|
|
||||||
|
let conversation = Conversation {
|
||||||
|
messages: vec![Message {
|
||||||
|
author: Author::Prompt,
|
||||||
|
sender_id: None,
|
||||||
|
timestamp,
|
||||||
|
content: MessageContent::Text("You are a bot".to_string()),
|
||||||
|
}],
|
||||||
|
};
|
||||||
|
|
||||||
|
let result =
|
||||||
|
inject_sender_context(conversation, TextGenerationSenderContextMode::MatrixUserId);
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
result.messages[0].content,
|
||||||
|
MessageContent::Text("You are a bot".to_string())
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_inject_sender_context_skips_messages_without_sender_id() {
|
||||||
|
let timestamp = Utc.with_ymd_and_hms(2026, 3, 23, 14, 30, 0).unwrap();
|
||||||
|
|
||||||
|
let conversation = Conversation {
|
||||||
|
messages: vec![Message {
|
||||||
|
author: Author::User,
|
||||||
|
sender_id: None,
|
||||||
|
timestamp,
|
||||||
|
content: MessageContent::Text("Transcribed text".to_string()),
|
||||||
|
}],
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = inject_sender_context(
|
||||||
|
conversation,
|
||||||
|
TextGenerationSenderContextMode::MatrixUserIdAndTimestamp,
|
||||||
|
);
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
result.messages[0].content,
|
||||||
|
MessageContent::Text("Transcribed text".to_string())
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_inject_sender_context_leaves_non_text_content_unchanged() {
|
||||||
|
let timestamp = Utc.with_ymd_and_hms(2026, 3, 23, 14, 30, 0).unwrap();
|
||||||
|
let user_id = OwnedUserId::try_from("@alice:example.com").unwrap();
|
||||||
|
let image_event_content = ImageMessageEventContent::plain(
|
||||||
|
"image.png".to_string(),
|
||||||
|
OwnedMxcUri::from("mxc://example.com/1234567890"),
|
||||||
|
);
|
||||||
|
|
||||||
|
let conversation = Conversation {
|
||||||
|
messages: vec![Message {
|
||||||
|
author: Author::User,
|
||||||
|
sender_id: Some(user_id),
|
||||||
|
timestamp,
|
||||||
|
content: MessageContent::Image(ImageDetails::new(
|
||||||
|
image_event_content.clone(),
|
||||||
|
mime::IMAGE_PNG,
|
||||||
|
vec![],
|
||||||
|
)),
|
||||||
|
}],
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = inject_sender_context(
|
||||||
|
conversation,
|
||||||
|
TextGenerationSenderContextMode::MatrixUserIdAndTimestamp,
|
||||||
|
);
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
result.messages[0].content,
|
||||||
|
MessageContent::Image(ImageDetails::new(
|
||||||
|
image_event_content,
|
||||||
|
mime::IMAGE_PNG,
|
||||||
|
vec![]
|
||||||
|
))
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_inject_sender_context_none_leaves_text_unchanged() {
|
||||||
|
let timestamp = Utc.with_ymd_and_hms(2026, 3, 23, 14, 30, 0).unwrap();
|
||||||
|
let user_id = OwnedUserId::try_from("@alice:example.com").unwrap();
|
||||||
|
|
||||||
|
let conversation = Conversation {
|
||||||
|
messages: vec![Message {
|
||||||
|
author: Author::User,
|
||||||
|
sender_id: Some(user_id),
|
||||||
|
timestamp,
|
||||||
|
content: MessageContent::Text("Hello bot".to_string()),
|
||||||
|
}],
|
||||||
|
};
|
||||||
|
|
||||||
|
let result = inject_sender_context(conversation, TextGenerationSenderContextMode::Disabled);
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
result.messages[0].content,
|
||||||
|
MessageContent::Text("Hello bot".to_string())
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -58,6 +58,18 @@ pub fn determine_controller(
|
|||||||
)
|
)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
MessagePayload::File(_file_message_content) => {
|
||||||
|
let prefix_requirement_type = message_context
|
||||||
|
.room_config_context()
|
||||||
|
.text_generation_prefix_requirement_type();
|
||||||
|
|
||||||
|
match prefix_requirement_type {
|
||||||
|
TextGenerationPrefixRequirementType::CommandPrefix => ControllerType::Ignore,
|
||||||
|
TextGenerationPrefixRequirementType::No => {
|
||||||
|
ControllerType::ChatCompletion(ChatCompletionControllerType::File)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
MessagePayload::Audio(_) => {
|
MessagePayload::Audio(_) => {
|
||||||
ControllerType::ChatCompletion(ChatCompletionControllerType::Audio)
|
ControllerType::ChatCompletion(ChatCompletionControllerType::Audio)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -64,6 +64,7 @@ mod tests {
|
|||||||
original_prompt: "Generate a picture of a dog",
|
original_prompt: "Generate a picture of a dog",
|
||||||
messages: vec![Message {
|
messages: vec![Message {
|
||||||
author: Author::User,
|
author: Author::User,
|
||||||
|
sender_id: None,
|
||||||
content: MessageContent::Text("Must be blue".to_owned()),
|
content: MessageContent::Text("Must be blue".to_owned()),
|
||||||
timestamp,
|
timestamp,
|
||||||
}],
|
}],
|
||||||
@@ -75,16 +76,19 @@ mod tests {
|
|||||||
messages: vec![
|
messages: vec![
|
||||||
Message {
|
Message {
|
||||||
author: Author::User,
|
author: Author::User,
|
||||||
|
sender_id: None,
|
||||||
content: MessageContent::Text("Must be blue".to_owned()),
|
content: MessageContent::Text("Must be blue".to_owned()),
|
||||||
timestamp,
|
timestamp,
|
||||||
},
|
},
|
||||||
Message {
|
Message {
|
||||||
author: Author::Assistant,
|
author: Author::Assistant,
|
||||||
|
sender_id: None,
|
||||||
content: MessageContent::Text("Whatever".to_owned()),
|
content: MessageContent::Text("Whatever".to_owned()),
|
||||||
timestamp,
|
timestamp,
|
||||||
},
|
},
|
||||||
Message {
|
Message {
|
||||||
author: Author::User,
|
author: Author::User,
|
||||||
|
sender_id: None,
|
||||||
content: MessageContent::Text(
|
content: MessageContent::Text(
|
||||||
"Must be 3-legged.\nMust be flying.".to_owned(),
|
"Must be 3-legged.\nMust be flying.".to_owned(),
|
||||||
),
|
),
|
||||||
@@ -99,21 +103,25 @@ mod tests {
|
|||||||
messages: vec![
|
messages: vec![
|
||||||
Message {
|
Message {
|
||||||
author: Author::User,
|
author: Author::User,
|
||||||
|
sender_id: None,
|
||||||
content: MessageContent::Text("Must be blue".to_owned()),
|
content: MessageContent::Text("Must be blue".to_owned()),
|
||||||
timestamp,
|
timestamp,
|
||||||
},
|
},
|
||||||
Message {
|
Message {
|
||||||
author: Author::Assistant,
|
author: Author::Assistant,
|
||||||
|
sender_id: None,
|
||||||
content: MessageContent::Text("Whatever".to_owned()),
|
content: MessageContent::Text("Whatever".to_owned()),
|
||||||
timestamp,
|
timestamp,
|
||||||
},
|
},
|
||||||
Message {
|
Message {
|
||||||
author: Author::User,
|
author: Author::User,
|
||||||
|
sender_id: None,
|
||||||
content: MessageContent::Text("Again".to_owned()),
|
content: MessageContent::Text("Again".to_owned()),
|
||||||
timestamp,
|
timestamp,
|
||||||
},
|
},
|
||||||
Message {
|
Message {
|
||||||
author: Author::User,
|
author: Author::User,
|
||||||
|
sender_id: None,
|
||||||
content: MessageContent::Text("again".to_owned()),
|
content: MessageContent::Text("again".to_owned()),
|
||||||
timestamp,
|
timestamp,
|
||||||
},
|
},
|
||||||
|
|||||||
@@ -1,5 +1,8 @@
|
|||||||
use chrono::{DateTime, Utc};
|
use chrono::{DateTime, Utc};
|
||||||
use mxlink::matrix_sdk::ruma::events::room::message::ImageMessageEventContent;
|
use mxlink::matrix_sdk::ruma::OwnedUserId;
|
||||||
|
use mxlink::matrix_sdk::ruma::events::room::message::{
|
||||||
|
FileMessageEventContent, ImageMessageEventContent,
|
||||||
|
};
|
||||||
use mxlink::mime::Mime;
|
use mxlink::mime::Mime;
|
||||||
|
|
||||||
use crate::agent::provider::ImageSource;
|
use crate::agent::provider::ImageSource;
|
||||||
@@ -14,6 +17,7 @@ pub enum Author {
|
|||||||
#[derive(Debug, Clone)]
|
#[derive(Debug, Clone)]
|
||||||
pub struct Message {
|
pub struct Message {
|
||||||
pub author: Author,
|
pub author: Author,
|
||||||
|
pub sender_id: Option<OwnedUserId>,
|
||||||
pub timestamp: DateTime<Utc>,
|
pub timestamp: DateTime<Utc>,
|
||||||
pub content: MessageContent,
|
pub content: MessageContent,
|
||||||
}
|
}
|
||||||
@@ -48,10 +52,35 @@ impl From<ImageDetails> for ImageSource {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Clone)]
|
||||||
|
pub struct FileDetails {
|
||||||
|
pub event_content: FileMessageEventContent,
|
||||||
|
pub mime: Mime,
|
||||||
|
pub data: Vec<u8>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl FileDetails {
|
||||||
|
pub fn new(event_content: FileMessageEventContent, mime: Mime, data: Vec<u8>) -> Self {
|
||||||
|
Self {
|
||||||
|
event_content,
|
||||||
|
mime,
|
||||||
|
data,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn filename(&self) -> String {
|
||||||
|
self.event_content
|
||||||
|
.filename
|
||||||
|
.clone()
|
||||||
|
.unwrap_or(self.event_content.body.clone())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone)]
|
#[derive(Debug, Clone)]
|
||||||
pub enum MessageContent {
|
pub enum MessageContent {
|
||||||
Text(String),
|
Text(String),
|
||||||
Image(ImageDetails),
|
Image(ImageDetails),
|
||||||
|
File(FileDetails),
|
||||||
}
|
}
|
||||||
|
|
||||||
impl PartialEq for MessageContent {
|
impl PartialEq for MessageContent {
|
||||||
@@ -62,6 +91,7 @@ impl PartialEq for MessageContent {
|
|||||||
// We can probably do better than this by inspecting `.event_conten1t.source`, but for now this is good enough.
|
// We can probably do better than this by inspecting `.event_conten1t.source`, but for now this is good enough.
|
||||||
a.filename() == b.filename()
|
a.filename() == b.filename()
|
||||||
}
|
}
|
||||||
|
(MessageContent::File(a), MessageContent::File(b)) => a.filename() == b.filename(),
|
||||||
_ => false,
|
_ => false,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -76,6 +106,11 @@ impl Conversation {
|
|||||||
///
|
///
|
||||||
/// Certain models (like Anthropic) cannot tolerate consecutive messages by the same author,
|
/// Certain models (like Anthropic) cannot tolerate consecutive messages by the same author,
|
||||||
/// so combining them helps avoid issues.
|
/// so combining them helps avoid issues.
|
||||||
|
///
|
||||||
|
/// When multiple text messages by the same author are merged, the resulting message keeps a
|
||||||
|
/// `sender_id` only if all merged messages came from the same sender. Mixed-sender merges are
|
||||||
|
/// possible for user turns in multi-user rooms, so `sender_id` is cleared in that case to
|
||||||
|
/// avoid incorrectly attributing the whole merged turn to the first sender.
|
||||||
/// See: https://github.com/etkecc/baibot/issues/13
|
/// See: https://github.com/etkecc/baibot/issues/13
|
||||||
pub fn combine_consecutive_messages(&self) -> Conversation {
|
pub fn combine_consecutive_messages(&self) -> Conversation {
|
||||||
// We'll likely get fewer messages, but let's reserve the maximum we expect.
|
// We'll likely get fewer messages, but let's reserve the maximum we expect.
|
||||||
@@ -106,6 +141,10 @@ impl Conversation {
|
|||||||
text.push('\n');
|
text.push('\n');
|
||||||
text.push_str(message_text_content);
|
text.push_str(message_text_content);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if last_message.sender_id != message.sender_id {
|
||||||
|
last_message.sender_id = None;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
Conversation {
|
Conversation {
|
||||||
@@ -122,7 +161,7 @@ impl Conversation {
|
|||||||
mod tests {
|
mod tests {
|
||||||
use super::*;
|
use super::*;
|
||||||
use chrono::{TimeZone, Utc};
|
use chrono::{TimeZone, Utc};
|
||||||
use mxlink::matrix_sdk::ruma::OwnedMxcUri;
|
use mxlink::matrix_sdk::ruma::{OwnedMxcUri, OwnedUserId};
|
||||||
use mxlink::mime;
|
use mxlink::mime;
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
@@ -145,21 +184,25 @@ mod tests {
|
|||||||
// User's turn
|
// User's turn
|
||||||
Message {
|
Message {
|
||||||
author: Author::User,
|
author: Author::User,
|
||||||
|
sender_id: None,
|
||||||
content: MessageContent::Text("Hello".to_string()),
|
content: MessageContent::Text("Hello".to_string()),
|
||||||
timestamp: timestamp_1,
|
timestamp: timestamp_1,
|
||||||
},
|
},
|
||||||
Message {
|
Message {
|
||||||
author: Author::User,
|
author: Author::User,
|
||||||
|
sender_id: None,
|
||||||
content: MessageContent::Text("How are you?".to_string()),
|
content: MessageContent::Text("How are you?".to_string()),
|
||||||
timestamp: timestamp_2,
|
timestamp: timestamp_2,
|
||||||
},
|
},
|
||||||
Message {
|
Message {
|
||||||
author: Author::User,
|
author: Author::User,
|
||||||
|
sender_id: None,
|
||||||
content: MessageContent::Text("I'm OK, btw.".to_string()),
|
content: MessageContent::Text("I'm OK, btw.".to_string()),
|
||||||
timestamp: timestamp_3,
|
timestamp: timestamp_3,
|
||||||
},
|
},
|
||||||
Message {
|
Message {
|
||||||
author: Author::User,
|
author: Author::User,
|
||||||
|
sender_id: None,
|
||||||
content: MessageContent::Image(ImageDetails::new(
|
content: MessageContent::Image(ImageDetails::new(
|
||||||
image_event_content.clone(),
|
image_event_content.clone(),
|
||||||
mime::IMAGE_PNG,
|
mime::IMAGE_PNG,
|
||||||
@@ -169,28 +212,33 @@ mod tests {
|
|||||||
},
|
},
|
||||||
Message {
|
Message {
|
||||||
author: Author::User,
|
author: Author::User,
|
||||||
|
sender_id: None,
|
||||||
content: MessageContent::Text("Above is an image.".to_string()),
|
content: MessageContent::Text("Above is an image.".to_string()),
|
||||||
timestamp: timestamp_4,
|
timestamp: timestamp_4,
|
||||||
},
|
},
|
||||||
Message {
|
Message {
|
||||||
author: Author::User,
|
author: Author::User,
|
||||||
|
sender_id: None,
|
||||||
content: MessageContent::Text("Would you take a look at it?".to_string()),
|
content: MessageContent::Text("Would you take a look at it?".to_string()),
|
||||||
timestamp: timestamp_4,
|
timestamp: timestamp_4,
|
||||||
},
|
},
|
||||||
// Assistant's turn
|
// Assistant's turn
|
||||||
Message {
|
Message {
|
||||||
author: Author::Assistant,
|
author: Author::Assistant,
|
||||||
|
sender_id: None,
|
||||||
content: MessageContent::Text("Hi there!".to_string()),
|
content: MessageContent::Text("Hi there!".to_string()),
|
||||||
timestamp: timestamp_2,
|
timestamp: timestamp_2,
|
||||||
},
|
},
|
||||||
Message {
|
Message {
|
||||||
author: Author::Assistant,
|
author: Author::Assistant,
|
||||||
|
sender_id: None,
|
||||||
content: MessageContent::Text("I'm doing well, thank you.".to_string()),
|
content: MessageContent::Text("I'm doing well, thank you.".to_string()),
|
||||||
timestamp: timestamp_3,
|
timestamp: timestamp_3,
|
||||||
},
|
},
|
||||||
// User's turn
|
// User's turn
|
||||||
Message {
|
Message {
|
||||||
author: Author::User,
|
author: Author::User,
|
||||||
|
sender_id: None,
|
||||||
content: MessageContent::Text("That's great!".to_string()),
|
content: MessageContent::Text("That's great!".to_string()),
|
||||||
timestamp: timestamp_3,
|
timestamp: timestamp_3,
|
||||||
},
|
},
|
||||||
@@ -239,4 +287,39 @@ mod tests {
|
|||||||
);
|
);
|
||||||
assert_eq!(conversation.messages[4].timestamp, timestamp_3);
|
assert_eq!(conversation.messages[4].timestamp, timestamp_3);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn combine_consecutive_messages_clears_sender_id_for_mixed_sender_turns() {
|
||||||
|
let timestamp_1 = Utc.with_ymd_and_hms(2024, 9, 20, 18, 34, 15).unwrap();
|
||||||
|
let timestamp_2 = Utc.with_ymd_and_hms(2024, 9, 20, 18, 34, 16).unwrap();
|
||||||
|
let sender_1 = OwnedUserId::try_from("@alice:example.com").unwrap();
|
||||||
|
let sender_2 = OwnedUserId::try_from("@bob:example.com").unwrap();
|
||||||
|
|
||||||
|
let conversation = Conversation {
|
||||||
|
messages: vec![
|
||||||
|
Message {
|
||||||
|
author: Author::User,
|
||||||
|
sender_id: Some(sender_1),
|
||||||
|
content: MessageContent::Text("Hello".to_string()),
|
||||||
|
timestamp: timestamp_1,
|
||||||
|
},
|
||||||
|
Message {
|
||||||
|
author: Author::User,
|
||||||
|
sender_id: Some(sender_2),
|
||||||
|
content: MessageContent::Text("Hi there".to_string()),
|
||||||
|
timestamp: timestamp_2,
|
||||||
|
},
|
||||||
|
],
|
||||||
|
};
|
||||||
|
|
||||||
|
let conversation = conversation.combine_consecutive_messages();
|
||||||
|
|
||||||
|
assert_eq!(conversation.messages.len(), 1);
|
||||||
|
assert_eq!(conversation.messages[0].sender_id, None);
|
||||||
|
assert_eq!(
|
||||||
|
conversation.messages[0].content,
|
||||||
|
MessageContent::Text("Hello\nHi there".to_string())
|
||||||
|
);
|
||||||
|
assert_eq!(conversation.messages[0].timestamp, timestamp_1);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -20,6 +20,7 @@ fn test_messages_by_the_bot_are_identified_correctly() {
|
|||||||
let llm_message = convert_matrix_message_to_llm_message(&matrix_message, &bot_user_id).unwrap();
|
let llm_message = convert_matrix_message_to_llm_message(&matrix_message, &bot_user_id).unwrap();
|
||||||
|
|
||||||
assert_eq!(llm_message.author, Author::Assistant);
|
assert_eq!(llm_message.author, Author::Assistant);
|
||||||
|
assert_eq!(llm_message.sender_id, Some(bot_user_id.clone()));
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
llm_message.content,
|
llm_message.content,
|
||||||
MessageContent::Text("Hello!".to_string())
|
MessageContent::Text("Hello!".to_string())
|
||||||
@@ -47,6 +48,7 @@ fn test_notice_messages_by_bot_with_speech_to_text_prefix_are_cleaned_up_and_con
|
|||||||
let llm_message = convert_matrix_message_to_llm_message(&matrix_message, &bot_user_id).unwrap();
|
let llm_message = convert_matrix_message_to_llm_message(&matrix_message, &bot_user_id).unwrap();
|
||||||
|
|
||||||
assert_eq!(llm_message.author, Author::User);
|
assert_eq!(llm_message.author, Author::User);
|
||||||
|
assert_eq!(llm_message.sender_id, None);
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
llm_message.content,
|
llm_message.content,
|
||||||
MessageContent::Text(source_message_text.to_string())
|
MessageContent::Text(source_message_text.to_string())
|
||||||
@@ -75,6 +77,30 @@ fn test_notice_error_messages_by_bot_are_ignored() {
|
|||||||
assert!(llm_message.is_none());
|
assert!(llm_message.is_none());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_user_messages_preserve_sender_id() {
|
||||||
|
let bot_user_id =
|
||||||
|
OwnedUserId::try_from("@bot:example.com").expect("Failed to parse bot user ID");
|
||||||
|
|
||||||
|
let user_id = OwnedUserId::try_from("@alice:example.com").expect("Failed to parse user ID");
|
||||||
|
|
||||||
|
let matrix_message = super::super::matrix::MatrixMessage {
|
||||||
|
sender_id: user_id.clone(),
|
||||||
|
content: super::super::matrix::MatrixMessageContent::Text("Hello!".to_owned()),
|
||||||
|
mentioned_users: vec![],
|
||||||
|
timestamp: chrono::Utc::now(),
|
||||||
|
};
|
||||||
|
|
||||||
|
let llm_message = convert_matrix_message_to_llm_message(&matrix_message, &bot_user_id).unwrap();
|
||||||
|
|
||||||
|
assert_eq!(llm_message.author, Author::User);
|
||||||
|
assert_eq!(llm_message.sender_id, Some(user_id));
|
||||||
|
assert_eq!(
|
||||||
|
llm_message.content,
|
||||||
|
MessageContent::Text("Hello!".to_string())
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_other_notice_messages_by_the_bot_are_ignored() {
|
fn test_other_notice_messages_by_the_bot_are_ignored() {
|
||||||
// Also see `test_notice_error_messages_by_bot_are_ignored()`.
|
// Also see `test_notice_error_messages_by_bot_are_ignored()`.
|
||||||
|
|||||||
@@ -1,15 +1,15 @@
|
|||||||
use tiktoken_rs::CoreBPE;
|
use tiktoken_rs::CoreBPE;
|
||||||
use tiktoken_rs::get_bpe_from_tokenizer;
|
use tiktoken_rs::bpe_for_tokenizer;
|
||||||
use tiktoken_rs::tokenizer;
|
use tiktoken_rs::tokenizer;
|
||||||
|
|
||||||
use super::{Author, Message, MessageContent};
|
use super::{Author, Message, MessageContent};
|
||||||
|
|
||||||
fn get_bpe_for_model(model: &str) -> CoreBPE {
|
fn get_bpe_for_model(model: &str) -> &'static CoreBPE {
|
||||||
let tokenizer = tokenizer::get_tokenizer(model)
|
let tokenizer = tokenizer::get_tokenizer(model)
|
||||||
.or_else(|| tokenizer::get_tokenizer("gpt-4"))
|
.or_else(|| tokenizer::get_tokenizer("gpt-4"))
|
||||||
.unwrap();
|
.unwrap();
|
||||||
|
|
||||||
get_bpe_from_tokenizer(tokenizer).unwrap()
|
bpe_for_tokenizer(tokenizer).unwrap()
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn shorten_messages_list_to_context_size(
|
pub fn shorten_messages_list_to_context_size(
|
||||||
@@ -26,7 +26,7 @@ pub fn shorten_messages_list_to_context_size(
|
|||||||
// We want to retain the prompt in all cases, so we always count it first.
|
// We want to retain the prompt in all cases, so we always count it first.
|
||||||
// We also always reserve enough tokens for the maximum response we expect.
|
// We also always reserve enough tokens for the maximum response we expect.
|
||||||
let mut current_context_length: u32 = if let Some(prompt_message) = prompt_message {
|
let mut current_context_length: u32 = if let Some(prompt_message) = prompt_message {
|
||||||
calculate_token_size_for_message(&bpe, model, prompt_message)
|
calculate_token_size_for_message(bpe, model, prompt_message)
|
||||||
+ max_response_tokens.unwrap_or(0)
|
+ max_response_tokens.unwrap_or(0)
|
||||||
} else {
|
} else {
|
||||||
0
|
0
|
||||||
@@ -37,7 +37,7 @@ pub fn shorten_messages_list_to_context_size(
|
|||||||
let mut messages_to_keep: Vec<Message> = Vec::new();
|
let mut messages_to_keep: Vec<Message> = Vec::new();
|
||||||
|
|
||||||
for message in messages {
|
for message in messages {
|
||||||
let tokens_for_message = calculate_token_size_for_message(&bpe, model, &message);
|
let tokens_for_message = calculate_token_size_for_message(bpe, model, &message);
|
||||||
|
|
||||||
if current_context_length + tokens_for_message > max_context_tokens {
|
if current_context_length + tokens_for_message > max_context_tokens {
|
||||||
break;
|
break;
|
||||||
@@ -74,6 +74,7 @@ fn calculate_token_size_for_message(bpe: &CoreBPE, model: &str, message: &Messag
|
|||||||
let text_length = match &message.content {
|
let text_length = match &message.content {
|
||||||
MessageContent::Text(text) => bpe.encode_with_special_tokens(text).len() as i32,
|
MessageContent::Text(text) => bpe.encode_with_special_tokens(text).len() as i32,
|
||||||
MessageContent::Image(..) => 0,
|
MessageContent::Image(..) => 0,
|
||||||
|
MessageContent::File(..) => 0,
|
||||||
};
|
};
|
||||||
|
|
||||||
(text_length + role_length + tokens_per_message + tokens_per_name) as u32
|
(text_length + role_length + tokens_per_message + tokens_per_name) as u32
|
||||||
@@ -88,11 +89,12 @@ pub mod test {
|
|||||||
|
|
||||||
let message = super::Message {
|
let message = super::Message {
|
||||||
author: super::Author::User,
|
author: super::Author::User,
|
||||||
|
sender_id: None,
|
||||||
content: super::MessageContent::Text("Hello there!".to_string()),
|
content: super::MessageContent::Text("Hello there!".to_string()),
|
||||||
timestamp: chrono::Utc::now(),
|
timestamp: chrono::Utc::now(),
|
||||||
};
|
};
|
||||||
|
|
||||||
let tokens = super::calculate_token_size_for_message(&bpe, model, &message);
|
let tokens = super::calculate_token_size_for_message(bpe, model, &message);
|
||||||
|
|
||||||
assert_eq!(8, tokens);
|
assert_eq!(8, tokens);
|
||||||
}
|
}
|
||||||
@@ -107,6 +109,7 @@ pub mod test {
|
|||||||
|
|
||||||
let prompt = super::Message {
|
let prompt = super::Message {
|
||||||
author: super::Author::Prompt,
|
author: super::Author::Prompt,
|
||||||
|
sender_id: None,
|
||||||
content: super::MessageContent::Text("You are a bot!".to_string()),
|
content: super::MessageContent::Text("You are a bot!".to_string()),
|
||||||
timestamp: chrono::Utc::now(),
|
timestamp: chrono::Utc::now(),
|
||||||
};
|
};
|
||||||
@@ -114,13 +117,14 @@ pub mod test {
|
|||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
prompt_length,
|
prompt_length,
|
||||||
super::calculate_token_size_for_message(&bpe, model, &prompt)
|
super::calculate_token_size_for_message(bpe, model, &prompt)
|
||||||
);
|
);
|
||||||
|
|
||||||
let mut conversation_messages = Vec::new();
|
let mut conversation_messages = Vec::new();
|
||||||
|
|
||||||
let first = super::Message {
|
let first = super::Message {
|
||||||
author: super::Author::User,
|
author: super::Author::User,
|
||||||
|
sender_id: None,
|
||||||
content: super::MessageContent::Text("Hello there!".to_string()),
|
content: super::MessageContent::Text("Hello there!".to_string()),
|
||||||
timestamp: chrono::Utc::now(),
|
timestamp: chrono::Utc::now(),
|
||||||
};
|
};
|
||||||
@@ -128,13 +132,14 @@ pub mod test {
|
|||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
first_length,
|
first_length,
|
||||||
super::calculate_token_size_for_message(&bpe, model, &first)
|
super::calculate_token_size_for_message(bpe, model, &first)
|
||||||
);
|
);
|
||||||
|
|
||||||
conversation_messages.push(first);
|
conversation_messages.push(first);
|
||||||
|
|
||||||
let second = super::Message {
|
let second = super::Message {
|
||||||
author: super::Author::Assistant,
|
author: super::Author::Assistant,
|
||||||
|
sender_id: None,
|
||||||
content: super::MessageContent::Text("Hello!".to_string()),
|
content: super::MessageContent::Text("Hello!".to_string()),
|
||||||
timestamp: chrono::Utc::now(),
|
timestamp: chrono::Utc::now(),
|
||||||
};
|
};
|
||||||
@@ -142,13 +147,14 @@ pub mod test {
|
|||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
second_length,
|
second_length,
|
||||||
super::calculate_token_size_for_message(&bpe, model, &second)
|
super::calculate_token_size_for_message(bpe, model, &second)
|
||||||
);
|
);
|
||||||
|
|
||||||
conversation_messages.push(second);
|
conversation_messages.push(second);
|
||||||
|
|
||||||
let third = super::Message {
|
let third = super::Message {
|
||||||
author: super::Author::User,
|
author: super::Author::User,
|
||||||
|
sender_id: None,
|
||||||
content: super::MessageContent::Text(
|
content: super::MessageContent::Text(
|
||||||
"This is the 3rd message in this conversation. It shall be preserved.".to_owned(),
|
"This is the 3rd message in this conversation. It shall be preserved.".to_owned(),
|
||||||
),
|
),
|
||||||
@@ -158,13 +164,14 @@ pub mod test {
|
|||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
third_length,
|
third_length,
|
||||||
super::calculate_token_size_for_message(&bpe, model, &third)
|
super::calculate_token_size_for_message(bpe, model, &third)
|
||||||
);
|
);
|
||||||
|
|
||||||
conversation_messages.push(third.clone());
|
conversation_messages.push(third.clone());
|
||||||
|
|
||||||
let forth = super::Message {
|
let forth = super::Message {
|
||||||
author: super::Author::Assistant,
|
author: super::Author::Assistant,
|
||||||
|
sender_id: None,
|
||||||
content: super::MessageContent::Text(
|
content: super::MessageContent::Text(
|
||||||
"This is yet another message that shall be preserved.".to_owned(),
|
"This is yet another message that shall be preserved.".to_owned(),
|
||||||
),
|
),
|
||||||
@@ -174,7 +181,7 @@ pub mod test {
|
|||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
forth_length,
|
forth_length,
|
||||||
super::calculate_token_size_for_message(&bpe, model, &forth)
|
super::calculate_token_size_for_message(bpe, model, &forth)
|
||||||
);
|
);
|
||||||
|
|
||||||
conversation_messages.push(forth.clone());
|
conversation_messages.push(forth.clone());
|
||||||
@@ -212,6 +219,7 @@ pub mod test {
|
|||||||
|
|
||||||
let prompt = super::Message {
|
let prompt = super::Message {
|
||||||
author: super::Author::User,
|
author: super::Author::User,
|
||||||
|
sender_id: None,
|
||||||
content: super::MessageContent::Text("あなたはボットです。".to_string()),
|
content: super::MessageContent::Text("あなたはボットです。".to_string()),
|
||||||
timestamp: chrono::Utc::now(),
|
timestamp: chrono::Utc::now(),
|
||||||
};
|
};
|
||||||
@@ -219,13 +227,14 @@ pub mod test {
|
|||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
prompt_length,
|
prompt_length,
|
||||||
super::calculate_token_size_for_message(&bpe, model, &prompt)
|
super::calculate_token_size_for_message(bpe, model, &prompt)
|
||||||
);
|
);
|
||||||
|
|
||||||
let mut conversation_messages = Vec::new();
|
let mut conversation_messages = Vec::new();
|
||||||
|
|
||||||
let first = super::Message {
|
let first = super::Message {
|
||||||
author: super::Author::User,
|
author: super::Author::User,
|
||||||
|
sender_id: None,
|
||||||
content: super::MessageContent::Text("こんにちは!".to_string()),
|
content: super::MessageContent::Text("こんにちは!".to_string()),
|
||||||
timestamp: chrono::Utc::now(),
|
timestamp: chrono::Utc::now(),
|
||||||
};
|
};
|
||||||
@@ -233,13 +242,14 @@ pub mod test {
|
|||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
first_length,
|
first_length,
|
||||||
super::calculate_token_size_for_message(&bpe, model, &first)
|
super::calculate_token_size_for_message(bpe, model, &first)
|
||||||
);
|
);
|
||||||
|
|
||||||
conversation_messages.push(first);
|
conversation_messages.push(first);
|
||||||
|
|
||||||
let second = super::Message {
|
let second = super::Message {
|
||||||
author: super::Author::Assistant,
|
author: super::Author::Assistant,
|
||||||
|
sender_id: None,
|
||||||
content: super::MessageContent::Text("こんにちは。今日は元気ですか。".to_string()),
|
content: super::MessageContent::Text("こんにちは。今日は元気ですか。".to_string()),
|
||||||
timestamp: chrono::Utc::now(),
|
timestamp: chrono::Utc::now(),
|
||||||
};
|
};
|
||||||
@@ -247,13 +257,14 @@ pub mod test {
|
|||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
second_length,
|
second_length,
|
||||||
super::calculate_token_size_for_message(&bpe, model, &second)
|
super::calculate_token_size_for_message(bpe, model, &second)
|
||||||
);
|
);
|
||||||
|
|
||||||
conversation_messages.push(second);
|
conversation_messages.push(second);
|
||||||
|
|
||||||
let third = super::Message {
|
let third = super::Message {
|
||||||
author: super::Author::User,
|
author: super::Author::User,
|
||||||
|
sender_id: None,
|
||||||
content: super::MessageContent::Text(
|
content: super::MessageContent::Text(
|
||||||
"これは第3のメッセージなので、保存されます。".to_string(),
|
"これは第3のメッセージなので、保存されます。".to_string(),
|
||||||
),
|
),
|
||||||
@@ -263,13 +274,14 @@ pub mod test {
|
|||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
third_length,
|
third_length,
|
||||||
super::calculate_token_size_for_message(&bpe, model, &third)
|
super::calculate_token_size_for_message(bpe, model, &third)
|
||||||
);
|
);
|
||||||
|
|
||||||
conversation_messages.push(third.clone());
|
conversation_messages.push(third.clone());
|
||||||
|
|
||||||
let forth = super::Message {
|
let forth = super::Message {
|
||||||
author: super::Author::Assistant,
|
author: super::Author::Assistant,
|
||||||
|
sender_id: None,
|
||||||
content: super::MessageContent::Text(
|
content: super::MessageContent::Text(
|
||||||
"これはもう一つの保存されますメッセージです。".to_string(),
|
"これはもう一つの保存されますメッセージです。".to_string(),
|
||||||
),
|
),
|
||||||
@@ -279,7 +291,7 @@ pub mod test {
|
|||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
forth_length,
|
forth_length,
|
||||||
super::calculate_token_size_for_message(&bpe, model, &forth)
|
super::calculate_token_size_for_message(bpe, model, &forth)
|
||||||
);
|
);
|
||||||
|
|
||||||
conversation_messages.push(forth.clone());
|
conversation_messages.push(forth.clone());
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
use mxlink::matrix_sdk::ruma::OwnedUserId;
|
use mxlink::matrix_sdk::ruma::OwnedUserId;
|
||||||
|
|
||||||
use super::entity::{Author, ImageDetails, Message, MessageContent};
|
use super::entity::{Author, FileDetails, ImageDetails, Message, MessageContent};
|
||||||
use crate::conversation::matrix::{MatrixMessage, MatrixMessageContent};
|
use crate::conversation::matrix::{MatrixMessage, MatrixMessageContent};
|
||||||
use crate::utils::text_to_speech as text_to_speech_utils;
|
use crate::utils::text_to_speech as text_to_speech_utils;
|
||||||
|
|
||||||
@@ -17,14 +17,17 @@ pub fn convert_matrix_message_to_llm_message(
|
|||||||
|
|
||||||
fn convert_bot_message(matrix_message: &MatrixMessage) -> Option<Message> {
|
fn convert_bot_message(matrix_message: &MatrixMessage) -> Option<Message> {
|
||||||
match &matrix_message.content {
|
match &matrix_message.content {
|
||||||
MatrixMessageContent::Text(text) => {
|
MatrixMessageContent::Text(text) => convert_bot_text_message(
|
||||||
convert_bot_text_message(text, &matrix_message.timestamp)
|
text,
|
||||||
}
|
&matrix_message.timestamp,
|
||||||
|
matrix_message.sender_id.clone(),
|
||||||
|
),
|
||||||
MatrixMessageContent::Notice(text) => {
|
MatrixMessageContent::Notice(text) => {
|
||||||
convert_bot_notice_message(text, &matrix_message.timestamp)
|
convert_bot_notice_message(text, &matrix_message.timestamp)
|
||||||
}
|
}
|
||||||
MatrixMessageContent::Image(image_content, mime_type, media_bytes) => Some(Message {
|
MatrixMessageContent::Image(image_content, mime_type, media_bytes) => Some(Message {
|
||||||
author: Author::Assistant,
|
author: Author::Assistant,
|
||||||
|
sender_id: Some(matrix_message.sender_id.clone()),
|
||||||
content: MessageContent::Image(ImageDetails::new(
|
content: MessageContent::Image(ImageDetails::new(
|
||||||
image_content.clone(),
|
image_content.clone(),
|
||||||
mime_type.clone(),
|
mime_type.clone(),
|
||||||
@@ -32,15 +35,27 @@ fn convert_bot_message(matrix_message: &MatrixMessage) -> Option<Message> {
|
|||||||
)),
|
)),
|
||||||
timestamp: matrix_message.timestamp.to_owned(),
|
timestamp: matrix_message.timestamp.to_owned(),
|
||||||
}),
|
}),
|
||||||
|
MatrixMessageContent::File(file_content, mime_type, media_bytes) => Some(Message {
|
||||||
|
author: Author::Assistant,
|
||||||
|
sender_id: Some(matrix_message.sender_id.clone()),
|
||||||
|
content: MessageContent::File(FileDetails::new(
|
||||||
|
file_content.clone(),
|
||||||
|
mime_type.clone(),
|
||||||
|
media_bytes.clone(),
|
||||||
|
)),
|
||||||
|
timestamp: matrix_message.timestamp.to_owned(),
|
||||||
|
}),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn convert_bot_text_message(
|
fn convert_bot_text_message(
|
||||||
text: &str,
|
text: &str,
|
||||||
timestamp: &chrono::DateTime<chrono::Utc>,
|
timestamp: &chrono::DateTime<chrono::Utc>,
|
||||||
|
sender_id: OwnedUserId,
|
||||||
) -> Option<Message> {
|
) -> Option<Message> {
|
||||||
Some(Message {
|
Some(Message {
|
||||||
author: Author::Assistant,
|
author: Author::Assistant,
|
||||||
|
sender_id: Some(sender_id),
|
||||||
content: MessageContent::Text(text.to_owned()),
|
content: MessageContent::Text(text.to_owned()),
|
||||||
timestamp: timestamp.to_owned(),
|
timestamp: timestamp.to_owned(),
|
||||||
})
|
})
|
||||||
@@ -59,8 +74,10 @@ fn convert_bot_notice_message(
|
|||||||
|
|
||||||
if let Some(text) = text_to_speech_utils::parse_transcribed_message_text(text) {
|
if let Some(text) = text_to_speech_utils::parse_transcribed_message_text(text) {
|
||||||
// This is a transcription message. We remove the prefix and consider it as a message sent by the user.
|
// This is a transcription message. We remove the prefix and consider it as a message sent by the user.
|
||||||
|
// sender_id is None because the original speaker is unknown.
|
||||||
return Some(Message {
|
return Some(Message {
|
||||||
author: Author::User,
|
author: Author::User,
|
||||||
|
sender_id: None,
|
||||||
content: MessageContent::Text(text.to_owned()),
|
content: MessageContent::Text(text.to_owned()),
|
||||||
timestamp: timestamp.to_owned(),
|
timestamp: timestamp.to_owned(),
|
||||||
});
|
});
|
||||||
@@ -73,16 +90,19 @@ fn convert_user_message(matrix_message: &MatrixMessage) -> Option<Message> {
|
|||||||
match &matrix_message.content {
|
match &matrix_message.content {
|
||||||
MatrixMessageContent::Text(text) => Some(Message {
|
MatrixMessageContent::Text(text) => Some(Message {
|
||||||
author: Author::User,
|
author: Author::User,
|
||||||
|
sender_id: Some(matrix_message.sender_id.clone()),
|
||||||
content: MessageContent::Text(text.clone()),
|
content: MessageContent::Text(text.clone()),
|
||||||
timestamp: matrix_message.timestamp.to_owned(),
|
timestamp: matrix_message.timestamp.to_owned(),
|
||||||
}),
|
}),
|
||||||
MatrixMessageContent::Notice(text) => Some(Message {
|
MatrixMessageContent::Notice(text) => Some(Message {
|
||||||
author: Author::User,
|
author: Author::User,
|
||||||
|
sender_id: Some(matrix_message.sender_id.clone()),
|
||||||
content: MessageContent::Text(text.clone()),
|
content: MessageContent::Text(text.clone()),
|
||||||
timestamp: matrix_message.timestamp.to_owned(),
|
timestamp: matrix_message.timestamp.to_owned(),
|
||||||
}),
|
}),
|
||||||
MatrixMessageContent::Image(image_content, mime_type, media_bytes) => Some(Message {
|
MatrixMessageContent::Image(image_content, mime_type, media_bytes) => Some(Message {
|
||||||
author: Author::User,
|
author: Author::User,
|
||||||
|
sender_id: Some(matrix_message.sender_id.clone()),
|
||||||
content: MessageContent::Image(ImageDetails::new(
|
content: MessageContent::Image(ImageDetails::new(
|
||||||
image_content.clone(),
|
image_content.clone(),
|
||||||
mime_type.clone(),
|
mime_type.clone(),
|
||||||
@@ -90,5 +110,15 @@ fn convert_user_message(matrix_message: &MatrixMessage) -> Option<Message> {
|
|||||||
)),
|
)),
|
||||||
timestamp: matrix_message.timestamp.to_owned(),
|
timestamp: matrix_message.timestamp.to_owned(),
|
||||||
}),
|
}),
|
||||||
|
MatrixMessageContent::File(file_content, mime_type, media_bytes) => Some(Message {
|
||||||
|
author: Author::User,
|
||||||
|
sender_id: Some(matrix_message.sender_id.clone()),
|
||||||
|
content: MessageContent::File(FileDetails::new(
|
||||||
|
file_content.clone(),
|
||||||
|
mime_type.clone(),
|
||||||
|
media_bytes.clone(),
|
||||||
|
)),
|
||||||
|
timestamp: matrix_message.timestamp.to_owned(),
|
||||||
|
}),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -2,7 +2,9 @@ use chrono::{DateTime, Utc};
|
|||||||
use regex::Regex;
|
use regex::Regex;
|
||||||
|
|
||||||
use mxlink::matrix_sdk::ruma::OwnedUserId;
|
use mxlink::matrix_sdk::ruma::OwnedUserId;
|
||||||
use mxlink::matrix_sdk::ruma::events::room::message::ImageMessageEventContent;
|
use mxlink::matrix_sdk::ruma::events::room::message::{
|
||||||
|
FileMessageEventContent, ImageMessageEventContent,
|
||||||
|
};
|
||||||
use mxlink::mime::Mime;
|
use mxlink::mime::Mime;
|
||||||
|
|
||||||
#[derive(Clone)]
|
#[derive(Clone)]
|
||||||
@@ -18,6 +20,7 @@ pub enum MatrixMessageContent {
|
|||||||
Text(String),
|
Text(String),
|
||||||
Notice(String),
|
Notice(String),
|
||||||
Image(ImageMessageEventContent, Mime, Vec<u8>),
|
Image(ImageMessageEventContent, Mime, Vec<u8>),
|
||||||
|
File(FileMessageEventContent, Mime, Vec<u8>),
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Clone)]
|
#[derive(Clone)]
|
||||||
|
|||||||
@@ -154,35 +154,35 @@ pub async fn process_matrix_messages(
|
|||||||
|
|
||||||
let mut message = message.clone();
|
let mut message = message.clone();
|
||||||
|
|
||||||
if i == 0 && !params.first_message_prefixes_to_strip.is_empty() {
|
if i == 0
|
||||||
if let MatrixMessageContent::Text(message_text) = &message.content {
|
&& !params.first_message_prefixes_to_strip.is_empty()
|
||||||
let mut message_text = message_text.clone();
|
&& let MatrixMessageContent::Text(message_text) = &message.content
|
||||||
|
{
|
||||||
|
let mut message_text = message_text.clone();
|
||||||
|
|
||||||
for prefix in ¶ms.first_message_prefixes_to_strip {
|
for prefix in ¶ms.first_message_prefixes_to_strip {
|
||||||
if let Some(message_text_stripped) = message_text.strip_prefix(prefix) {
|
if let Some(message_text_stripped) = message_text.strip_prefix(prefix) {
|
||||||
message_text = message_text_stripped.to_owned();
|
message_text = message_text_stripped.to_owned();
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
message.content = MatrixMessageContent::Text(message_text.trim().to_owned());
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
message.content = MatrixMessageContent::Text(message_text.trim().to_owned());
|
||||||
}
|
}
|
||||||
|
|
||||||
// We only strip `bot_user_prefixes_to_strip`-defined prefixes from messages that mention the bot user.
|
// We only strip `bot_user_prefixes_to_strip`-defined prefixes from messages that mention the bot user.
|
||||||
if !params.bot_user_prefixes_to_strip.is_empty()
|
if !params.bot_user_prefixes_to_strip.is_empty()
|
||||||
&& message.mentioned_users.contains(¶ms.bot_user_id)
|
&& message.mentioned_users.contains(¶ms.bot_user_id)
|
||||||
|
&& let MatrixMessageContent::Text(message_text) = &message.content
|
||||||
{
|
{
|
||||||
if let MatrixMessageContent::Text(message_text) = &message.content {
|
let mut message_text = message_text.clone();
|
||||||
let mut message_text = message_text.clone();
|
|
||||||
|
|
||||||
for prefix in ¶ms.bot_user_prefixes_to_strip {
|
for prefix in ¶ms.bot_user_prefixes_to_strip {
|
||||||
if let Some(message_text_stripped) = message_text.strip_prefix(prefix) {
|
if let Some(message_text_stripped) = message_text.strip_prefix(prefix) {
|
||||||
message_text = message_text_stripped.to_owned();
|
message_text = message_text_stripped.to_owned();
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
message.content = MatrixMessageContent::Text(message_text.trim().to_owned());
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
message.content = MatrixMessageContent::Text(message_text.trim().to_owned());
|
||||||
}
|
}
|
||||||
|
|
||||||
messages_filtered.push(message);
|
messages_filtered.push(message);
|
||||||
@@ -234,6 +234,7 @@ pub async fn convert_matrix_native_event_to_matrix_message(
|
|||||||
MessageType::Text(text_content) => (text_content.body.clone(), false),
|
MessageType::Text(text_content) => (text_content.body.clone(), false),
|
||||||
MessageType::Notice(notice_content) => (notice_content.body.clone(), true),
|
MessageType::Notice(notice_content) => (notice_content.body.clone(), true),
|
||||||
MessageType::Image(image_content) => (image_content.body.clone(), false),
|
MessageType::Image(image_content) => (image_content.body.clone(), false),
|
||||||
|
MessageType::File(file_content) => (file_content.body.clone(), false),
|
||||||
_ => return Ok(None),
|
_ => return Ok(None),
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -291,6 +292,67 @@ pub async fn convert_matrix_native_event_to_matrix_message(
|
|||||||
}));
|
}));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if let MessageType::File(file_content) = &room_message.msgtype {
|
||||||
|
let media_request = mxlink::matrix_sdk::media::MediaRequestParameters {
|
||||||
|
source: file_content.source.to_owned(),
|
||||||
|
format: mxlink::matrix_sdk::media::MediaFormat::File,
|
||||||
|
};
|
||||||
|
|
||||||
|
let file_name = file_content
|
||||||
|
.filename
|
||||||
|
.clone()
|
||||||
|
.unwrap_or(file_content.body.clone());
|
||||||
|
|
||||||
|
let mime_type = file_content
|
||||||
|
.info
|
||||||
|
.as_ref()
|
||||||
|
.and_then(|info| info.mimetype.clone())
|
||||||
|
.and_then(|mimetype| mimetype.parse::<mxlink::mime::Mime>().ok())
|
||||||
|
.unwrap_or_else(|| get_mime_type_from_file_name(&file_name));
|
||||||
|
|
||||||
|
tracing::debug!("Determined mime type {} for file {}", mime_type, file_name);
|
||||||
|
|
||||||
|
if mime_type == mxlink::mime::APPLICATION_OCTET_STREAM {
|
||||||
|
tracing::debug!(
|
||||||
|
"Skipping file {} with unsupported MIME type {}. It will be represented as a text message.",
|
||||||
|
file_name,
|
||||||
|
mime_type,
|
||||||
|
);
|
||||||
|
|
||||||
|
return Ok(Some(MatrixMessage {
|
||||||
|
sender_id: matrix_native_event.sender().to_owned(),
|
||||||
|
content: MatrixMessageContent::Text(format!(
|
||||||
|
"[A file ({}) was attached but skipped because its content type ({}) is not supported. Let the user know.]",
|
||||||
|
file_name, mime_type,
|
||||||
|
)),
|
||||||
|
mentioned_users,
|
||||||
|
timestamp,
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
|
||||||
|
let span = tracing::debug_span!("get_media_content", file_name = %file_name, mime_type = %mime_type);
|
||||||
|
|
||||||
|
let media_bytes = matrix_link
|
||||||
|
.client()
|
||||||
|
.media()
|
||||||
|
.get_media_content(&media_request, true)
|
||||||
|
.instrument(span)
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
tracing::debug!(
|
||||||
|
"Downloaded {} bytes for file {}",
|
||||||
|
media_bytes.len(),
|
||||||
|
file_name
|
||||||
|
);
|
||||||
|
|
||||||
|
return Ok(Some(MatrixMessage {
|
||||||
|
sender_id: matrix_native_event.sender().to_owned(),
|
||||||
|
content: MatrixMessageContent::File(file_content.clone(), mime_type, media_bytes),
|
||||||
|
mentioned_users,
|
||||||
|
timestamp,
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
|
||||||
Ok(Some(MatrixMessage {
|
Ok(Some(MatrixMessage {
|
||||||
sender_id: matrix_native_event.sender().to_owned(),
|
sender_id: matrix_native_event.sender().to_owned(),
|
||||||
content: if is_notice {
|
content: if is_notice {
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
use std::path::PathBuf;
|
use std::path::PathBuf;
|
||||||
|
|
||||||
use mxlink::helpers::encryption::EncryptionKey;
|
use mxlink::helpers::encryption::EncryptionKey;
|
||||||
|
use mxlink::matrix_sdk::ruma::{OwnedDeviceId, OwnedUserId};
|
||||||
use serde::{Deserialize, Deserializer, Serialize};
|
use serde::{Deserialize, Deserializer, Serialize};
|
||||||
|
|
||||||
use crate::{
|
use crate::{
|
||||||
@@ -38,7 +39,7 @@ pub struct Config {
|
|||||||
impl Config {
|
impl Config {
|
||||||
pub fn validate(&self) -> anyhow::Result<()> {
|
pub fn validate(&self) -> anyhow::Result<()> {
|
||||||
self.homeserver.validate()?;
|
self.homeserver.validate()?;
|
||||||
self.user.validate()?;
|
self.user.validate(&self.homeserver.server_name)?;
|
||||||
self.persistence.validate()?;
|
self.persistence.validate()?;
|
||||||
self.room.validate()?;
|
self.room.validate()?;
|
||||||
self.access.validate()?;
|
self.access.validate()?;
|
||||||
@@ -57,6 +58,19 @@ impl Config {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub enum ConfigUserAuth {
|
||||||
|
UserPassword {
|
||||||
|
username: String,
|
||||||
|
password: String,
|
||||||
|
},
|
||||||
|
AccessToken {
|
||||||
|
user_id: OwnedUserId,
|
||||||
|
device_id: OwnedDeviceId,
|
||||||
|
access_token: String,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
#[derive(Debug, Serialize, Deserialize)]
|
#[derive(Debug, Serialize, Deserialize)]
|
||||||
pub struct ConfigHomeserver {
|
pub struct ConfigHomeserver {
|
||||||
pub server_name: String,
|
pub server_name: String,
|
||||||
@@ -88,9 +102,10 @@ impl ConfigHomeserver {
|
|||||||
/// - `Default`: Use the built-in default avatar (null, empty string, or missing in config)
|
/// - `Default`: Use the built-in default avatar (null, empty string, or missing in config)
|
||||||
/// - `Keep`: Don't touch the avatar, keep whatever is already set ("keep" in config)
|
/// - `Keep`: Don't touch the avatar, keep whatever is already set ("keep" in config)
|
||||||
/// - `Custom(String)`: Use a custom avatar from the specified file path
|
/// - `Custom(String)`: Use a custom avatar from the specified file path
|
||||||
#[derive(Debug, Clone, PartialEq, Serialize)]
|
#[derive(Debug, Clone, Default, PartialEq, Serialize)]
|
||||||
pub enum Avatar {
|
pub enum Avatar {
|
||||||
/// Use the built-in default avatar
|
/// Use the built-in default avatar
|
||||||
|
#[default]
|
||||||
Default,
|
Default,
|
||||||
/// Keep the current avatar, don't change it
|
/// Keep the current avatar, don't change it
|
||||||
Keep,
|
Keep,
|
||||||
@@ -98,12 +113,6 @@ pub enum Avatar {
|
|||||||
Custom(String),
|
Custom(String),
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Default for Avatar {
|
|
||||||
fn default() -> Self {
|
|
||||||
Avatar::Default
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<'de> Deserialize<'de> for Avatar {
|
impl<'de> Deserialize<'de> for Avatar {
|
||||||
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
|
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
|
||||||
where
|
where
|
||||||
@@ -132,7 +141,15 @@ impl Avatar {
|
|||||||
#[derive(Debug, Serialize, Deserialize)]
|
#[derive(Debug, Serialize, Deserialize)]
|
||||||
pub struct ConfigUser {
|
pub struct ConfigUser {
|
||||||
pub mxid_localpart: String,
|
pub mxid_localpart: String,
|
||||||
pub password: String,
|
|
||||||
|
#[serde(default)]
|
||||||
|
pub password: Option<String>,
|
||||||
|
|
||||||
|
#[serde(default)]
|
||||||
|
pub access_token: Option<String>,
|
||||||
|
|
||||||
|
#[serde(default)]
|
||||||
|
pub device_id: Option<String>,
|
||||||
|
|
||||||
#[serde(default = "super::defaults::name")]
|
#[serde(default = "super::defaults::name")]
|
||||||
pub name: String,
|
pub name: String,
|
||||||
@@ -145,7 +162,7 @@ pub struct ConfigUser {
|
|||||||
}
|
}
|
||||||
|
|
||||||
impl ConfigUser {
|
impl ConfigUser {
|
||||||
pub fn validate(&self) -> anyhow::Result<()> {
|
pub fn validate(&self, homeserver_server_name: &str) -> anyhow::Result<()> {
|
||||||
if self.mxid_localpart.is_empty() {
|
if self.mxid_localpart.is_empty() {
|
||||||
return Err(anyhow::anyhow!(
|
return Err(anyhow::anyhow!(
|
||||||
"The user.mxid_localpart ({}) configuration must be set",
|
"The user.mxid_localpart ({}) configuration must be set",
|
||||||
@@ -153,12 +170,7 @@ impl ConfigUser {
|
|||||||
));
|
));
|
||||||
}
|
}
|
||||||
|
|
||||||
if self.password.is_empty() {
|
self.auth_config(homeserver_server_name)?;
|
||||||
return Err(anyhow::anyhow!(
|
|
||||||
"The user.password ({}) configuration must be set",
|
|
||||||
super::env::BAIBOT_USER_PASSWORD
|
|
||||||
));
|
|
||||||
}
|
|
||||||
|
|
||||||
if self.name.is_empty() {
|
if self.name.is_empty() {
|
||||||
return Err(anyhow::anyhow!(
|
return Err(anyhow::anyhow!(
|
||||||
@@ -171,6 +183,57 @@ impl ConfigUser {
|
|||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn auth_config(&self, homeserver_server_name: &str) -> anyhow::Result<ConfigUserAuth> {
|
||||||
|
let password = self.password.as_deref().filter(|value| !value.is_empty());
|
||||||
|
let access_token = self
|
||||||
|
.access_token
|
||||||
|
.as_deref()
|
||||||
|
.filter(|value| !value.is_empty());
|
||||||
|
|
||||||
|
match (password, access_token) {
|
||||||
|
(Some(_), Some(_)) => Err(anyhow::anyhow!(
|
||||||
|
"Set exactly one authentication method: either user.password ({}) OR user.access_token ({}) + user.device_id ({})",
|
||||||
|
super::env::BAIBOT_USER_PASSWORD,
|
||||||
|
super::env::BAIBOT_USER_ACCESS_TOKEN,
|
||||||
|
super::env::BAIBOT_USER_DEVICE_ID
|
||||||
|
)),
|
||||||
|
(None, None) => Err(anyhow::anyhow!(
|
||||||
|
"Set one authentication method: either user.password ({}) OR user.access_token ({}) + user.device_id ({})",
|
||||||
|
super::env::BAIBOT_USER_PASSWORD,
|
||||||
|
super::env::BAIBOT_USER_ACCESS_TOKEN,
|
||||||
|
super::env::BAIBOT_USER_DEVICE_ID
|
||||||
|
)),
|
||||||
|
(Some(password), None) => Ok(ConfigUserAuth::UserPassword {
|
||||||
|
username: self.mxid_localpart.to_owned(),
|
||||||
|
password: password.to_owned(),
|
||||||
|
}),
|
||||||
|
(None, Some(access_token)) => {
|
||||||
|
let device_id = self
|
||||||
|
.device_id
|
||||||
|
.as_deref()
|
||||||
|
.filter(|value| !value.is_empty())
|
||||||
|
.ok_or_else(|| {
|
||||||
|
anyhow::anyhow!(
|
||||||
|
"user.device_id ({}) must be set when using access token authentication",
|
||||||
|
super::env::BAIBOT_USER_DEVICE_ID
|
||||||
|
)
|
||||||
|
})?;
|
||||||
|
|
||||||
|
let user_id = OwnedUserId::try_from(format!(
|
||||||
|
"@{}:{}",
|
||||||
|
self.mxid_localpart, homeserver_server_name
|
||||||
|
))
|
||||||
|
.map_err(|e| anyhow::anyhow!("Invalid user ID: {e}"))?;
|
||||||
|
|
||||||
|
Ok(ConfigUserAuth::AccessToken {
|
||||||
|
user_id,
|
||||||
|
device_id: OwnedDeviceId::from(device_id),
|
||||||
|
access_token: access_token.to_owned(),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Default, Serialize, Deserialize)]
|
#[derive(Debug, Default, Serialize, Deserialize)]
|
||||||
@@ -181,13 +244,13 @@ pub struct ConfigUserEncryption {
|
|||||||
|
|
||||||
impl ConfigUserEncryption {
|
impl ConfigUserEncryption {
|
||||||
pub fn validate(&self) -> anyhow::Result<()> {
|
pub fn validate(&self) -> anyhow::Result<()> {
|
||||||
if let Some(passphrase) = &self.recovery_passphrase {
|
if let Some(passphrase) = &self.recovery_passphrase
|
||||||
if passphrase.is_empty() {
|
&& passphrase.is_empty()
|
||||||
return Err(anyhow::anyhow!(
|
{
|
||||||
"The user.encryption.recovery_passphrase ({}) configuration must either be null or set to a non-empty passphrase",
|
return Err(anyhow::anyhow!(
|
||||||
super::env::BAIBOT_USER_ENCRYPTION_RECOVERY_PASSPHRASE
|
"The user.encryption.recovery_passphrase ({}) configuration must either be null or set to a non-empty passphrase",
|
||||||
));
|
super::env::BAIBOT_USER_ENCRYPTION_RECOVERY_PASSPHRASE
|
||||||
}
|
));
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
@@ -473,3 +536,7 @@ impl TryInto<GlobalConfig> for ConfigInitialGlobalConfig {
|
|||||||
Ok(entity)
|
Ok(entity)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
#[path = "config_tests.rs"]
|
||||||
|
mod config_tests;
|
||||||
|
|||||||
117
src/entity/cfg/config_tests.rs
Normal file
117
src/entity/cfg/config_tests.rs
Normal file
@@ -0,0 +1,117 @@
|
|||||||
|
use super::{Avatar, ConfigUser, ConfigUserAuth, ConfigUserEncryption};
|
||||||
|
use crate::entity::cfg::env;
|
||||||
|
|
||||||
|
fn base_user() -> ConfigUser {
|
||||||
|
ConfigUser {
|
||||||
|
mxid_localpart: "baibot".to_owned(),
|
||||||
|
password: None,
|
||||||
|
access_token: None,
|
||||||
|
device_id: None,
|
||||||
|
name: "baibot".to_owned(),
|
||||||
|
encryption: ConfigUserEncryption {
|
||||||
|
recovery_passphrase: None,
|
||||||
|
recovery_reset_allowed: false,
|
||||||
|
},
|
||||||
|
avatar: Avatar::Default,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn auth_config_uses_password_mode() {
|
||||||
|
let mut user = base_user();
|
||||||
|
user.password = Some("secret".to_owned());
|
||||||
|
|
||||||
|
let auth = user
|
||||||
|
.auth_config("example.com")
|
||||||
|
.expect("password auth should be valid");
|
||||||
|
|
||||||
|
match auth {
|
||||||
|
ConfigUserAuth::UserPassword { username, password } => {
|
||||||
|
assert_eq!(username, "baibot");
|
||||||
|
assert_eq!(password, "secret");
|
||||||
|
}
|
||||||
|
ConfigUserAuth::AccessToken { .. } => {
|
||||||
|
panic!("expected password auth mode");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn auth_config_uses_access_token_mode() {
|
||||||
|
let mut user = base_user();
|
||||||
|
user.access_token = Some("token123".to_owned());
|
||||||
|
user.device_id = Some("DEVICE1".to_owned());
|
||||||
|
|
||||||
|
let auth = user
|
||||||
|
.auth_config("example.com")
|
||||||
|
.expect("access token auth should be valid");
|
||||||
|
|
||||||
|
match auth {
|
||||||
|
ConfigUserAuth::AccessToken {
|
||||||
|
user_id,
|
||||||
|
device_id,
|
||||||
|
access_token,
|
||||||
|
} => {
|
||||||
|
assert_eq!(user_id.as_str(), "@baibot:example.com");
|
||||||
|
assert_eq!(device_id.as_str(), "DEVICE1");
|
||||||
|
assert_eq!(access_token, "token123");
|
||||||
|
}
|
||||||
|
ConfigUserAuth::UserPassword { .. } => {
|
||||||
|
panic!("expected access token auth mode");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn auth_config_rejects_both_auth_methods() {
|
||||||
|
let mut user = base_user();
|
||||||
|
user.password = Some("secret".to_owned());
|
||||||
|
user.access_token = Some("token123".to_owned());
|
||||||
|
user.device_id = Some("DEVICE1".to_owned());
|
||||||
|
|
||||||
|
let err = user
|
||||||
|
.auth_config("example.com")
|
||||||
|
.expect_err("both auth methods should be rejected");
|
||||||
|
|
||||||
|
assert!(
|
||||||
|
err.to_string()
|
||||||
|
.contains("exactly one authentication method")
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn auth_config_rejects_missing_auth() {
|
||||||
|
let user = base_user();
|
||||||
|
|
||||||
|
let err = user
|
||||||
|
.auth_config("example.com")
|
||||||
|
.expect_err("missing auth should be rejected");
|
||||||
|
|
||||||
|
assert!(err.to_string().contains("Set one authentication method"));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn auth_config_rejects_access_token_without_device_id() {
|
||||||
|
let mut user = base_user();
|
||||||
|
user.access_token = Some("token123".to_owned());
|
||||||
|
|
||||||
|
let err = user
|
||||||
|
.auth_config("example.com")
|
||||||
|
.expect_err("access token mode without device_id should be rejected");
|
||||||
|
|
||||||
|
assert!(err.to_string().contains(env::BAIBOT_USER_DEVICE_ID));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn auth_config_treats_empty_strings_as_unset() {
|
||||||
|
let mut user = base_user();
|
||||||
|
user.password = Some(String::new());
|
||||||
|
user.access_token = Some(String::new());
|
||||||
|
user.device_id = Some(String::new());
|
||||||
|
|
||||||
|
let err = user
|
||||||
|
.auth_config("example.com")
|
||||||
|
.expect_err("empty auth values should be treated as unset");
|
||||||
|
|
||||||
|
assert!(err.to_string().contains("Set one authentication method"));
|
||||||
|
}
|
||||||
@@ -5,6 +5,8 @@ pub const BAIBOT_HOMESERVER_URL: &str = "BAIBOT_HOMESERVER_URL";
|
|||||||
|
|
||||||
pub const BAIBOT_USER_MXID_LOCALPART: &str = "BAIBOT_USER_MXID_LOCALPART";
|
pub const BAIBOT_USER_MXID_LOCALPART: &str = "BAIBOT_USER_MXID_LOCALPART";
|
||||||
pub const BAIBOT_USER_PASSWORD: &str = "BAIBOT_USER_PASSWORD";
|
pub const BAIBOT_USER_PASSWORD: &str = "BAIBOT_USER_PASSWORD";
|
||||||
|
pub const BAIBOT_USER_ACCESS_TOKEN: &str = "BAIBOT_USER_ACCESS_TOKEN";
|
||||||
|
pub const BAIBOT_USER_DEVICE_ID: &str = "BAIBOT_USER_DEVICE_ID";
|
||||||
pub const BAIBOT_USER_NAME: &str = "BAIBOT_USER_NAME";
|
pub const BAIBOT_USER_NAME: &str = "BAIBOT_USER_NAME";
|
||||||
pub const BAIBOT_USER_AVATAR: &str = "BAIBOT_USER_AVATAR";
|
pub const BAIBOT_USER_AVATAR: &str = "BAIBOT_USER_AVATAR";
|
||||||
pub const BAIBOT_USER_ENCRYPTION_RECOVERY_PASSPHRASE: &str =
|
pub const BAIBOT_USER_ENCRYPTION_RECOVERY_PASSPHRASE: &str =
|
||||||
|
|||||||
@@ -2,4 +2,4 @@ mod config;
|
|||||||
pub mod defaults;
|
pub mod defaults;
|
||||||
pub mod env;
|
pub mod env;
|
||||||
|
|
||||||
pub use config::{Avatar, Config};
|
pub use config::{Avatar, Config, ConfigUserAuth};
|
||||||
|
|||||||
@@ -1,5 +1,6 @@
|
|||||||
use mxlink::matrix_sdk::ruma::events::room::message::{
|
use mxlink::matrix_sdk::ruma::events::room::message::{
|
||||||
AudioMessageEventContent, ImageMessageEventContent, MessageType, TextMessageEventContent,
|
AudioMessageEventContent, FileMessageEventContent, ImageMessageEventContent, MessageType,
|
||||||
|
TextMessageEventContent,
|
||||||
};
|
};
|
||||||
use mxlink::matrix_sdk::ruma::{OwnedEventId, OwnedUserId};
|
use mxlink::matrix_sdk::ruma::{OwnedEventId, OwnedUserId};
|
||||||
|
|
||||||
@@ -30,6 +31,7 @@ pub enum MessagePayload {
|
|||||||
Text(TextMessageEventContent),
|
Text(TextMessageEventContent),
|
||||||
Audio(AudioMessageEventContent),
|
Audio(AudioMessageEventContent),
|
||||||
Image(ImageMessageEventContent),
|
Image(ImageMessageEventContent),
|
||||||
|
File(FileMessageEventContent),
|
||||||
|
|
||||||
Reaction {
|
Reaction {
|
||||||
key: String,
|
key: String,
|
||||||
@@ -57,6 +59,7 @@ impl TryInto<MessagePayload> for MessageType {
|
|||||||
MessagePayload::Audio(audio_content)
|
MessagePayload::Audio(audio_content)
|
||||||
}
|
}
|
||||||
MessageType::Image(image_content) => MessagePayload::Image(image_content),
|
MessageType::Image(image_content) => MessagePayload::Image(image_content),
|
||||||
|
MessageType::File(file_content) => MessagePayload::File(file_content),
|
||||||
other => {
|
other => {
|
||||||
return Err(format!("Unsupported message type: {:?}", other));
|
return Err(format!("Unsupported message type: {:?}", other));
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -5,8 +5,9 @@ use super::roomconfig::RoomConfig;
|
|||||||
|
|
||||||
use crate::entity::roomconfig::{
|
use crate::entity::roomconfig::{
|
||||||
SpeechToTextFlowType, SpeechToTextMessageTypeForNonThreadedOnlyTranscribedMessages,
|
SpeechToTextFlowType, SpeechToTextMessageTypeForNonThreadedOnlyTranscribedMessages,
|
||||||
TextGenerationAutoUsage, TextGenerationPrefixRequirementType, TextToSpeechBotMessagesFlowType,
|
TextGenerationAutoUsage, TextGenerationPrefixRequirementType, TextGenerationSenderContextMode,
|
||||||
TextToSpeechUserMessagesFlowType, defaults as roomconfig_defaults,
|
TextToSpeechBotMessagesFlowType, TextToSpeechUserMessagesFlowType,
|
||||||
|
defaults as roomconfig_defaults,
|
||||||
};
|
};
|
||||||
|
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
@@ -135,6 +136,20 @@ impl RoomConfigContext {
|
|||||||
.unwrap_or(false)
|
.unwrap_or(false)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn text_generation_sender_context_mode(&self) -> TextGenerationSenderContextMode {
|
||||||
|
self.room_config
|
||||||
|
.settings
|
||||||
|
.text_generation
|
||||||
|
.sender_context_mode
|
||||||
|
.or({
|
||||||
|
self.global_config
|
||||||
|
.fallback_room_settings
|
||||||
|
.text_generation
|
||||||
|
.sender_context_mode
|
||||||
|
})
|
||||||
|
.unwrap_or(roomconfig_defaults::TEXT_GENERATION_SENDER_CONTEXT_MODE)
|
||||||
|
}
|
||||||
|
|
||||||
pub fn text_generation_prefix_requirement_type(&self) -> TextGenerationPrefixRequirementType {
|
pub fn text_generation_prefix_requirement_type(&self) -> TextGenerationPrefixRequirementType {
|
||||||
self.room_config
|
self.room_config
|
||||||
.settings
|
.settings
|
||||||
|
|||||||
@@ -1,5 +1,7 @@
|
|||||||
use super::{SpeechToTextFlowType, SpeechToTextMessageTypeForNonThreadedOnlyTranscribedMessages};
|
use super::{SpeechToTextFlowType, SpeechToTextMessageTypeForNonThreadedOnlyTranscribedMessages};
|
||||||
use super::{TextGenerationAutoUsage, TextGenerationPrefixRequirementType};
|
use super::{
|
||||||
|
TextGenerationAutoUsage, TextGenerationPrefixRequirementType, TextGenerationSenderContextMode,
|
||||||
|
};
|
||||||
use super::{TextToSpeechBotMessagesFlowType, TextToSpeechUserMessagesFlowType};
|
use super::{TextToSpeechBotMessagesFlowType, TextToSpeechUserMessagesFlowType};
|
||||||
|
|
||||||
pub const TEXT_GENERATION_PREFIX_REQUIREMENT_TYPE: TextGenerationPrefixRequirementType =
|
pub const TEXT_GENERATION_PREFIX_REQUIREMENT_TYPE: TextGenerationPrefixRequirementType =
|
||||||
@@ -7,6 +9,9 @@ pub const TEXT_GENERATION_PREFIX_REQUIREMENT_TYPE: TextGenerationPrefixRequireme
|
|||||||
|
|
||||||
pub const TEXT_GENERATION_AUTO_USAGE: TextGenerationAutoUsage = TextGenerationAutoUsage::Always;
|
pub const TEXT_GENERATION_AUTO_USAGE: TextGenerationAutoUsage = TextGenerationAutoUsage::Always;
|
||||||
|
|
||||||
|
pub const TEXT_GENERATION_SENDER_CONTEXT_MODE: TextGenerationSenderContextMode =
|
||||||
|
TextGenerationSenderContextMode::Disabled;
|
||||||
|
|
||||||
pub const TEXT_TO_SPEECH_BOT_MESSAGES_FLOW_TYPE: TextToSpeechBotMessagesFlowType =
|
pub const TEXT_TO_SPEECH_BOT_MESSAGES_FLOW_TYPE: TextToSpeechBotMessagesFlowType =
|
||||||
TextToSpeechBotMessagesFlowType::OnDemandForVoice;
|
TextToSpeechBotMessagesFlowType::OnDemandForVoice;
|
||||||
|
|
||||||
|
|||||||
@@ -16,7 +16,9 @@ pub use handler::RoomSettingsHandler;
|
|||||||
pub use speech_to_text::{
|
pub use speech_to_text::{
|
||||||
SpeechToTextFlowType, SpeechToTextMessageTypeForNonThreadedOnlyTranscribedMessages,
|
SpeechToTextFlowType, SpeechToTextMessageTypeForNonThreadedOnlyTranscribedMessages,
|
||||||
};
|
};
|
||||||
pub use text_generation::{TextGenerationAutoUsage, TextGenerationPrefixRequirementType};
|
pub use text_generation::{
|
||||||
|
TextGenerationAutoUsage, TextGenerationPrefixRequirementType, TextGenerationSenderContextMode,
|
||||||
|
};
|
||||||
pub use text_to_speech::{TextToSpeechBotMessagesFlowType, TextToSpeechUserMessagesFlowType};
|
pub use text_to_speech::{TextToSpeechBotMessagesFlowType, TextToSpeechUserMessagesFlowType};
|
||||||
|
|
||||||
#[derive(Clone, Debug, Deserialize, Serialize, EventContent)]
|
#[derive(Clone, Debug, Deserialize, Serialize, EventContent)]
|
||||||
|
|||||||
@@ -14,6 +14,9 @@ pub struct RoomSettingsTextGeneration {
|
|||||||
/// When enabled, the bot will automatically tokenize messages and try to shorten the message context intelligently.
|
/// When enabled, the bot will automatically tokenize messages and try to shorten the message context intelligently.
|
||||||
pub context_management_enabled: Option<bool>,
|
pub context_management_enabled: Option<bool>,
|
||||||
|
|
||||||
|
/// Controls how each message in the conversation context is annotated with sender metadata.
|
||||||
|
pub sender_context_mode: Option<TextGenerationSenderContextMode>,
|
||||||
|
|
||||||
/// Allows customizing the system prompt that the agent would use
|
/// Allows customizing the system prompt that the agent would use
|
||||||
pub prompt_override: Option<String>,
|
pub prompt_override: Option<String>,
|
||||||
|
|
||||||
@@ -111,3 +114,46 @@ impl std::fmt::Display for TextGenerationAutoUsage {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Clone, Copy, Debug, Deserialize, Serialize, PartialEq)]
|
||||||
|
pub enum TextGenerationSenderContextMode {
|
||||||
|
#[serde(rename = "disabled")]
|
||||||
|
Disabled,
|
||||||
|
|
||||||
|
#[serde(rename = "matrix_user_id")]
|
||||||
|
MatrixUserId,
|
||||||
|
|
||||||
|
#[serde(rename = "matrix_user_id_and_timestamp")]
|
||||||
|
MatrixUserIdAndTimestamp,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl TextGenerationSenderContextMode {
|
||||||
|
pub fn choices() -> Vec<Self> {
|
||||||
|
vec![
|
||||||
|
Self::Disabled,
|
||||||
|
Self::MatrixUserId,
|
||||||
|
Self::MatrixUserIdAndTimestamp,
|
||||||
|
]
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn from_str(s: &str) -> Option<Self> {
|
||||||
|
match s {
|
||||||
|
"disabled" => Some(Self::Disabled),
|
||||||
|
"matrix_user_id" => Some(Self::MatrixUserId),
|
||||||
|
"matrix_user_id_and_timestamp" => Some(Self::MatrixUserIdAndTimestamp),
|
||||||
|
_ => None,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl std::fmt::Display for TextGenerationSenderContextMode {
|
||||||
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||||
|
match self {
|
||||||
|
TextGenerationSenderContextMode::Disabled => write!(f, "disabled"),
|
||||||
|
TextGenerationSenderContextMode::MatrixUserId => write!(f, "matrix_user_id"),
|
||||||
|
TextGenerationSenderContextMode::MatrixUserIdAndTimestamp => {
|
||||||
|
write!(f, "matrix_user_id_and_timestamp")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -6,8 +6,8 @@ use mxlink::helpers::account_data_config::RoomConfigManager as AccountDataRoomCo
|
|||||||
pub use entity::{RoomConfig, RoomConfigCarrierContent, RoomSettings, RoomSettingsHandler};
|
pub use entity::{RoomConfig, RoomConfigCarrierContent, RoomSettings, RoomSettingsHandler};
|
||||||
pub use entity::{
|
pub use entity::{
|
||||||
SpeechToTextFlowType, SpeechToTextMessageTypeForNonThreadedOnlyTranscribedMessages,
|
SpeechToTextFlowType, SpeechToTextMessageTypeForNonThreadedOnlyTranscribedMessages,
|
||||||
TextGenerationAutoUsage, TextGenerationPrefixRequirementType, TextToSpeechBotMessagesFlowType,
|
TextGenerationAutoUsage, TextGenerationPrefixRequirementType, TextGenerationSenderContextMode,
|
||||||
TextToSpeechUserMessagesFlowType,
|
TextToSpeechBotMessagesFlowType, TextToSpeechUserMessagesFlowType,
|
||||||
};
|
};
|
||||||
|
|
||||||
pub type RoomConfigurationManager =
|
pub type RoomConfigurationManager =
|
||||||
|
|||||||
@@ -249,6 +249,13 @@ pub fn status_text_generation_entry_context_management(value: bool, set_where: &
|
|||||||
format!("- ♻️ Context management: `{}` ({})\n", value, set_where)
|
format!("- ♻️ Context management: `{}` ({})\n", value, set_where)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn status_text_generation_entry_sender_context(
|
||||||
|
value: impl std::fmt::Display,
|
||||||
|
set_where: &str,
|
||||||
|
) -> String {
|
||||||
|
format!("- 👤 Sender context mode: `{}` ({})\n", value, set_where)
|
||||||
|
}
|
||||||
|
|
||||||
pub fn status_text_generation_entry_prompt(value: &str, set_where: &str) -> String {
|
pub fn status_text_generation_entry_prompt(value: &str, set_where: &str) -> String {
|
||||||
let value = value.trim();
|
let value = value.trim();
|
||||||
|
|
||||||
|
|||||||
@@ -132,6 +132,18 @@ pub fn text_generation_context_management_intro() -> String {
|
|||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn text_generation_sender_context_heading() -> &'static str {
|
||||||
|
"👤 Sender Context Mode"
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn text_generation_sender_context_intro() -> String {
|
||||||
|
format!(
|
||||||
|
"{}\n{}",
|
||||||
|
"Controls whether the bot attaches sender information to conversation messages before sending them to the model.",
|
||||||
|
"`disabled` leaves messages unchanged, `matrix_user_id` adds `[sender=@alice:example.com]`, and `matrix_user_id_and_timestamp` adds `[sender=@alice:example.com sent_at=2026-03-23T14:30:00Z]`. Enabling this sends Matrix user IDs, and optionally timestamps, to the model provider.",
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
pub fn text_generation_prompt_override_heading() -> &'static str {
|
pub fn text_generation_prompt_override_heading() -> &'static str {
|
||||||
"⌨️ Prompt Override"
|
"⌨️ Prompt Override"
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -17,18 +17,7 @@ pub fn get_file_extension(mime_type: &mime::Mime) -> String {
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub fn get_mime_type_from_file_name(file_name: &str) -> mime::Mime {
|
pub fn get_mime_type_from_file_name(file_name: &str) -> mime::Mime {
|
||||||
let extension = file_name.rsplit('.').next().unwrap_or("");
|
mime_guess::from_path(file_name)
|
||||||
|
.first()
|
||||||
match extension.to_lowercase().as_str() {
|
.unwrap_or(mime::APPLICATION_OCTET_STREAM)
|
||||||
"jpg" | "jpeg" => mime::IMAGE_JPEG,
|
|
||||||
"png" => mime::IMAGE_PNG,
|
|
||||||
"gif" => mime::IMAGE_GIF,
|
|
||||||
"webp" => "image/webp".parse().unwrap(),
|
|
||||||
"svg" => mime::IMAGE_SVG,
|
|
||||||
"tiff" | "tif" => "image/tiff".parse().unwrap(),
|
|
||||||
"bmp" => "image/bmp".parse().unwrap(),
|
|
||||||
"heic" | "heif" => "image/heic".parse().unwrap(),
|
|
||||||
"avif" => "image/avif".parse().unwrap(),
|
|
||||||
_ => mime::APPLICATION_OCTET_STREAM,
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user