Added Claude 3, parameter dropping.

plasma-umass · Mar 21, 2024 · a0d1e62 · a0d1e62
1 parent 73a1162
commit a0d1e62
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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 ---