mirror of
https://github.com/charmbracelet/crush.git
synced 2025-08-02 05:20:46 +03:00
fix: method name
This commit is contained in:
@@ -23,8 +23,8 @@ func NewLazySlice[K any](load func() []K) *LazySlice[K] {
|
||||
return s
|
||||
}
|
||||
|
||||
// Iter returns an iterator that yields elements from the slice.
|
||||
func (s *LazySlice[K]) Iter() iter.Seq[K] {
|
||||
// Seq returns an iterator that yields elements from the slice.
|
||||
func (s *LazySlice[K]) Seq() iter.Seq[K] {
|
||||
s.mu.Lock()
|
||||
inner := s.inner
|
||||
s.mu.Unlock()
|
||||
|
||||
@@ -8,7 +8,7 @@ import (
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
func TestLazySlice_Iter(t *testing.T) {
|
||||
func TestLazySlice_Seq(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
data := []string{"a", "b", "c"}
|
||||
@@ -19,14 +19,14 @@ func TestLazySlice_Iter(t *testing.T) {
|
||||
})
|
||||
|
||||
var result []string
|
||||
for v := range s.Iter() {
|
||||
for v := range s.Seq() {
|
||||
result = append(result, v)
|
||||
}
|
||||
|
||||
assert.Equal(t, data, result)
|
||||
}
|
||||
|
||||
func TestLazySlice_IterWaitsForLoading(t *testing.T) {
|
||||
func TestLazySlice_SeqWaitsForLoading(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
var loaded atomic.Bool
|
||||
@@ -42,11 +42,11 @@ func TestLazySlice_IterWaitsForLoading(t *testing.T) {
|
||||
assert.False(t, loaded.Load(), "should not be loaded immediately")
|
||||
|
||||
var result []string
|
||||
for v := range s.Iter() {
|
||||
for v := range s.Seq() {
|
||||
result = append(result, v)
|
||||
}
|
||||
|
||||
assert.True(t, loaded.Load(), "should be loaded after Iter")
|
||||
assert.True(t, loaded.Load(), "should be loaded after Seq")
|
||||
assert.Equal(t, data, result)
|
||||
}
|
||||
|
||||
@@ -58,7 +58,7 @@ func TestLazySlice_EmptySlice(t *testing.T) {
|
||||
})
|
||||
|
||||
var result []string
|
||||
for v := range s.Iter() {
|
||||
for v := range s.Seq() {
|
||||
result = append(result, v)
|
||||
}
|
||||
|
||||
@@ -76,7 +76,7 @@ func TestLazySlice_EarlyBreak(t *testing.T) {
|
||||
})
|
||||
|
||||
var result []string
|
||||
for v := range s.Iter() {
|
||||
for v := range s.Seq() {
|
||||
result = append(result, v)
|
||||
if len(result) == 2 {
|
||||
break
|
||||
|
||||
@@ -445,7 +445,7 @@ func (a *agent) createUserMessage(ctx context.Context, sessionID, content string
|
||||
|
||||
func (a *agent) streamAndHandleEvents(ctx context.Context, sessionID string, msgHistory []message.Message) (message.Message, *message.Message, error) {
|
||||
ctx = context.WithValue(ctx, tools.SessionIDContextKey, sessionID)
|
||||
eventChan := a.provider.StreamResponse(ctx, msgHistory, slices.Collect(a.tools.Iter()))
|
||||
eventChan := a.provider.StreamResponse(ctx, msgHistory, slices.Collect(a.tools.Seq()))
|
||||
|
||||
assistantMsg, err := a.messages.Create(ctx, sessionID, message.CreateMessageParams{
|
||||
Role: message.Assistant,
|
||||
@@ -494,7 +494,7 @@ func (a *agent) streamAndHandleEvents(ctx context.Context, sessionID string, msg
|
||||
default:
|
||||
// Continue processing
|
||||
var tool tools.BaseTool
|
||||
for availableTool := range a.tools.Iter() {
|
||||
for availableTool := range a.tools.Seq() {
|
||||
if availableTool.Info().Name == toolCall.Name {
|
||||
tool = availableTool
|
||||
break
|
||||
|
||||
Reference in New Issue
Block a user