tkmind/tkmind_go
1
0
Fork 0
Code Issues Pull Requests Actions 5 Packages Projects Releases Wiki Activity

feat: Allow Ollama for non-tool models for chat only (#3308)

Browse Source
This commit is contained in:
Shravan Dhakal
2025-07-11 10:04:38 -07:00
committed by GitHub
parent 3cdb642cc8
commit 4ba991e52e
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/goose/src/providers/ollama.rs
+5 -2
View File
@@ -127,14 +127,17 @@ impl Provider for OllamaProvider {
messages: &[Message], messages: &[Message],
tools: &[Tool], tools: &[Tool],
) -> Result<(Message, ProviderUsage), ProviderError> { ) -> Result<(Message, ProviderUsage), ProviderError> {
let config = crate::config::Config::global();
let goose_mode = config.get_param("GOOSE_MODE").unwrap_or("auto".to_string());
let filtered_tools = if goose_mode == "chat" { &[] } else { tools };
let payload = create_request( let payload = create_request(
&self.model, &self.model,
system, system,
messages, messages,
tools, filtered_tools,
&super::utils::ImageFormat::OpenAi, &super::utils::ImageFormat::OpenAi,
)?; )?;
let response = self.post(payload.clone()).await?; let response = self.post(payload.clone()).await?;
let message = response_to_message(response.clone())?; let message = response_to_message(response.clone())?;