Skip to content

Commit 4265af5

Browse files
committed
Those events are not linked to an agent
Signed-off-by: David Gageot <david.gageot@docker.com>
1 parent 5a76853 commit 4265af5

2 files changed

Lines changed: 6 additions & 6 deletions

File tree

pkg/runtime/event.go

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -558,12 +558,12 @@ type RAGIndexingStartedEvent struct {
558558
StrategyName string `json:"strategy_name"`
559559
}
560560

561-
func RAGIndexingStarted(ragName, strategyName, agentName string) Event {
561+
func RAGIndexingStarted(ragName, strategyName string) Event {
562562
return &RAGIndexingStartedEvent{
563563
Type: "rag_indexing_started",
564564
RAGName: ragName,
565565
StrategyName: strategyName,
566-
AgentContext: newAgentContext(agentName),
566+
AgentContext: newAgentContext(""),
567567
}
568568
}
569569

@@ -596,12 +596,12 @@ type RAGIndexingCompletedEvent struct {
596596
StrategyName string `json:"strategy_name"`
597597
}
598598

599-
func RAGIndexingCompleted(ragName, strategyName, agentName string) Event {
599+
func RAGIndexingCompleted(ragName, strategyName string) Event {
600600
return &RAGIndexingCompletedEvent{
601601
Type: "rag_indexing_completed",
602602
RAGName: ragName,
603603
StrategyName: strategyName,
604-
AgentContext: newAgentContext(agentName),
604+
AgentContext: newAgentContext(""),
605605
}
606606
}
607607

pkg/runtime/rag.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -50,13 +50,13 @@ func (r *LocalRuntime) forwardRAGEvents(ctx context.Context, ragManagers []*rag.
5050

5151
switch ragEvent.Type {
5252
case types.EventTypeIndexingStarted:
53-
sendEvent(RAGIndexingStarted(ragName, ragEvent.StrategyName, agentName))
53+
sendEvent(RAGIndexingStarted(ragName, ragEvent.StrategyName))
5454
case types.EventTypeIndexingProgress:
5555
if ragEvent.Progress != nil {
5656
sendEvent(RAGIndexingProgress(ragName, ragEvent.StrategyName, ragEvent.Progress.Current, ragEvent.Progress.Total, agentName))
5757
}
5858
case types.EventTypeIndexingComplete:
59-
sendEvent(RAGIndexingCompleted(ragName, ragEvent.StrategyName, agentName))
59+
sendEvent(RAGIndexingCompleted(ragName, ragEvent.StrategyName))
6060
case types.EventTypeUsage:
6161
// Convert RAG usage to TokenUsageEvent so TUI displays it
6262
sendEvent(NewTokenUsageEvent("", agentName, &Usage{

0 commit comments

Comments
 (0)