Update models.go (#17)

madebywelch · Mar 15, 2024 · 71e3db7 · 71e3db7
1 parent 69dc742
commit 71e3db7
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pkg/anthropic/models.go b/pkg/anthropic/models.go
@@ -13,7 +13,8 @@ const (
 
 	// Fastest and most compact model for near-instant responsiveness
 	// Claude 3 Haiku: Coming soon. Not yet available as of Mar 4 2024.
-
+	Claude3Haiku Model = "claude-3-haiku-20240307"
+
 	// Updated version of Claude 2 with improved accuracy
 	ClaudeV2_1 Model = "claude-2.1"
 
@@ -63,7 +64,7 @@ const (
 
 func (m Model) IsMessageCompatible() bool {
 	switch m {
-	case Claude3Opus, Claude3Sonnet, ClaudeV2_1:
+	case Claude3Opus, Claude3Sonnet, Claude3Haiku, ClaudeV2_1:
 		return true
 	}
 	return false