git.iko.soy/crush
on main help sign in
fix: check if token is expired before sending the request (#1641)
4529235Carlos Alexandro Becker
internal/agent/coordinator.go+7
compare with main
unifiedsplit
@@ -132,6 +132,13 @@ func (c *coordinator) Run(ctx context.Context, sessionID string, prompt string,
132
132
133 mergedOptions, temp, topP, topK, freqPenalty, presPenalty := mergeCallOptions(model, providerCfg)
133 mergedOptions, temp, topP, topK, freqPenalty, presPenalty := mergeCallOptions(model, providerCfg)
134
134
135+ if providerCfg.OAuthToken != nil && providerCfg.OAuthToken.IsExpired() {
136+ slog.Info("Token needs to be refreshed", "provider", providerCfg.ID)
137+ if err := c.refreshOAuth2Token(ctx, providerCfg); err != nil {
138+ return nil, err
139+ }
140+ }
141+
135 run := func() (*fantasy.AgentResult, error) {
142 run := func() (*fantasy.AgentResult, error) {
136 return c.currentAgent.Run(ctx, SessionAgentCall{
143 return c.currentAgent.Run(ctx, SessionAgentCall{
137 SessionID: sessionID,
144 SessionID: sessionID,