mirror of
https://github.com/SilasMarvin/lsp-ai.git
synced 2024-09-11 12:25:48 +03:00
fix gemini_config
This commit is contained in:
parent
c8993bf740
commit
be577c19e0
@ -441,15 +441,22 @@ mod test {
|
|||||||
"completion": {
|
"completion": {
|
||||||
"model": "model1",
|
"model": "model1",
|
||||||
"parameters": {
|
"parameters": {
|
||||||
"messages": [
|
"contents": [
|
||||||
{
|
{
|
||||||
"role": "system",
|
"role": "user",
|
||||||
"content": "Test",
|
"parts":[{
|
||||||
},
|
"text": "Pretend you're a snowman and stay in character for each response."}]
|
||||||
|
},
|
||||||
{
|
{
|
||||||
"role": "user",
|
"role": "model",
|
||||||
"content": "Test {CONTEXT} - {CODE}"
|
"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,
|
"max_new_tokens": 32,
|
||||||
}
|
}
|
||||||
|
@ -259,7 +259,6 @@ mod test {
|
|||||||
dbg!(response.generated_text);
|
dbg!(response.generated_text);
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
// gemini_chat_do_generate TODO
|
|
||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
async fn gemini_chat_do_generate() -> anyhow::Result<()> {
|
async fn gemini_chat_do_generate() -> anyhow::Result<()> {
|
||||||
let configuration: config::Gemini = serde_json::from_value(json!({
|
let configuration: config::Gemini = serde_json::from_value(json!({
|
||||||
|
Loading…
Reference in New Issue
Block a user