From a0d1e6244a94b7b306eb31f0aa3bd0319e3f8422 Mon Sep 17 00:00:00 2001
From: Emery Berger <emery.berger@gmail.com>
Date: Thu, 21 Mar 2024 17:32:41 -0400
Subject: [PATCH] Added Claude 3, parameter dropping.

---
 src/coverup/coverup.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/coverup/coverup.py b/src/coverup/coverup.py
index 9ec3ef6..4da6e80 100644
--- a/src/coverup/coverup.py
+++ b/src/coverup/coverup.py
@@ -28,6 +28,8 @@
 # Turn off most logging
 litellm.set_verbose = False
 logging.getLogger().setLevel(logging.ERROR)
+# Ignore unavailable parameters
+litellm.drop_params=True
 
 def parse_args(args=None):
     import argparse
@@ -656,8 +658,8 @@ def main():
         if 'OPENAI_ORGANIZATION' in os.environ:
             openai.organization=os.environ['OPENAI_ORGANIZATION']
     else:
-        args.model = "bedrock/anthropic.claude-v2:1"   
-
+        # args.model = "bedrock/anthropic.claude-v2:1"   
+        args.model = "bedrock/anthropic.claude-3-sonnet-20240229-v1:0"
     log_write('startup', f"Command: {' '.join(sys.argv)}")
 
     # --- (1) load or measure initial coverage, figure out segmentation ---