Function createLocalAgent

internal/multi_test.go:147–181

(transform func(string) string)

}

func createLocalAgent(transform func(string) string) (*agent.LocalAgent, error) {
	processFunc := func(ctx context.Context, conversationID string, execID string, start *proto.AgentStart, e agent.Executor, handler agent.OutputHandler) error {
		for _, msg := range start.Messages {
			content := msg.GetContent()
			if content == nil {
				continue
			}
			textContent := content.GetText()
			if textContent == nil {
				continue
			}
			if err := handler(&proto.AgentOutputs{
				Messages: []*proto.Message{
					{
						Role: "assistant",
						Content: &proto.Content{
							Type: &proto.Content_Text{
								Text: &proto.TextContent{
									Text: transform(textContent.Text),
								},
							},
						},
					},
				},
			}); err != nil {
				return err
			}
		}
		return nil
	}

	return agent.NewLocalAgent(agent.LocalAgentConfig{
		ProcessFunc: processFunc,
	})
}

type mockPlanner struct{}