agent.go 339 B

1234567891011121314151617
  1. package agent
  2. import (
  3. "context"
  4. "fmt"
  5. "github.com/cloudwego/eino/flow/agent/react"
  6. )
  7. func GetAgent(ctx context.Context, name string) (*react.Agent, string, error) {
  8. switch name {
  9. case "coder":
  10. agent, err := NewCoderAgent(ctx)
  11. return agent, CoderSystemPrompt(), err
  12. }
  13. return nil, "", fmt.Errorf("agent %s not found", name)
  14. }