summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2023-06-19 08:44:26 +0800
committerGitHub <noreply@github.com>2023-06-19 08:44:26 +0800
commitec51b84290215d11d5045de8baa9b89f2447b1e9 (patch)
treebd93eb96b8695dbe55236377bdbc50770ab13164
parent00ce1575715329a21726a506b13c0c2aef7f3f82 (diff)
downloadaichat-ec51b84290215d11d5045de8baa9b89f2447b1e9.tar.gz
feat: add new model gpt-3.5-turbo-16k (#126)
-rw-r--r--README.md1
-rw-r--r--src/config/mod.rs3
2 files changed, 3 insertions, 1 deletions
diff --git a/README.md b/README.md
index 1f3d32b..7c00630 100644
--- a/README.md
+++ b/README.md
@@ -183,6 +183,7 @@ dry_run false
> .model gpt-4
> .model gpt-4-32k
> .model gpt-3.5-turbo
+> .model gpt-3.5-turbo-16k
```
### `.prompt` - use GPT prompt
diff --git a/src/config/mod.rs b/src/config/mod.rs
index 900f129..aec9c93 100644
--- a/src/config/mod.rs
+++ b/src/config/mod.rs
@@ -23,10 +23,11 @@ use std::{
sync::Arc,
};
-pub const MODELS: [(&str, usize); 3] = [
+pub const MODELS: [(&str, usize); 4] = [
("gpt-4", 8192),
("gpt-4-32k", 32768),
("gpt-3.5-turbo", 4096),
+ ("gpt-3.5-turbo-16k", 16384),
];
const CONFIG_FILE_NAME: &str = "config.yaml";