diff options
| author | sigoden <sigoden@gmail.com> | 2024-03-02 21:11:28 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-03-02 21:11:28 +0800 |
| commit | 3c16aff59145134885f32056fa3be9bd6bed2cce (patch) | |
| tree | f285ee126374b8815840815c7334e1d48fae2791 /src/config/mod.rs | |
| parent | d275c33632271bf9cdbc01ef005deb9ff851370b (diff) | |
| download | aichat-3c16aff59145134885f32056fa3be9bd6bed2cce.tar.gz | |
feat: add `-c/--code` to generate only code (#327)
Diffstat (limited to 'src/config/mod.rs')
| -rw-r--r-- | src/config/mod.rs | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/config/mod.rs b/src/config/mod.rs index f8d97c2..6b99483 100644 --- a/src/config/mod.rs +++ b/src/config/mod.rs @@ -288,6 +288,11 @@ impl Config { self.set_role_obj(role) } + pub fn set_code_role(&mut self) -> Result<()> { + let role = Role::for_code(); + self.set_role_obj(role) + } + pub fn set_role_obj(&mut self, role: Role) -> Result<()> { if let Some(session) = self.session.as_mut() { session.update_role(Some(role.clone()))?; |