Bug fix.

plasma-umass · Apr 11, 2024 · 73c4c16 · 73c4c16
1 parent cbc13ff
commit 73c4c16
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/coverup/coverup.py b/src/coverup/coverup.py
@@ -532,10 +532,8 @@ def pl(item, singular, plural = None):
 """
 
     content['claude'] = f"""
-<file path="{seg.filename}">
-<module name="{module_name}">
+<file path="{seg.filename}" module_name="{module_name}">
 {seg.get_excerpt()}
-</module>
 </file>
 
 <instructions>
@@ -680,7 +678,7 @@ def pl(item, singular, plural = None):
 """
             messages.append({
                 "role": "user",
-                "content": content[args.model]
+                "content": content[args.prompt_family]
             })
             log_write(seg, messages[-1]['content'])
             state.inc_counter('U')