fix gemini_config

This commit is contained in:
Asuka Minato 2024-06-19 09:36:16 +09:00
parent c8993bf740
commit be577c19e0
2 changed files with 14 additions and 8 deletions

View File

@ -441,15 +441,22 @@ mod test {
"completion": {
"model": "model1",
"parameters": {
"messages": [
"contents": [
{
"role": "system",
"content": "Test",
},
"role": "user",
"parts":[{
"text": "Pretend you're a snowman and stay in character for each response."}]
},
{
"role": "user",
"content": "Test {CONTEXT} - {CODE}"
}
"role": "model",
"parts":[{
"text": "Hello! It's so cold! Isn't that great?"}]
},
{
"role": "user",
"parts":[{
"text": "What's your favorite season of the year?"}]
}
],
"max_new_tokens": 32,
}

View File

@ -259,7 +259,6 @@ mod test {
dbg!(response.generated_text);
Ok(())
}
// gemini_chat_do_generate TODO
#[tokio::test]
async fn gemini_chat_do_generate() -> anyhow::Result<()> {
let configuration: config::Gemini = serde_json::from_value(json!({