From e7258e38aeb46281fda474b8b7fcc3eee35edd9f Mon Sep 17 00:00:00 2001 From: Kujtim Hoxha Date: Sun, 23 Mar 2025 22:25:31 +0100 Subject: initial agent setup --- internal/llm/agent/agent.go | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) create mode 100644 internal/llm/agent/agent.go (limited to 'internal/llm/agent/agent.go') diff --git a/internal/llm/agent/agent.go b/internal/llm/agent/agent.go new file mode 100644 index 000000000..e02fc0f78 --- /dev/null +++ b/internal/llm/agent/agent.go @@ -0,0 +1,17 @@ +package agent + +import ( + "context" + "fmt" + + "github.com/cloudwego/eino/flow/agent/react" +) + +func GetAgent(ctx context.Context, name string) (*react.Agent, string, error) { + switch name { + case "coder": + agent, err := NewCoderAgent(ctx) + return agent, CoderSystemPrompt(), err + } + return nil, "", fmt.Errorf("agent %s not found", name) +} -- cgit v1.2.3