move formats/ollama

This commit is contained in:
Jack Amadeo
2026-06-08 16:19:51 -04:00
parent 10e59c223e
commit 1dc29bfb49
4 changed files with 9 additions and 11 deletions
+1
View File
@@ -1,2 +1,3 @@
pub mod anthropic;
pub mod ollama;
pub mod openai;
@@ -10,21 +10,19 @@
//! - qwen3-coder-32b
use crate::conversation::message::{Message, MessageContent};
use crate::conversation::token_usage::ProviderUsage;
use async_stream::try_stream;
use chrono;
use futures::Stream;
use goose_providers::{
conversation::token_usage::ProviderUsage,
formats::openai::{self, is_valid_function_name},
};
use regex::Regex;
use rmcp::model::{object, CallToolRequestParams, ErrorCode, ErrorData, Role};
use serde_json::Value;
use std::borrow::Cow;
use uuid::Uuid;
pub use goose_providers::formats::openai::{
create_request, format_messages, format_tools, get_usage, validate_tool_schemas,
pub use super::openai::{
self, create_request, format_messages, format_tools, get_usage, is_valid_function_name,
validate_tool_schemas,
};
/// Parse XML-style tool calls from content (Ollama/Qwen3-coder fallback format).
@@ -3,7 +3,6 @@ pub mod bedrock;
pub mod databricks;
pub mod gcpvertexai;
pub mod google;
pub mod ollama;
pub mod openai_responses;
pub mod openrouter;
pub mod snowflake;
+4 -4
View File
@@ -10,13 +10,13 @@ use super::utils::RequestLog;
use crate::config::declarative_providers::DeclarativeProviderConfig;
use crate::conversation::message::Message;
use crate::model::ModelConfig;
use crate::providers::formats::ollama::{create_request, response_to_streaming_message_ollama};
use anyhow::{Error, Result};
use async_stream::try_stream;
use async_trait::async_trait;
use futures::future::BoxFuture;
use futures::TryStreamExt;
use goose_providers::errors::ProviderError;
use goose_providers::formats::ollama::{create_request, response_to_streaming_message_ollama};
use goose_providers::images::ImageFormat;
use goose_providers::models::ModelConfigParams;
use reqwest::Response;
@@ -565,7 +565,7 @@ mod tests {
#[test]
fn test_raw_create_request_contains_unsupported_ollama_fields() {
use crate::providers::formats::ollama::create_request;
use goose_providers::formats::ollama::create_request;
let model_config = ModelConfig::new("llama3.1")
.unwrap()
@@ -600,7 +600,7 @@ mod tests {
#[test]
fn test_apply_ollama_options_preserves_stream_options_by_default() {
use crate::providers::formats::ollama::create_request;
use goose_providers::formats::ollama::create_request;
let _guard = env_lock::lock_env([
("GOOSE_INPUT_LIMIT", None::<&str>),
@@ -650,7 +650,7 @@ mod tests {
#[test]
fn test_apply_ollama_options_strips_stream_options_when_disabled() {
use crate::providers::formats::ollama::create_request;
use goose_providers::formats::ollama::create_request;
let _guard = env_lock::lock_env([
("GOOSE_INPUT_LIMIT", None::<&str>),