Merge pull request #58 from jpmcb/openai-api-rs-2.0.0

feat: Upgrade to openai-api-rs 2.0.0

Cargo.toml

@@ -18,7 +18,7 @@ rayon = "1"
 reqwest = { version = "0.11", default-features = false, features = ["rustls-tls", "json"] }
 serde = "1"
 tokenizers = "0.14"
-openai-api-rs = "1.0"
+openai-api-rs = "2.0"
 zip = "0.6"
 rust-fuzzy-search = "0.1"
 text-splitter = "0.4"

src/conversation/mod.rs

@@ -93,7 +93,8 @@ impl<D: RepositoryEmbeddingsDB, M: EmbeddingsModel> Conversation<D, M> {
         #[allow(unused_labels)]
         'conversation: loop {
             //Generate a request with the message history and functions
-            let request = generate_completion_request(self.messages.clone(), FunctionCallType::Auto);
+            let request =
+                generate_completion_request(self.messages.clone(), FunctionCallType::Auto);
 
             match self.send_request(request) {
                 Ok(response) => {

src/conversation/prompts.rs

@@ -1,6 +1,6 @@
 use openai_api_rs::v1::chat_completion::{
-    ChatCompletionMessage, ChatCompletionRequest, Function as F, FunctionParameters,
-    JSONSchemaDefine, JSONSchemaType, FunctionCallType,
+    ChatCompletionMessage, ChatCompletionRequest, Function as F, FunctionCallType,
+    FunctionParameters, JSONSchemaDefine, JSONSchemaType,
 };
 use std::collections::HashMap;
 
@@ -17,23 +17,10 @@ pub fn generate_completion_request(
     messages: Vec<ChatCompletionMessage>,
     function_call: FunctionCallType,
 ) -> ChatCompletionRequest {
-
-    ChatCompletionRequest {
-        model: CHAT_COMPLETION_MODEL.into(),
-        messages,
-        functions: Some(functions()),
-        function_call: Some(function_call),
-        temperature: Some(CHAT_COMPLETION_TEMPERATURE),
-        top_p: None,
-        n: None,
-        stream: None,
-        stop: None,
-        max_tokens: None,
-        presence_penalty: None,
-        frequency_penalty: None,
-        logit_bias: None,
-        user: None,
-    }
+    ChatCompletionRequest::new(CHAT_COMPLETION_MODEL.to_string(), messages)
+        .functions(functions())
+        .function_call(function_call)
+        .temperature(CHAT_COMPLETION_TEMPERATURE)
 }
 
 pub fn functions() -> Vec<F> {