diff options
| -rw-r--r-- | README.md | 4 | ||||
| -rw-r--r-- | config.example.yaml | 16 | ||||
| -rw-r--r-- | src/config/mod.rs | 2 | ||||
| -rw-r--r-- | src/config/role.rs | 1 | ||||
| -rw-r--r-- | src/repl/mod.rs | 12 |
5 files changed, 22 insertions, 13 deletions
@@ -4,7 +4,7 @@ [](https://crates.io/crates/aichat) [](https://discord.gg/mr3ZZUB9hG) -AIChat is an all-in-one AI CLI tool featuring Chat-REPL, RAG, Function Calling, AI Agents, and more. +AIChat is an all-in-one AI CLI tool featuring Chat-REPL, Shell Assistant, RAG, Function Calling, AI Agents, and more.  @@ -45,7 +45,7 @@ AIChat is an all-in-one AI CLI tool featuring Chat-REPL, RAG, Function Calling, - Moonshot (paid, function-calling) - Deepseek (paid) - ZhipuAI: GLM-4 (paid, vision, function-calling) -- LingYiWanWu: Yi-Large (paid, vision) +- LingYiWanWu: Yi-* (paid, vision) - Other openAI-compatible platforms ## Install diff --git a/config.example.yaml b/config.example.yaml index 318db2d..90c480d 100644 --- a/config.example.yaml +++ b/config.example.yaml @@ -1,14 +1,14 @@ # ---- llm ---- model: openai:gpt-3.5-turbo # Specify the language model to use -temperature: null # Set default temperature parameter -top_p: null # Set default top-p parameter +temperature: null # Set default temperature parameter, range (0, 2) +top_p: null # Set default top-p parameter, range (0, 1) # ---- apperence ---- highlight: true # Controls syntax highlighting light_theme: false # Activates a light color theme when true. ENV: AICHAT_LIGHT_THEME # Custom REPL prompt, see https://github.com/sigoden/aichat/wiki/Custom-REPL-Prompt for more detils left_prompt: - '{color.green}{?session {?agent {agent}#}{session}{?role /}}{!session {?agent {agent}}}{role}{?rag @{rag}}{color.cyan}{?session )}{!session >}{color.reset} ' + '{color.green}{?session {?agent {agent}>}{session}{?role /}}{!session {?agent {agent}>}}{role}{?rag @{rag}}{color.cyan}{?session )}{!session >}{color.reset} ' right_prompt: '{color.purple}{?session {?consume_tokens {consume_tokens}({consume_percent}%)}{!consume_tokens {consume_tokens}}}{color.reset}' @@ -79,17 +79,17 @@ clients: # - type: xxxx # name: xxxx # Only use it to distinguish clients with the same client type. Optional # models: - # - name: xxxx + # - name: xxxx # Chat model # max_input_tokens: 100000 # supports_vision: true # supports_function_calling: true - # - name: xxxx - # type: embedding # Embedding model + # - name: xxxx # Embedding model + # type: embedding # max_input_tokens: 2048 # default_chunk_size: 2000 # max_batch_size: 100 - # - name: xxxx - # type: rerank # Rerank model + # - name: xxxx # Rerank model + # type: rerank # max_input_tokens: 2048 # patches: # <regex>: # The regex to match model names, e.g. '.*' 'gpt-4o' 'gpt-4o|gpt-4-.*' diff --git a/src/config/mod.rs b/src/config/mod.rs index 2bc8d5f..4c0e1e7 100644 --- a/src/config/mod.rs +++ b/src/config/mod.rs @@ -75,7 +75,7 @@ And answer according to the language of the user's question. Given the context information, answer the query. Query: __INPUT__"#; -const LEFT_PROMPT: &str = "{color.green}{?session {?agent {agent}#}{session}{?role /}}{!session {?agent {agent}}}{role}{?rag @{rag}}{color.cyan}{?session )}{!session >}{color.reset} "; +const LEFT_PROMPT: &str = "{color.green}{?session {?agent {agent}>}{session}{?role /}}{!session {?agent {agent}>}}{role}{?rag @{rag}}{color.cyan}{?session )}{!session >}{color.reset} "; const RIGHT_PROMPT: &str = "{color.purple}{?session {?consume_tokens {consume_tokens}({consume_percent}%)}{!consume_tokens {consume_tokens}}}{color.reset}"; #[derive(Debug, Clone, Deserialize)] diff --git a/src/config/role.rs b/src/config/role.rs index 0dbfa00..29d3f3b 100644 --- a/src/config/role.rs +++ b/src/config/role.rs @@ -247,7 +247,6 @@ impl RoleLike for Role { } fn set_model(&mut self, model: &Model) { - self.model_id = Some(model.id()); self.model = model.clone(); } diff --git a/src/repl/mod.rs b/src/repl/mod.rs index 1df3fdb..8083820 100644 --- a/src/repl/mod.rs +++ b/src/repl/mod.rs @@ -123,7 +123,7 @@ lazy_static! { "Include files with the message", AssertState::pass() ), - ReplCommand::new(".continue", "Continue response", AssertState::pass()), + ReplCommand::new(".continue", "Continue the response", AssertState::pass()), ReplCommand::new( ".regenerate", "Regenerate the last response", @@ -443,6 +443,16 @@ Type ".help" for additional help. KeyCode::Enter, ReedlineEvent::Edit(vec![EditCommand::InsertNewline]), ); + keybindings.add_binding( + KeyModifiers::SHIFT, + KeyCode::Enter, + ReedlineEvent::Edit(vec![EditCommand::InsertNewline]), + ); + keybindings.add_binding( + KeyModifiers::ALT, + KeyCode::Enter, + ReedlineEvent::Edit(vec![EditCommand::InsertNewline]), + ); } fn create_edit_mode(config: &GlobalConfig) -> Box<dyn EditMode> { |