Skip to content

Commit ea3cb5d

Browse files
committed
Don't compute the messages 3 times
Signed-off-by: David Gageot <david.gageot@docker.com>
1 parent 0f65d42 commit ea3cb5d

1 file changed

Lines changed: 6 additions & 6 deletions

File tree

pkg/runtime/runtime.go

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -160,15 +160,16 @@ func (r *runtime) RunStream(ctx context.Context, sess *session.Session) <-chan E
160160
telemetryClient.RecordSessionStart(ctx, r.currentAgent, sess.ID)
161161
}
162162

163+
a := r.team.Agent(r.currentAgent)
164+
model := a.Model()
165+
modelID := model.ID()
166+
167+
messages := sess.GetMessages(a)
163168
if sess.SendUserMessage {
164-
events <- UserMessage(sess.GetMessages(r.CurrentAgent())[len(sess.GetMessages(r.CurrentAgent()))-1].Content)
169+
events <- UserMessage(messages[len(messages)-1].Content)
165170
}
166171

167172
events <- StreamStarted()
168-
a := r.team.Agent(r.currentAgent)
169-
170-
model := a.Model()
171-
modelID := model.ID()
172173

173174
defer r.finalizeEventChannel(ctx, sess, events)
174175

@@ -226,7 +227,6 @@ func (r *runtime) RunStream(ctx context.Context, sess *session.Session) <-chan E
226227
return
227228
}
228229
slog.Debug("Starting conversation loop iteration", "agent", a.Name())
229-
messages := sess.GetMessages(a)
230230
slog.Debug("Retrieved messages for processing", "agent", a.Name(), "message_count", len(messages))
231231

232232
// Retry loop for getting agent tools with OAuth handling

0 commit comments

Comments
 (0)