From ec51b84290215d11d5045de8baa9b89f2447b1e9 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Mon, 19 Jun 2023 08:44:26 +0800
Subject: feat: add new model gpt-3.5-turbo-16k (#126)

---
 README.md         | 1 +
 src/config/mod.rs | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

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