@@ -14,6 +14,7 @@ import (
1414 "github.com/docker/cagent/pkg/chat"
1515 latest "github.com/docker/cagent/pkg/config/v2"
1616 "github.com/docker/cagent/pkg/environment"
17+ "github.com/docker/cagent/pkg/httpclient"
1718 "github.com/docker/cagent/pkg/model/provider/base"
1819 "github.com/docker/cagent/pkg/model/provider/options"
1920 "github.com/docker/cagent/pkg/tools"
@@ -49,8 +50,9 @@ func NewClient(ctx context.Context, cfg *latest.ModelConfig, env environment.Pro
4950 }
5051
5152 client , err := genai .NewClient (ctx , & genai.ClientConfig {
52- APIKey : apiKey ,
53- Backend : genai .BackendGeminiAPI ,
53+ APIKey : apiKey ,
54+ Backend : genai .BackendGeminiAPI ,
55+ HTTPClient : httpclient .NewHttpClient (),
5456 })
5557 if err != nil {
5658 return nil , err
@@ -75,8 +77,9 @@ func NewClient(ctx context.Context, cfg *latest.ModelConfig, env environment.Pro
7577 }
7678
7779 return genai .NewClient (ctx , & genai.ClientConfig {
78- APIKey : authToken ,
79- Backend : genai .BackendGeminiAPI ,
80+ APIKey : authToken ,
81+ Backend : genai .BackendGeminiAPI ,
82+ HTTPClient : httpclient .NewHttpClient (),
8083 HTTPOptions : genai.HTTPOptions {
8184 BaseURL : gateway ,
8285 Headers : http.Header {
0 commit comments