Adjust interface

Author: intitni

Pro

@@ -3,44 +3,10 @@ import OpenAIService
 
 public struct ChatContext {
     public struct RetrievedContent {
-        public enum Priority: Equatable, Comparable {
-            case bottom
-            case low
-            case medium
-            case high
-            case top
-            case custom(Int)
-
-            public var rawValue: Int {
-                switch self {
-                case .bottom:
-                    return 0
-                case .low:
-                    return 400
-                case .medium:
-                    return 600
-                case .high:
-                    return 800
-                case .top:
-                    return 1_000_000_000
-                case let .custom(value):
-                    return value
-                }
-            }
-
-            public static func < (lhs: Self, rhs: Self) -> Bool {
-                lhs.rawValue < rhs.rawValue
-            }
-
-            public static func == (lhs: Self, rhs: Self) -> Bool {
-                lhs.rawValue == rhs.rawValue
-            }
-        }
-
         public var content: String
-        public var priority: Priority
+        public var priority: Int
 
-        public init(content: String, priority: Priority) {
+        public init(content: String, priority: Int) {
             self.content = content
             self.priority = priority
         }
@@ -64,20 +30,6 @@ public struct ChatContext {
     }
 }
 
-public func + (
-    lhs: ChatContext.RetrievedContent.Priority,
-    rhs: Int
-) -> ChatContext.RetrievedContent.Priority {
-    .custom(lhs.rawValue + rhs)
-}
-
-public func - (
-    lhs: ChatContext.RetrievedContent.Priority,
-    rhs: Int
-) -> ChatContext.RetrievedContent.Priority {
-    .custom(lhs.rawValue - rhs)
-}
-
 public protocol ChatContextCollector {
     func generateContext(
         history: [ChatMessage],

Tool/Sources/ChatContextCollector/ChatContextCollector.swift

@@ -27,13 +27,8 @@ public final class ActiveDocumentChatContextCollector: ChatContextCollector {
                 var removedCode = context
                 removedCode.focusedContext = nil
                 return .init(
-                    systemPrompt: "",
-                    retrievedContent: [
-                        .init(
-                            content: extractSystemPrompt(removedCode),
-                            priority: .high
-                        ),
-                    ],
+                    systemPrompt: extractSystemPrompt(removedCode),
+                    retrievedContent: [],
                     functions: []
                 )
             }
@@ -71,10 +66,8 @@ public final class ActiveDocumentChatContextCollector: ChatContextCollector {
         }
 
         return .init(
-            systemPrompt: "",
-            retrievedContent: [
-                .init(content: extractSystemPrompt(context), priority: .high)
-            ],
+            systemPrompt: extractSystemPrompt(context),
+            retrievedContent: [],
             functions: functions
         )
     }

Tool/Sources/ChatContextCollectors/ActiveDocumentChatContextCollector/ActiveDocumentChatContextCollector.swift

@@ -78,9 +78,7 @@ public struct LegacyActiveDocumentChatContextCollector: ChatContextCollector {
         }()
 
         return .init(
-            systemPrompt: "",
-            retrievedContent: [
-                .init(content: """
+            systemPrompt: """
                 Active Document Context:###
                 Document Relative Path: \(relativePath)
                 Selection Range Start: \
@@ -100,8 +98,8 @@ public struct LegacyActiveDocumentChatContextCollector: ChatContextCollector {
                         .joined(separator: "\n") ?? "N/A"
                 )
                 ###
-                """, priority: .high),
-            ],
+                """,
+            retrievedContent: [],
             functions: []
         )
     }