Iceberg/chat-api
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: ollama errors centralized

Browse Source
This commit is contained in:
LucasDLTG
2026-05-11 11:16:16 +02:00
parent 7bc489939f
commit b77715268d
9 changed files with 78 additions and 111 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/ollama_provider.rs
+8 -11
View File
@@ -3,8 +3,8 @@ use wiremock::matchers::{method, path};
use wiremock::{Mock, MockServer, ResponseTemplate};
use chat::dto::api;
use chat::errors::OllamaError;
use chat::providers::ollama::client::OllamaProvider;
use chat::providers::ollama::errors::OllamaError;
// ── helpers ──────────────────────────────────────────────────────────────────
@@ -147,7 +147,7 @@ async fn test_completions_model_not_found() {
let err = provider.completions(&req).await.unwrap_err();
assert!(matches!(err, chat::errors::OllamaError::ModelNotFound(_)));
assert!(matches!(err, OllamaError::ModelNotFound(_)));
}
// ── chat_completions ──────────────────────────────────────────────────────────
@@ -223,7 +223,7 @@ async fn test_chat_completions_missing_messages() {
let err = provider.chat_completions(&req).await.unwrap_err();
assert!(matches!(err, chat::errors::OllamaError::MissingMessages));
assert!(matches!(err, OllamaError::MissingMessages));
}
#[tokio::test]
@@ -250,7 +250,7 @@ async fn test_chat_completions_no_user_message() {
let err = provider.chat_completions(&req).await.unwrap_err();
assert!(matches!(err, chat::errors::OllamaError::MissingMessages));
assert!(matches!(err, OllamaError::MissingMessages));
}
#[tokio::test]
@@ -277,7 +277,7 @@ async fn test_chat_completions_model_not_found() {
let err = provider.chat_completions(&req).await.unwrap_err();
assert!(matches!(err, chat::errors::OllamaError::ModelNotFound(_)));
assert!(matches!(err, OllamaError::ModelNotFound(_)));
}
// // ── load_model ────────────────────────────────────────────────────────────────
@@ -321,7 +321,7 @@ async fn test_load_model_not_found() {
let err = provider.load_model("gpt-4", Some("10m")).await.unwrap_err();
assert!(matches!(err, chat::errors::OllamaError::ModelNotFound(_)));
assert!(matches!(err, OllamaError::ModelNotFound(_)));
}
#[tokio::test]
@@ -339,10 +339,7 @@ async fn test_load_model_invalid_keep_alive() {
.await
.unwrap_err();
assert!(matches!(
err,
chat::errors::OllamaError::InvalidKeepAlive(_)
));
assert!(matches!(err, OllamaError::InvalidKeepAlive(_)));
}
#[tokio::test]
@@ -414,5 +411,5 @@ async fn test_unload_model_not_found() {
.await;
let err = provider.unload_model("gpt-4").await.unwrap_err();
assert!(matches!(err, chat::errors::OllamaError::ModelNotFound(_)));
assert!(matches!(err, OllamaError::ModelNotFound(_)));
}