@@ -809,10 +809,11 @@ func TestDoGenerate(t *testing.T) {
"content": "Hello, World!",
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-3.5-turbo")
result, err := model.Generate(context.Background(), fantasy.Call{
@@ -841,10 +842,11 @@ func TestDoGenerate(t *testing.T) {
},
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-3.5-turbo")
result, err := model.Generate(context.Background(), fantasy.Call{
@@ -865,13 +867,14 @@ func TestDoGenerate(t *testing.T) {
server.prepareJSONResponse(map[string]any{})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-3.5-turbo")
- _, err := model.Generate(context.Background(), fantasy.Call{
+ _, err = model.Generate(context.Background(), fantasy.Call{
Prompt: testPrompt,
})
@@ -905,10 +908,11 @@ func TestDoGenerate(t *testing.T) {
},
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-3.5-turbo")
result, err := model.Generate(context.Background(), fantasy.Call{
@@ -931,10 +935,11 @@ func TestDoGenerate(t *testing.T) {
"logprobs": testLogprobs,
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-3.5-turbo")
result, err := model.Generate(context.Background(), fantasy.Call{
@@ -965,10 +970,11 @@ func TestDoGenerate(t *testing.T) {
"finish_reason": "stop",
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-3.5-turbo")
result, err := model.Generate(context.Background(), fantasy.Call{
@@ -989,10 +995,11 @@ func TestDoGenerate(t *testing.T) {
"finish_reason": "eos",
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-3.5-turbo")
result, err := model.Generate(context.Background(), fantasy.Call{
@@ -1013,13 +1020,14 @@ func TestDoGenerate(t *testing.T) {
"content": "",
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-3.5-turbo")
- _, err := model.Generate(context.Background(), fantasy.Call{
+ _, err = model.Generate(context.Background(), fantasy.Call{
Prompt: testPrompt,
})
@@ -1045,13 +1053,14 @@ func TestDoGenerate(t *testing.T) {
server.prepareJSONResponse(map[string]any{})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-3.5-turbo")
- _, err := model.Generate(context.Background(), fantasy.Call{
+ _, err = model.Generate(context.Background(), fantasy.Call{
Prompt: testPrompt,
ProviderOptions: NewProviderOptions(&ProviderOptions{
LogitBias: map[string]int64{
@@ -1087,13 +1096,14 @@ func TestDoGenerate(t *testing.T) {
"content": "",
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("o1-mini")
- _, err := model.Generate(context.Background(), fantasy.Call{
+ _, err = model.Generate(context.Background(), fantasy.Call{
Prompt: testPrompt,
ProviderOptions: NewProviderOptions(
&ProviderOptions{
@@ -1127,13 +1137,14 @@ func TestDoGenerate(t *testing.T) {
"content": "",
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-4o")
- _, err := model.Generate(context.Background(), fantasy.Call{
+ _, err = model.Generate(context.Background(), fantasy.Call{
Prompt: testPrompt,
ProviderOptions: NewProviderOptions(&ProviderOptions{
TextVerbosity: fantasy.Opt("low"),
@@ -1165,13 +1176,14 @@ func TestDoGenerate(t *testing.T) {
"content": "",
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-3.5-turbo")
- _, err := model.Generate(context.Background(), fantasy.Call{
+ _, err = model.Generate(context.Background(), fantasy.Call{
Prompt: testPrompt,
Tools: []fantasy.Tool{
fantasy.FunctionTool{
@@ -1237,10 +1249,11 @@ func TestDoGenerate(t *testing.T) {
},
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-3.5-turbo")
result, err := model.Generate(context.Background(), fantasy.Call{
@@ -1295,10 +1308,11 @@ func TestDoGenerate(t *testing.T) {
},
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-3.5-turbo")
result, err := model.Generate(context.Background(), fantasy.Call{
@@ -1337,10 +1351,11 @@ func TestDoGenerate(t *testing.T) {
},
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-4o-mini")
result, err := model.Generate(context.Background(), fantasy.Call{
@@ -1372,10 +1387,11 @@ func TestDoGenerate(t *testing.T) {
},
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-4o-mini")
result, err := model.Generate(context.Background(), fantasy.Call{
@@ -1400,10 +1416,11 @@ func TestDoGenerate(t *testing.T) {
server.prepareJSONResponse(map[string]any{})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("o1-preview")
result, err := model.Generate(context.Background(), fantasy.Call{
@@ -1448,13 +1465,14 @@ func TestDoGenerate(t *testing.T) {
server.prepareJSONResponse(map[string]any{})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("o1-preview")
- _, err := model.Generate(context.Background(), fantasy.Call{
+ _, err = model.Generate(context.Background(), fantasy.Call{
Prompt: testPrompt,
MaxOutputTokens: &[]int64{1000}[0],
})
@@ -1492,10 +1510,11 @@ func TestDoGenerate(t *testing.T) {
},
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("o1-preview")
result, err := model.Generate(context.Background(), fantasy.Call{
@@ -1519,13 +1538,14 @@ func TestDoGenerate(t *testing.T) {
"model": "o1-preview",
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("o1-preview")
- _, err := model.Generate(context.Background(), fantasy.Call{
+ _, err = model.Generate(context.Background(), fantasy.Call{
Prompt: testPrompt,
ProviderOptions: NewProviderOptions(&ProviderOptions{
MaxCompletionTokens: fantasy.Opt(int64(255)),
@@ -1557,13 +1577,14 @@ func TestDoGenerate(t *testing.T) {
"content": "",
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-3.5-turbo")
- _, err := model.Generate(context.Background(), fantasy.Call{
+ _, err = model.Generate(context.Background(), fantasy.Call{
Prompt: testPrompt,
ProviderOptions: NewProviderOptions(&ProviderOptions{
Prediction: map[string]any{
@@ -1601,13 +1622,14 @@ func TestDoGenerate(t *testing.T) {
"content": "",
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-3.5-turbo")
- _, err := model.Generate(context.Background(), fantasy.Call{
+ _, err = model.Generate(context.Background(), fantasy.Call{
Prompt: testPrompt,
ProviderOptions: NewProviderOptions(&ProviderOptions{
Store: fantasy.Opt(true),
@@ -1639,13 +1661,14 @@ func TestDoGenerate(t *testing.T) {
"content": "",
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-3.5-turbo")
- _, err := model.Generate(context.Background(), fantasy.Call{
+ _, err = model.Generate(context.Background(), fantasy.Call{
Prompt: testPrompt,
ProviderOptions: NewProviderOptions(&ProviderOptions{
Metadata: map[string]any{
@@ -1681,13 +1704,14 @@ func TestDoGenerate(t *testing.T) {
"content": "",
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-3.5-turbo")
- _, err := model.Generate(context.Background(), fantasy.Call{
+ _, err = model.Generate(context.Background(), fantasy.Call{
Prompt: testPrompt,
ProviderOptions: NewProviderOptions(&ProviderOptions{
PromptCacheKey: fantasy.Opt("test-cache-key-123"),
@@ -1719,13 +1743,14 @@ func TestDoGenerate(t *testing.T) {
"content": "",
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-3.5-turbo")
- _, err := model.Generate(context.Background(), fantasy.Call{
+ _, err = model.Generate(context.Background(), fantasy.Call{
Prompt: testPrompt,
ProviderOptions: NewProviderOptions(&ProviderOptions{
SafetyIdentifier: fantasy.Opt("test-safety-identifier-123"),
@@ -1755,10 +1780,11 @@ func TestDoGenerate(t *testing.T) {
server.prepareJSONResponse(map[string]any{})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-4o-search-preview")
result, err := model.Generate(context.Background(), fantasy.Call{
@@ -1789,13 +1815,14 @@ func TestDoGenerate(t *testing.T) {
"content": "",
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("o3-mini")
- _, err := model.Generate(context.Background(), fantasy.Call{
+ _, err = model.Generate(context.Background(), fantasy.Call{
Prompt: testPrompt,
ProviderOptions: NewProviderOptions(&ProviderOptions{
ServiceTier: fantasy.Opt("flex"),
@@ -1825,10 +1852,11 @@ func TestDoGenerate(t *testing.T) {
server.prepareJSONResponse(map[string]any{})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-4o-mini")
result, err := model.Generate(context.Background(), fantasy.Call{
@@ -1858,13 +1886,14 @@ func TestDoGenerate(t *testing.T) {
server.prepareJSONResponse(map[string]any{})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-4o-mini")
- _, err := model.Generate(context.Background(), fantasy.Call{
+ _, err = model.Generate(context.Background(), fantasy.Call{
Prompt: testPrompt,
ProviderOptions: NewProviderOptions(&ProviderOptions{
ServiceTier: fantasy.Opt("priority"),
@@ -1894,10 +1923,11 @@ func TestDoGenerate(t *testing.T) {
server.prepareJSONResponse(map[string]any{})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-3.5-turbo")
result, err := model.Generate(context.Background(), fantasy.Call{
@@ -2201,10 +2231,11 @@ func TestDoStream(t *testing.T) {
"logprobs": testLogprobs,
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-3.5-turbo")
stream, err := model.Stream(context.Background(), fantasy.Call{
@@ -2257,10 +2288,11 @@ func TestDoStream(t *testing.T) {
server.prepareToolStreamResponse()
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-3.5-turbo")
stream, err := model.Stream(context.Background(), fantasy.Call{
@@ -2343,10 +2375,11 @@ func TestDoStream(t *testing.T) {
},
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-3.5-turbo")
stream, err := model.Stream(context.Background(), fantasy.Call{
@@ -2382,10 +2415,11 @@ func TestDoStream(t *testing.T) {
server.prepareErrorStreamResponse()
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-3.5-turbo")
stream, err := model.Stream(context.Background(), fantasy.Call{
@@ -2423,13 +2457,14 @@ func TestDoStream(t *testing.T) {
"content": []string{},
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-3.5-turbo")
- _, err := model.Stream(context.Background(), fantasy.Call{
+ _, err = model.Stream(context.Background(), fantasy.Call{
Prompt: testPrompt,
})
@@ -2471,10 +2506,11 @@ func TestDoStream(t *testing.T) {
},
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-3.5-turbo")
stream, err := model.Stream(context.Background(), fantasy.Call{
@@ -2521,10 +2557,11 @@ func TestDoStream(t *testing.T) {
},
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-3.5-turbo")
stream, err := model.Stream(context.Background(), fantasy.Call{
@@ -2564,13 +2601,14 @@ func TestDoStream(t *testing.T) {
"content": []string{},
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-3.5-turbo")
- _, err := model.Stream(context.Background(), fantasy.Call{
+ _, err = model.Stream(context.Background(), fantasy.Call{
Prompt: testPrompt,
ProviderOptions: NewProviderOptions(&ProviderOptions{
Store: fantasy.Opt(true),
@@ -2606,13 +2644,14 @@ func TestDoStream(t *testing.T) {
"content": []string{},
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-3.5-turbo")
- _, err := model.Stream(context.Background(), fantasy.Call{
+ _, err = model.Stream(context.Background(), fantasy.Call{
Prompt: testPrompt,
ProviderOptions: NewProviderOptions(&ProviderOptions{
Metadata: map[string]any{
@@ -2652,13 +2691,14 @@ func TestDoStream(t *testing.T) {
"content": []string{},
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("o3-mini")
- _, err := model.Stream(context.Background(), fantasy.Call{
+ _, err = model.Stream(context.Background(), fantasy.Call{
Prompt: testPrompt,
ProviderOptions: NewProviderOptions(&ProviderOptions{
ServiceTier: fantasy.Opt("flex"),
@@ -2694,13 +2734,14 @@ func TestDoStream(t *testing.T) {
"content": []string{},
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("gpt-4o-mini")
- _, err := model.Stream(context.Background(), fantasy.Call{
+ _, err = model.Stream(context.Background(), fantasy.Call{
Prompt: testPrompt,
ProviderOptions: NewProviderOptions(&ProviderOptions{
ServiceTier: fantasy.Opt("priority"),
@@ -2737,10 +2778,11 @@ func TestDoStream(t *testing.T) {
"model": "o1-preview",
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("o1-preview")
stream, err := model.Stream(context.Background(), fantasy.Call{
@@ -2783,10 +2825,11 @@ func TestDoStream(t *testing.T) {
},
})
- provider := New(
+ provider, err := New(
WithAPIKey("test-api-key"),
WithBaseURL(server.server.URL),
)
+ require.NoError(t, err)
model, _ := provider.LanguageModel("o1-preview")
stream, err := model.Stream(context.Background(), fantasy.Call{