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

Use Port of Context (pctx) for code mode (#6765)

Browse Source 
Signed-off-by: Elias Posen <elias@posen.ch>
Signed-off-by: Adrian Cole <adrian@tetrate.io>
Co-authored-by: Adrian Cole <adrian@tetrate.io>
This commit is contained in:
Elias Posen
2026-02-03 12:15:49 -05:00
committed by GitHub
parent 1f95f07065
commit 8631caa890
13 changed files with 3384 additions and 1845 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/goose/src/agents/code_execution_extension.rs
+334 -1124
View File
File diff suppressed because it is too large Load Diff
crates/goose/src/agents/extension_manager.rs
+2 -2
View File
@@ -1194,11 +1194,11 @@ impl ExtensionManager {
// Some models strip the tool prefix, so auto-add it for known code_execution tools
let tool_name_str = tool_call.name.to_string();
let prefixed_name = if !tool_name_str.contains("__") {
let code_exec_tools = ["execute_code", "read_module", "search_modules"];
let code_exec_tools = ["execute", "list_functions", "get_function_details"];
if code_exec_tools.contains(&tool_name_str.as_str())
&& self.extensions.lock().await.contains_key("code_execution")
{
format!("code_execution__{}", tool_name_str)
format!("code_execution__{tool_name_str}")
} else {
tool_name_str
}