feat: support Sourcegraph Amp cli (#43)

Author: Harsh9485

README.md

@@ -1,6 +1,6 @@
 # AgentAPI
 
-Control [Claude Code](https://github.com/anthropics/claude-code), [Goose](https://github.com/block/goose), [Aider](https://github.com/Aider-AI/aider), [Gemini](https://github.com/google-gemini/gemini-cli) and [Codex](https://github.com/openai/codex) with an HTTP API.
+Control [Claude Code](https://github.com/anthropics/claude-code), [Goose](https://github.com/block/goose), [Aider](https://github.com/Aider-AI/aider), [Gemini](https://github.com/google-gemini/gemini-cli), [Sourcegraph Amp](https://github.com/sourcegraph/amp-cli) and [Codex](https://github.com/openai/codex) with an HTTP API.
 
 ![agentapi-chat](https://github.com/user-attachments/assets/57032c9f-4146-4b66-b219-09e38ab7690d)
 

cmd/server/server.go

@@ -28,6 +28,7 @@ const (
 	AgentTypeAider  AgentType = msgfmt.AgentTypeAider
 	AgentTypeCodex  AgentType = msgfmt.AgentTypeCodex
 	AgentTypeGemini AgentType = msgfmt.AgentTypeGemini
+	AgentTypeAmp    AgentType = msgfmt.AgentTypeAmp
 	AgentTypeCustom AgentType = msgfmt.AgentTypeCustom
 )
 
@@ -38,6 +39,7 @@ var agentTypeMap = map[AgentType]bool{
 	AgentTypeAider:  true,
 	AgentTypeCodex:  true,
 	AgentTypeGemini: true,
+	AgentTypeAmp:    true,
 	AgentTypeCustom: true,
 }
 

cmd/server/server_test.go

@@ -47,6 +47,11 @@ func TestParseAgentType(t *testing.T) {
 			agentTypeVar: "",
 			want:         AgentTypeGemini,
 		},
+		{
+			firstArg:     "amp",
+			agentTypeVar: "",
+			want:         AgentTypeAmp,
+		},
 		{
 			firstArg:     "goose",
 			agentTypeVar: "",

lib/msgfmt/msgfmt.go

@@ -206,6 +206,7 @@ const (
 	AgentTypeAider  AgentType = "aider"
 	AgentTypeCodex  AgentType = "codex"
 	AgentTypeGemini AgentType = "gemini"
+	AgentTypeAmp    AgentType = "amp"
 	AgentTypeCustom AgentType = "custom"
 )
 
@@ -235,6 +236,8 @@ func FormatAgentMessage(agentType AgentType, message string, userInput string) s
 		return formatCodexMessage(message, userInput)
 	case AgentTypeGemini:
 		return formatGenericMessage(message, userInput)
+	case AgentTypeAmp:
+		return formatGenericMessage(message, userInput)
 	case AgentTypeCustom:
 		return formatGenericMessage(message, userInput)
 	default:

lib/msgfmt/msgfmt_test.go

@@ -218,7 +218,7 @@ func TestTrimEmptyLines(t *testing.T) {
 
 func TestFormatAgentMessage(t *testing.T) {
 	dir := "testdata/format"
-	agentTypes := []AgentType{AgentTypeClaude, AgentTypeGoose, AgentTypeAider, AgentTypeGemini, AgentTypeCodex, AgentTypeCustom}
+	agentTypes := []AgentType{AgentTypeClaude, AgentTypeGoose, AgentTypeAider, AgentTypeGemini, AgentTypeAmp, AgentTypeCodex, AgentTypeCustom}
 	for _, agentType := range agentTypes {
 		t.Run(string(agentType), func(t *testing.T) {
 			cases, err := testdataDir.ReadDir(path.Join(dir, string(agentType)))

lib/msgfmt/testdata/format/amp/first_message/expected.txt

@@ -0,0 +1 @@
+    Welcome to Amp

lib/msgfmt/testdata/format/amp/first_message/msg.txt

@@ -0,0 +1,5 @@
+    Welcome to Amp
+
+╭──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
+│                                                                                                                                                  │
+╰──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯

lib/msgfmt/testdata/format/amp/first_message/user.txt

@@ -0,0 +1,13 @@
+∴ Thinking
+  The user is asking me to identify which file contains this specific TypeScript code snippet. The code shows a useEffect hook that sets up polling
+for messages and server status. I need to search for this code in the codebase to find the file it belongs to.
+
+  I should use the Grep tool to search for some unique parts of this code to locate the file.
+
+✓ Grep Set up polling for messages and server status
+
+✓ Grep const messageInterval = setInterval\(fetchMessages, 1000\)
+
+✓ List Directory /mnt/c/Users/Username/test-amp
+
+No files found in the workspace containing this code.

lib/msgfmt/testdata/format/amp/multi-line-input/expected.txt

@@ -0,0 +1,33 @@
+┃ Which file is this code from?
+┃
+┃   ```ts
+┃   // Set up polling for messages and server status
+┃   useEffect(() => {
+┃     // Check server status initially
+┃     checkServerStatus();
+┃
+┃     // Set up polling intervals
+┃     const messageInterval = setInterval(fetchMessages, 1000);
+┃     const statusInterval = setInterval(checkServerStatus, 250);
+┃
+┃     // Clean up intervals on component unmount
+┃     return () => {
+┃       clearInterval(messageInterval);
+┃       clearInterval(statusInterval);
+┃     };
+┃   }, []);
+┃   ```
+
+∴ Thinking
+  The user is asking me to identify which file contains this specific TypeScript code snippet. The code shows a useEffect hook that sets up polling
+for messages and server status. I need to search for this code in the codebase to find the file it belongs to.
+
+  I should use the Grep tool to search for some unique parts of this code to locate the file.
+
+✓ Grep Set up polling for messages and server status
+
+✓ Grep const messageInterval = setInterval\(fetchMessages, 1000\)
+
+✓ List Directory /mnt/c/Users/Username/test-amp
+
+No files found in the workspace containing this code.