toki/iop
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix(openai): artifact pair 도구 선택을 고정한다

Browse source
This commit is contained in:
toki 2026-08-15 14:41:01 +09:00
parent 949db240a9
commit af96a1e03a
2 changed files with 11 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
apps/edge/internal/openai/provider_model_rewrite.go
View file

@ -215,7 +215,9 @@ func prepareHotPathSelectorCanonicalTools(tunnel edgeservice.SubmitProviderTunne
},
},
}}
root["tool_choice"] = "required"
root["tool_choice"] = map[string]any{
"type": "function", "function": map[string]any{"name": hotPathArtifactPairToolName},
}
delete(root, "parallel_tool_calls")
return json.Marshal(root)
}

11
apps/edge/internal/openai/provider_model_rewrite_test.go
View file

@ -394,7 +394,9 @@ func TestHotPathSelectorCanonicalWriteToolReplacesCallerCommandSchema(t *testing
if function["name"] != hotPathArtifactPairToolName || properties["plan_content"] == nil || properties["review_content"] == nil || properties["command"] != nil {
t.Fatalf("canonical write function=%+v", function)
}
if request["tool_choice"] != "required" || request["parallel_tool_calls"] != nil {
choice := request["tool_choice"].(map[string]any)
chosenFunction := choice["function"].(map[string]any)
if choice["type"] != "function" || chosenFunction["name"] != hotPathArtifactPairToolName || request["parallel_tool_calls"] != nil {
t.Fatalf("pair-write must force one atomic tool call: %+v", request)
}
}
@ -447,7 +449,8 @@ func TestAnthropicCallerWorkspaceSelectorUsesCanonicalProviderOperation(t *testi
t.Fatalf("OpenAI selector tunnel=%+v", prepared)
}
tool := tools[0].(map[string]any)
if tool["name"] != hotPathArtifactPairToolName || request["tool_choice"] != "required" || request["max_output_tokens"] != float64(maxHotPathSelectorOutputTokens) {
choice := request["tool_choice"].(map[string]any)
if tool["name"] != hotPathArtifactPairToolName || choice["type"] != "function" || choice["name"] != hotPathArtifactPairToolName || request["max_output_tokens"] != float64(maxHotPathSelectorOutputTokens) {
t.Fatalf("OpenAI selector request=%+v", request)
}
if !strings.Contains(request["instructions"].(string), "Operation: pair-write") {
@ -459,7 +462,9 @@ func TestAnthropicCallerWorkspaceSelectorUsesCanonicalProviderOperation(t *testi
t.Fatalf("Gemini selector tunnel=%+v", prepared)
}
function := tools[0].(map[string]any)["function"].(map[string]any)
if function["name"] != hotPathArtifactPairToolName || request["tool_choice"] != "required" || request["max_tokens"] != float64(maxHotPathSelectorOutputTokens) {
choice := request["tool_choice"].(map[string]any)
chosenFunction := choice["function"].(map[string]any)
if function["name"] != hotPathArtifactPairToolName || choice["type"] != "function" || chosenFunction["name"] != hotPathArtifactPairToolName || request["max_tokens"] != float64(maxHotPathSelectorOutputTokens) {
t.Fatalf("Gemini selector request=%+v", request)
}
messages := anySlice(request["messages"])