Add one-shot example (#4103)

sweepai · Jun 23, 2024 · ec7ccd7 · ec7ccd7
1 parent ef78d6a
commit ec7ccd7
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 4 deletions.
diff --git a/sweep_chat/components/App.tsx b/sweep_chat/components/App.tsx
@@ -64,6 +64,7 @@ import { Label } from './ui/label'
 import PulsingLoader from './shared/PulsingLoader'
 import {
   DEFAULT_K,
+  DEFAULT_MODEL,
   modelMap,
 } from '@/lib/constants'
 import {
@@ -116,7 +117,7 @@ function App({ defaultMessageId = '' }: { defaultMessageId?: string }) {
   const [k, setK] = useLocalStorage<number>('k', DEFAULT_K)
   const [model, setModel] = useLocalStorage<keyof typeof modelMap>(
     'model',
-    'gpt-4o'
+    DEFAULT_MODEL
   )
   const [snippets, setSnippets] = useState<Snippet[]>([])
   const [searchMessage, setSearchMessage] = useState<string>('')

diff --git a/sweep_chat/lib/constants.ts b/sweep_chat/lib/constants.ts
@@ -38,11 +38,13 @@ const languageMapping: Record<string, Extension> = {
 }
 
 const DEFAULT_K: number = 8
+const DEFAULT_MODEL = 'claude-3-5-sonnet-20240620'
 
 export {
   codeStyle,
   modelMap,
   DEFAULT_K,
+  DEFAULT_MODEL,
   roleToColor,
   typeNameToColor,
   languageMapping,

diff --git a/sweepai/chat/search_prompts.py b/sweepai/chat/search_prompts.py
@@ -63,10 +63,20 @@
 [the function call goes here, using the valid XML format for function calls]
 </function_call>""" + example_tool_calls
 
-anthropic_format_message = """You MUST follow the following XML-based format, including <analysis> and <user_response> XML blocks:
+anthropic_format_message = """### Guidelines
 
-### Format
+<examples>
+If the user asks you to "Review this PR" or "Explain this PR", your organized response should contain the following sections:
+Overview of the PR
+Changes in file_1
+Changes in file_2
+Impact and purpose of the changes
+</examples>
+
+You MUST follow the following XML-based format, including <analysis> and <user_response> XML blocks:
 
+### Format
+<format>
 You must respond with the following two distinct sections:
 
 # 1. Summary and analysis
@@ -87,7 +97,8 @@
 When showing relevant examples of code, only show MINIMAL excerpts of code that address the user's question. Do NOT copy the whole file, but only the lines that are relevant to the user's question.
 
 To suggest code changes, first list each section of each file that you would like to change. Then for each section, write a <code_change> block for that change. These changes should be atomic -- to change multiple parts of the file, you MUST write multiple separate <code_change> blocks.
-</user_response>"""
+</user_response>
+</format>"""
 
 openai_format_message = """You MUST follow the following XML-based format, including <user_response> and </user_respose> tags: