summaryrefslogtreecommitdiff
path: root/internal/hexaicli/run_more_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'internal/hexaicli/run_more_test.go')
-rw-r--r--internal/hexaicli/run_more_test.go51
1 files changed, 27 insertions, 24 deletions
diff --git a/internal/hexaicli/run_more_test.go b/internal/hexaicli/run_more_test.go
index ae29563..bd88d56 100644
--- a/internal/hexaicli/run_more_test.go
+++ b/internal/hexaicli/run_more_test.go
@@ -1,44 +1,47 @@
package hexaicli
import (
- "bytes"
- "context"
- "testing"
+ "bytes"
+ "context"
+ "testing"
- "codeberg.org/snonux/hexai/internal/appconfig"
- "codeberg.org/snonux/hexai/internal/llm"
+ "codeberg.org/snonux/hexai/internal/appconfig"
+ "codeberg.org/snonux/hexai/internal/llm"
)
type streamClient struct{}
func (streamClient) Chat(ctx context.Context, msgs []llm.Message, opts ...llm.RequestOption) (string, error) {
- return "X", nil
+ return "X", nil
}
func (streamClient) Name() string { return "fake" }
func (streamClient) DefaultModel() string { return "m" }
func (streamClient) ChatStream(ctx context.Context, msgs []llm.Message, onDelta func(string), opts ...llm.RequestOption) error {
- onDelta("A")
- onDelta("B")
- return nil
+ onDelta("A")
+ onDelta("B")
+ return nil
}
func TestRunChat_Streaming(t *testing.T) {
- var out, errw bytes.Buffer
- input := "hello"
- msgs := []llm.Message{{Role: "user", Content: input}}
- if err := runChat(context.Background(), streamClient{}, msgs, input, &out, &errw); err != nil {
- t.Fatalf("runChat failed: %v", err)
- }
- if out.String() != "AB" {
- t.Fatalf("unexpected stream output: %q", out.String())
- }
+ var out, errw bytes.Buffer
+ input := "hello"
+ msgs := []llm.Message{{Role: "user", Content: input}}
+ if err := runChat(context.Background(), streamClient{}, msgs, input, &out, &errw); err != nil {
+ t.Fatalf("runChat failed: %v", err)
+ }
+ if out.String() != "AB" {
+ t.Fatalf("unexpected stream output: %q", out.String())
+ }
}
func TestBuildMessagesFromConfig(t *testing.T) {
- cfg := appconfig.App{PromptCLIDefaultSystem: "DEF", PromptCLIExplainSystem: "EXP"}
- msgs := buildMessagesFromConfig(cfg, "tell me")
- if msgs[0].Content != "DEF" { t.Fatalf("default system wrong: %q", msgs[0].Content) }
- msgs = buildMessagesFromConfig(cfg, "please explain")
- if msgs[0].Content != "EXP" { t.Fatalf("explain system wrong: %q", msgs[0].Content) }
+ cfg := appconfig.App{PromptCLIDefaultSystem: "DEF", PromptCLIExplainSystem: "EXP"}
+ msgs := buildMessagesFromConfig(cfg, "tell me")
+ if msgs[0].Content != "DEF" {
+ t.Fatalf("default system wrong: %q", msgs[0].Content)
+ }
+ msgs = buildMessagesFromConfig(cfg, "please explain")
+ if msgs[0].Content != "EXP" {
+ t.Fatalf("explain system wrong: %q", msgs[0].Content)
+ }
}
-