PNArchives/Memoh
1
0
Fork 0
mirror of https://github.com/memohai/Memoh.git synced 2026-04-27 07:16:19 +09:00
Code Issues Packages Projects Releases Wiki Activity

refactor(agent): merge read_media tool into read tool (#326)

Browse Source
This commit is contained in:
Ringo.Typowriter
2026-04-04 20:56:00 +08:00
committed by GitHub
parent 5cfbaa40e2
commit 09c523f0b8
8 changed files with 127 additions and 323 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/conversation/flow/read_media_prompt_test.go
+8 -6
View File
@@ -8,7 +8,9 @@ import (
agentpkg "github.com/memohai/memoh/internal/agent"
)
func TestPrepareRunConfigIncludesReadMediaWhenImageInputIsSupported(t *testing.T) {
const imageReadHint = "also supports images: PNG, JPEG, GIF, WebP"
func TestPrepareRunConfigIncludesImageReadHintWhenImageInputIsSupported(t *testing.T) {
t.Parallel()
resolver := &Resolver{}
@@ -21,12 +23,12 @@ func TestPrepareRunConfigIncludesReadMediaWhenImageInputIsSupported(t *testing.T
}
prepared := resolver.prepareRunConfig(context.Background(), cfg)
if !strings.Contains(prepared.System, "`read_media`") {
t.Fatalf("expected system prompt to include read_media tool, got:\n%s", prepared.System)
if !strings.Contains(prepared.System, imageReadHint) {
t.Fatalf("expected system prompt to contain %q, got:\n%s", imageReadHint, prepared.System)
}
}
func TestPrepareRunConfigOmitsReadMediaWhenImageInputIsUnsupported(t *testing.T) {
func TestPrepareRunConfigOmitsImageReadHintWhenImageInputIsUnsupported(t *testing.T) {
t.Parallel()
resolver := &Resolver{}
@@ -39,7 +41,7 @@ func TestPrepareRunConfigOmitsReadMediaWhenImageInputIsUnsupported(t *testing.T)
}
prepared := resolver.prepareRunConfig(context.Background(), cfg)
if strings.Contains(prepared.System, "`read_media`") {
t.Fatalf("expected system prompt to omit read_media tool, got:\n%s", prepared.System)
if strings.Contains(prepared.System, imageReadHint) {
t.Fatalf("expected system prompt to NOT contain %q, got:\n%s", imageReadHint, prepared.System)
}
}