Skip to content

Commit

Permalink
Formatted code
Browse files Browse the repository at this point in the history
  • Loading branch information
pepperoni21 committed Dec 6, 2023
1 parent 36659c2 commit d74c623
Showing 1 changed file with 9 additions and 3 deletions.
12 changes: 9 additions & 3 deletions tests/create_model.rs
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
use ollama_rs::{Ollama, models::create::CreateModelRequest};
use ollama_rs::{models::create::CreateModelRequest, Ollama};
use tokio_stream::StreamExt;

#[tokio::test]
Expand All @@ -7,7 +7,10 @@ async fn test_create_model_stream() {
let ollama = Ollama::default();

let mut res = ollama
.create_model_stream(CreateModelRequest::path("model".into(), "/tmp/Modelfile.example".into()))
.create_model_stream(CreateModelRequest::path(
"model".into(),
"/tmp/Modelfile.example".into(),
))
.await
.unwrap();

Expand All @@ -33,7 +36,10 @@ async fn test_create_model() {
let ollama = Ollama::default();

let res = ollama
.create_model(CreateModelRequest::path("model".into(), "/tmp/Modelfile.example".into()))
.create_model(CreateModelRequest::path(
"model".into(),
"/tmp/Modelfile.example".into(),
))
.await
.unwrap();

Expand Down

0 comments on commit d74c623

Please sign in to comment.