From bcb0cd29a62701ed8666bd237b8533e9a9898fb5 Mon Sep 17 00:00:00 2001
From: SilasMarvin <19626586+SilasMarvin@users.noreply.github.com>
Date: Sat, 27 Jul 2024 11:43:58 -0700
Subject: [PATCH] Tests passing

---
 crates/lsp-ai/src/config.rs                       | 4 ++--
 crates/lsp-ai/src/memory_backends/vector_store.rs | 3 ++-
 crates/lsp-ai/src/transformer_backends/gemini.rs  | 6 ++++++
 3 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/crates/lsp-ai/src/config.rs b/crates/lsp-ai/src/config.rs
index 102a857..1a8eed3 100644
--- a/crates/lsp-ai/src/config.rs
+++ b/crates/lsp-ai/src/config.rs
@@ -105,7 +105,7 @@ pub enum VectorDataType {
 }
 
 #[derive(Debug, Clone, Deserialize)]
-pub struct VectorStore {
+pub(crate) struct VectorStore {
     pub crawl: Option<Crawl>,
     #[serde(default)]
     pub splitter: ValidSplitter,
@@ -433,7 +433,7 @@ impl Config {
         }
     }
 
-    pub fn default_with_vector_store(vector_store: VectorStore) -> Self {
+    pub(crate) fn default_with_vector_store(vector_store: VectorStore) -> Self {
         Self {
             config: ValidConfig {
                 memory: ValidMemoryBackend::VectorStore(vector_store),
diff --git a/crates/lsp-ai/src/memory_backends/vector_store.rs b/crates/lsp-ai/src/memory_backends/vector_store.rs
index 22950ca..53e024d 100644
--- a/crates/lsp-ai/src/memory_backends/vector_store.rs
+++ b/crates/lsp-ai/src/memory_backends/vector_store.rs
@@ -827,7 +827,8 @@ assert multiply_two_numbers(2, 3) == 6
         assert!(chunks.len() == 1);
         assert_eq!(
             chunks[0].text,
-            r#"--file:///filler.py--\n# Multiplies two numbers
+            r#"--file:///filler.py--
+# Multiplies two numbers
 def multiply_two_numbers(x, y):
     return
 
diff --git a/crates/lsp-ai/src/transformer_backends/gemini.rs b/crates/lsp-ai/src/transformer_backends/gemini.rs
index be731de..a4b14ad 100644
--- a/crates/lsp-ai/src/transformer_backends/gemini.rs
+++ b/crates/lsp-ai/src/transformer_backends/gemini.rs
@@ -59,11 +59,15 @@ impl GeminiContent {
 #[serde(deny_unknown_fields)]
 pub(crate) struct GeminiGenerationConfig {
     #[serde(default)]
+    #[serde(rename = "stopSequences")]
     pub(crate) stop_sequences: Vec<String>,
+    #[serde(rename = "maxOutputTokens")]
     #[serde(default = "max_tokens_default")]
     pub(crate) max_output_tokens: usize,
     pub(crate) temperature: Option<f32>,
+    #[serde(rename = "topP")]
     pub(crate) top_p: Option<f32>,
+    #[serde(rename = "topK")]
     pub(crate) top_k: Option<f32>,
 }
 
@@ -72,7 +76,9 @@ pub(crate) struct GeminiGenerationConfig {
 #[serde(rename = "camelCase")]
 pub(crate) struct GeminiRunParams {
     contents: Vec<GeminiContent>,
+    #[serde(rename = "systemInstruction")]
     system_instruction: GeminiContent,
+    #[serde(rename = "generationConfig")]
     generation_config: Option<GeminiGenerationConfig>,
 }