summaryrefslogtreecommitdiffstats
path: root/src/main.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2025-01-22 20:51:10 +0800
committerGitHub <noreply@github.com>2025-01-22 20:51:10 +0800
commite0417e8d5bebe25476aaafa22a9ee23d9bd61457 (patch)
tree8edcd3aa9bcd1b012a3a429ad6240e6186caef36 /src/main.rs
parentdf4440a2a049d26c61a540d3254cd885345fbd6c (diff)
downloadaichat-e0417e8d5bebe25476aaafa22a9ee23d9bd61457.tar.gz
feat: add `--sync-models` cli option (#1114)
Diffstat (limited to 'src/main.rs')
-rw-r--r--src/main.rs6
1 files changed, 6 insertions, 0 deletions
diff --git a/src/main.rs b/src/main.rs
index e30efde..5251bad 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -46,6 +46,7 @@ async fn main() -> Result<()> {
WorkingMode::Cmd
};
let info_flag = cli.info
+ || cli.sync_models
|| cli.list_models
|| cli.list_roles
|| cli.list_agents
@@ -64,6 +65,11 @@ async fn main() -> Result<()> {
async fn run(config: GlobalConfig, cli: Cli, text: Option<String>) -> Result<()> {
let abort_signal = create_abort_signal();
+ if cli.sync_models {
+ let url = config.read().sync_models_url();
+ return Config::sync_models(&url, abort_signal.clone()).await;
+ }
+
if cli.list_models {
for model in list_models(&config.read(), ModelType::Chat) {
println!("{}", model.id());