fix(perplexity): allow to specify chat messages

CHANGELOG.md

@@ -1,3 +1,9 @@
+## What's Changed in perplexity-v0.1.1
+* fix(perplexity): allow to specify chat messages
+* chore: clean-up
+
+**Full Changelog**: https://github.com///compare/perplexity-v0.1.0...perplexity-v0.1.1
+
 ## What's Changed in opai-v0.3.1
 * refactor(openai): use struct instead of enum for `O1`
 

examples/perplexity-chat-completion/Cargo.toml → examples/perplexity-chat/Cargo.toml

@@ -1,5 +1,5 @@
 [package]
-name = "perplexity-chat-completion"
+name = "perplexity-chat"
 version = "0.1.0"
 edition = "2021"
 publish = false

examples/perplexity-chat/README.md

@@ -0,0 +1 @@
+# Perplexity Chat Example

examples/perplexity-chat-completion/src/main.rs → examples/perplexity-chat/src/main.rs

@@ -1,5 +1,5 @@
 use perplexity::{
-    client::{Client, CreateChatCompletion, Model},
+    client::{Client, CreateChatCompletion, Message, Model, Role},
     config::Config,
 };
 
@@ -11,7 +11,11 @@ async fn main() {
     let config = Config::new(api_key);
     let client = Client::new(config).unwrap();
 
-    let message = CreateChatCompletion::new(Model::Llama31SonarLargeOnline);
+    let messages: Vec<Message> = vec![Message {
+        role: Role::User,
+        content: "Find me the best pad thai restaurant in Bangkok".to_string(),
+    }];
+    let message = CreateChatCompletion::new(Model::Llama31SonarLargeOnline, messages);
     let result = client.create_completion(message).await.unwrap();
     println!("{:?}", result);
 }

perplexity/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "perplexity"
-version = "0.1.0"
+version = "0.1.1"
 edition.workspace = true
 authors.workspace = true
 description = "Perplexity Rust SDK"

perplexity/src/client.rs

@@ -99,7 +99,7 @@ pub enum FinishReason {
 }
 
 /// The message generated by the model.
-#[derive(Debug, Serialize, Deserialize)]
+#[derive(Debug, Clone, Serialize, Deserialize)]
 pub struct Message {
     /// The contents of the message in this turn of conversation.
     pub content: String,
@@ -118,7 +118,7 @@ pub struct CompletionDelta {
     pub role: Role,
 }
 
-#[derive(Debug, Serialize, Deserialize)]
+#[derive(Debug, Clone, Serialize, Deserialize)]
 #[serde(rename_all = "lowercase")]
 pub enum Role {
     System,
@@ -143,6 +143,9 @@ pub struct CreateChatCompletion {
     /// The name of the model that will complete your prompt.
     pub model: Model,
 
+    /// The conversation messages.
+    pub messages: Vec<Message>,
+
     /// The maximum number of completion tokens returned by the API. The total number of tokens requested in **max_tokens** plus the number of prompt tokens sent in messages must not exceed the context window token limit of model requested. If left unspecified, then the model will generate tokens until either it reaches its stop token or the end of its context window.
     #[serde(skip_serializing_if = "Option::is_none")]
     pub max_tokens: Option<u64>,
@@ -200,9 +203,10 @@ pub struct CreateChatCompletion {
 }
 
 impl CreateChatCompletion {
-    pub fn new(model: Model) -> Self {
+    pub fn new(model: Model, messages: Vec<Message>) -> Self {
         Self {
             model,
+            messages,
             ..Default::default()
         }
     }
@@ -277,6 +281,7 @@ impl Default for CreateChatCompletion {
     fn default() -> Self {
         Self {
             model: Model::Llama31SonarLargeOnline,
+            messages: Vec::new(),
             stream: false,
             max_tokens: None,
             temperature: 0.2,