Adjust system prompt

Author: intitni

Core/Sources/ChatContextCollectors/SystemInfoChatContextCollector/SystemInfoChatContextCollector.swift

@@ -19,10 +19,12 @@ public final class SystemInfoChatContextCollector: ChatContextCollector {
     ) -> ChatContext {
         return .init(
             systemPrompt: """
-                    Current Time: \(
-                        Self.dateFormatter.string(from: Date())
-                    ) (You can use it to calculate time in another time zone)
-                    """,
+            ## System Info
+            
+            Current Time: \(
+                Self.dateFormatter.string(from: Date())
+            ) (You can use it to calculate time in another time zone)
+            """,
             retrievedContent: [],
             functions: []
         )

Core/Sources/ChatService/DynamicContextController.swift

@@ -69,7 +69,7 @@ final class DynamicContextController {
         let extraSystemPrompt = contexts
             .map(\.systemPrompt)
             .filter { !$0.isEmpty }
-            .joined(separator: "\n")
+            .joined(separator: "\n\n")
 
         let contextPrompts = contexts
             .flatMap(\.retrievedContent)

Pro

@@ -101,9 +101,8 @@ public final class ActiveDocumentChatContextCollector: ChatContextCollector {
                 : "When you don't known what I am asking, I am probably referring to the code."
         )
 
-        Editing Document Context: ###
+        ### Editing Document Context
         """
-        let end = "###"
         let relativePath = "Document Relative Path: \(context.relativePath)"
         let language = "Language: \(context.language.rawValue)"
 
@@ -160,7 +159,6 @@ public final class ActiveDocumentChatContextCollector: ChatContextCollector {
                 code,
                 codeAnnotations,
                 fileAnnotations,
-                end,
             ]
             .filter { !$0.isEmpty }
             .joined(separator: "\n\n")
@@ -180,10 +178,9 @@ public final class ActiveDocumentChatContextCollector: ChatContextCollector {
                 language,
                 lineAnnotations,
                 selectionRange,
-                end,
             ]
             .filter { !$0.isEmpty }
-            .joined(separator: "\n")
+            .joined(separator: "\n\n")
         }
     }
 

Tool/Sources/ChatContextCollectors/ActiveDocumentChatContextCollector/ActiveDocumentChatContextCollector.swift

@@ -135,9 +135,13 @@ public actor AutoManagedChatGPTMemory: ChatGPTMemory {
         for (index, content) in retrievedContent.filter({ !$0.isEmpty }).enumerated() {
             if index == 0 {
                 if !appendToSystemPrompt("""
-
+                
+                
+                ## Relevant Content
+                
                 Below are information related to the conversation, separated by \(separator)
 
+                
                 """) { break }
             } else {
                 if !appendToSystemPrompt("\n\(separator)\n") { break }