respect model specified in config
This commit is contained in:
parent
69e19c3b87
commit
741a71fb3b
|
@ -11,6 +11,15 @@ pub enum Model {
|
||||||
GPT35,
|
GPT35,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
impl ToString for Model {
|
||||||
|
fn to_string(&self) -> String {
|
||||||
|
match self {
|
||||||
|
Self::Claude12 => String::from("claude-instant-1.2"),
|
||||||
|
Self::GPT35 => String::from("gpt-3.5-turbo-0125"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||||
pub struct Config {
|
pub struct Config {
|
||||||
pub model: Model,
|
pub model: Model,
|
||||||
|
|
|
@ -96,15 +96,13 @@ async fn get_vqd(cli: &Client) -> Result<String, Box<dyn Error>> {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn get_res<'a>(cli: &Client, query: String, vqd: String, cache: &'a mut Cache) {
|
async fn get_res<'a>(cli: &Client, query: String, vqd: String, cache: &'a mut Cache, config: &Config) {
|
||||||
let payload = ChatPayload {
|
let payload = ChatPayload {
|
||||||
model: "claude-instant-1.2".into(),
|
model: config.model.to_string(),
|
||||||
messages: vec![ ChatMessagePayload { role: "user".into(), content: query } ]
|
messages: vec![ ChatMessagePayload { role: "user".into(), content: query } ]
|
||||||
};
|
};
|
||||||
let payload = serde_json::to_string(&payload).unwrap();
|
let payload = serde_json::to_string(&payload).unwrap();
|
||||||
|
|
||||||
// println!("{payload}\n\n{:#?}", headers);return;
|
|
||||||
|
|
||||||
let req = cli.post("https://duckduckgo.com/duckchat/v1/chat")
|
let req = cli.post("https://duckduckgo.com/duckchat/v1/chat")
|
||||||
// .headers(get_headers())
|
// .headers(get_headers())
|
||||||
.header("Content-Type", "application/json")
|
.header("Content-Type", "application/json")
|
||||||
|
@ -197,6 +195,6 @@ async fn main() {
|
||||||
None => get_vqd(&cli).await.unwrap()
|
None => get_vqd(&cli).await.unwrap()
|
||||||
};
|
};
|
||||||
|
|
||||||
get_res(&cli, query, vqd, &mut cache).await;
|
get_res(&cli, query, vqd, &mut cache, &config).await;
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue