Skip to content

Commit 3e4d4ee

Browse files
committed
Format file
1 parent 65aa3c2 commit 3e4d4ee

1 file changed

Lines changed: 14 additions & 9 deletions

File tree

Tool/Sources/OpenAIService/Memory/AutoManagedChatGPTMemoryStrategy/AutoManagedChatGPTMemoryGoogleAIStrategy.swift

Lines changed: 14 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -40,17 +40,19 @@ extension AutoManagedChatGPTMemory {
4040
reformattedHistory.append(message)
4141
continue
4242
}
43-
43+
4444
let lastMessage = reformattedHistory[lastIndex]
45-
46-
if ModelContent.convertRole(lastMessage.role) == ModelContent.convertRole(message.role) {
45+
46+
if ModelContent.convertRole(lastMessage.role) == ModelContent
47+
.convertRole(message.role)
48+
{
4749
let newMessage = ChatMessage(
4850
role: message.role == .assistant ? .assistant : .user,
4951
content: """
5052
\(ModelContent.convertContent(of: lastMessage))
51-
53+
5254
======
53-
55+
5456
\(ModelContent.convertContent(of: message))
5557
"""
5658
)
@@ -59,9 +61,12 @@ extension AutoManagedChatGPTMemory {
5961
reformattedHistory.append(message)
6062
}
6163
}
62-
64+
6365
if let newUserMessage {
64-
if let last = reformattedHistory.last, ModelContent.convertRole(last.role) == ModelContent.convertRole(newUserMessage.role) {
66+
if let last = reformattedHistory.last,
67+
ModelContent.convertRole(last.role) == ModelContent
68+
.convertRole(newUserMessage.role)
69+
{
6570
// Add dummy message
6671
let dummyMessage = ChatMessage(
6772
role: .assistant,
@@ -90,7 +95,7 @@ extension ModelContent {
9095
return "model"
9196
}
9297
}
93-
98+
9499
static func convertContent(of message: ChatMessage) -> String {
95100
switch message.role {
96101
case .user, .system, .function:
@@ -106,7 +111,7 @@ extension ModelContent {
106111
}
107112
}
108113
}
109-
114+
110115
init(_ message: ChatMessage) {
111116
let role = Self.convertRole(message.role)
112117
let parts = [ModelContent.Part.text(Self.convertContent(of: message))]

0 commit comments

Comments
 (0)