|
|
|
|
@@ -176,7 +176,7 @@ func TestOpenAICompatibleClientGenerateSuccess(t *testing.T) {
|
|
|
|
|
if err != nil {
|
|
|
|
|
t.Fatalf("unexpected constructor error: %v", err)
|
|
|
|
|
}
|
|
|
|
|
t.Setenv("SCRIPTORIUM_TEST_API_KEY", "secret-key")
|
|
|
|
|
t.Setenv("PROMPTKIT_TEST_API_KEY", "secret-key")
|
|
|
|
|
|
|
|
|
|
resp, err := client.Generate(context.Background(), domain.GenerateRequest{
|
|
|
|
|
Prompt: domain.RenderedPrompt{Messages: []domain.RenderedMessage{
|
|
|
|
|
@@ -189,7 +189,7 @@ func TestOpenAICompatibleClientGenerateSuccess(t *testing.T) {
|
|
|
|
|
MaxTokens: 123,
|
|
|
|
|
TopP: 0.7,
|
|
|
|
|
ServiceTier: "priority",
|
|
|
|
|
APIKeyEnv: "SCRIPTORIUM_TEST_API_KEY",
|
|
|
|
|
APIKeyEnv: "PROMPTKIT_TEST_API_KEY",
|
|
|
|
|
},
|
|
|
|
|
StructuredOutput: &domain.StructuredOutputSpec{
|
|
|
|
|
Type: domain.StructuredOutputJSONSchema,
|
|
|
|
|
@@ -276,7 +276,7 @@ func TestOpenAICompatibleClientGenerateSuccess(t *testing.T) {
|
|
|
|
|
|
|
|
|
|
func TestOpenAICompatibleClientDirectAPIKeyPreferredOverEnv(t *testing.T) {
|
|
|
|
|
const directKey = "direct-llm-key"
|
|
|
|
|
t.Setenv("SCRIPTORIUM_TEST_API_KEY", "env-key")
|
|
|
|
|
t.Setenv("PROMPTKIT_TEST_API_KEY", "env-key")
|
|
|
|
|
|
|
|
|
|
var gotAuth string
|
|
|
|
|
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
|
|
|
@@ -294,7 +294,7 @@ func TestOpenAICompatibleClientDirectAPIKeyPreferredOverEnv(t *testing.T) {
|
|
|
|
|
Prompt: domain.RenderedPrompt{Messages: []domain.RenderedMessage{{Role: "user", Content: "hi"}}},
|
|
|
|
|
Target: domain.ExecutionTarget{
|
|
|
|
|
Model: "model",
|
|
|
|
|
APIKeyEnv: "SCRIPTORIUM_TEST_API_KEY",
|
|
|
|
|
APIKeyEnv: "PROMPTKIT_TEST_API_KEY",
|
|
|
|
|
APIKey: directKey,
|
|
|
|
|
},
|
|
|
|
|
})
|
|
|
|
|
@@ -870,7 +870,7 @@ func TestOpenAICompatibleClientAPIKeyEnvMissing(t *testing.T) {
|
|
|
|
|
|
|
|
|
|
_, err = client.Generate(context.Background(), domain.GenerateRequest{
|
|
|
|
|
Prompt: domain.RenderedPrompt{Messages: []domain.RenderedMessage{{Role: "user", Content: "hi"}}},
|
|
|
|
|
Target: domain.ExecutionTarget{APIKeyEnv: "SCRIPTORIUM_MISSING_KEY"},
|
|
|
|
|
Target: domain.ExecutionTarget{APIKeyEnv: "PROMPTKIT_MISSING_KEY"},
|
|
|
|
|
})
|
|
|
|
|
if err == nil {
|
|
|
|
|
t.Fatal("expected missing API key env error")
|
|
|
|
|
|