mirror of
https://github.com/aaif-goose/goose.git
synced 2026-07-03 14:10:03 +02:00
move formats/ollama
This commit is contained in:
@@ -1,2 +1,3 @@
|
||||
pub mod anthropic;
|
||||
pub mod ollama;
|
||||
pub mod openai;
|
||||
|
||||
+4
-6
@@ -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;
|
||||
|
||||
@@ -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>),
|
||||
|
||||
Reference in New Issue
Block a user