PaLM API

开始使用Rust的PaLM API。

用法

从MakerSuite获取API密钥，然后在此处配置。

use palm_api::palm::create_client;

let client = create_client(PALM_API_KEY.to_string());

使用PalmClient的generate_text()方法让模型完成一些初始文本。

use palm_api::palm::new_text_body;

let mut text_body = new_text_body();
text_body.set_text_prompt("The opposite of hot is".to_string());
let response = client
    .generate_text("text-bison-001".to_string(), text_body)
    .expect("An error has occured.");
println!("{}", response.candidates.unwrap()[0].output);

使用PalmClient的chat()方法与模型进行讨论。

use palm_api::palm::new_chat_body;

let mut chat_body = new_chat_body();
chat_body.append_message("Hello.".to_string());
let response = client
    .chat("chat-bison-001".to_string(), chat_body)
    .expect("An error has occured.");
let response2 = client
    .reply(response, "What can you do?".to_string(), 0)
    .expect("An error has occured.");
println!("{}", response2.candidates.unwrap()[0].content);