Rename openai to ollama

bosun-ai · Sep 8, 2024 · 11682f4 · 11682f4
1 parent 39c8094
commit 11682f4
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/swiftide-integrations/src/ollama/mod.rs b/swiftide-integrations/src/ollama/mod.rs
@@ -157,29 +157,29 @@ mod test {
 
     #[test]
     fn test_default_embed_model() {
-        let openai = Ollama::builder()
+        let ollama = Ollama::builder()
             .default_embed_model("mxbai-embed-large")
             .build()
             .unwrap();
         assert_eq!(
-            openai.default_options.embed_model,
+            ollama.default_options.embed_model,
             Some("mxbai-embed-large".to_string())
         );
     }
 
     #[test]
     fn test_default_models() {
-        let openai = Ollama::builder()
+        let ollama = Ollama::builder()
             .default_embed_model("mxbai-embed-large")
             .default_prompt_model("llama3.1")
             .build()
             .unwrap();
         assert_eq!(
-            openai.default_options.embed_model,
+            ollama.default_options.embed_model,
             Some("mxbai-embed-large".to_string())
         );
         assert_eq!(
-            openai.default_options.prompt_model,
+            ollama.default_options.prompt_model,
             Some("llama3.1".to_string())
         );
     }