Add current time to system prompt

Author: intitni

Core/Package.swift

@@ -180,6 +180,7 @@ let package = Package(
                 // context collectors
                 "WebChatContextCollector",
                 "ActiveDocumentChatContextCollector",
+                "SystemInfoChatContextCollector",
 
                 .product(name: "AppMonitoring", package: "Tool"),
                 .product(name: "Environment", package: "Tool"),
@@ -346,6 +347,15 @@ let package = Package(
             path: "Sources/ChatContextCollectors/WebChatContextCollector"
         ),
 
+        .target(
+            name: "SystemInfoChatContextCollector",
+            dependencies: [
+                "ChatContextCollector",
+                .product(name: "OpenAIService", package: "Tool"),
+            ],
+            path: "Sources/ChatContextCollectors/SystemInfoChatContextCollector"
+        ),
+
         .target(
             name: "ActiveDocumentChatContextCollector",
             dependencies: [

Core/Sources/ChatContextCollectors/SystemInfoChatContextCollector/SystemInfoChatContextCollector.swift

@@ -0,0 +1,27 @@
+import ChatContextCollector
+import Foundation
+import OpenAIService
+
+public final class SystemInfoChatContextCollector: ChatContextCollector {
+    static let dateFormatter: DateFormatter = {
+        let formatter = DateFormatter()
+        formatter.dateFormat = "EEEE, yyyy-MM-dd HH:mm:ssZ"
+        return formatter
+    }()
+
+    public init() {}
+
+    public func generateContext(
+        history: [ChatMessage],
+        scopes: Set<String>,
+        content: String
+    ) -> ChatContext? {
+        return .init(
+            systemPrompt: """
+            Current Time: \(Self.dateFormatter.string(from: Date())) (You can use it to calculate time in another time zone)
+            """,
+            functions: []
+        )
+    }
+}
+

Core/Sources/ChatService/AllContextCollector.swift

@@ -1,8 +1,10 @@
 import ActiveDocumentChatContextCollector
 import ChatContextCollector
+import SystemInfoChatContextCollector
 import WebChatContextCollector
 
 let allContextCollectors: [any ChatContextCollector] = [
+    SystemInfoChatContextCollector(),
     ActiveDocumentChatContextCollector(),
     WebChatContextCollector(),
 ]