mirror of
https://wget.la/https://github.com/leookun/cursor-byok
synced 2026-08-21 05:26:59 +08:00
update prompt
This commit is contained in:
@@ -0,0 +1,24 @@
|
||||
package prompt
|
||||
|
||||
import "testing"
|
||||
|
||||
func TestRenderPromptTemplateReplacesModelPlaceholders(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
got := RenderPromptTemplate(
|
||||
"id={{FAKE_MODEL_ID}} name={{FAKE_MODEL_NAME}}",
|
||||
"Test Model",
|
||||
)
|
||||
if want := "id=Test Model name=Test Model"; got != want {
|
||||
t.Fatalf("RenderPromptTemplate() = %q, want %q", got, want)
|
||||
}
|
||||
}
|
||||
|
||||
func TestRenderPromptTemplateUsesFallbackModelName(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
got := RenderPromptTemplate("{{FAKE_MODEL_NAME}}", " ")
|
||||
if want := "当前请求模型"; got != want {
|
||||
t.Fatalf("RenderPromptTemplate() = %q, want %q", got, want)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user