From e0417e8d5bebe25476aaafa22a9ee23d9bd61457 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Wed, 22 Jan 2025 20:51:10 +0800
Subject: feat: add `--sync-models` cli option (#1114)

---
 src/main.rs | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'src/main.rs')

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());
-- 
cgit v1.2.3