Skip to content

Commit c382804

Browse files
committed
Refactor
Signed-off-by: Milen Pivchev <milen.pivchev@gmail.com>
1 parent 7fa7e17 commit c382804

2 files changed

Lines changed: 15 additions & 29 deletions

File tree

Sources/NextcloudKit/Models/Assistant/v2/Chat.swift

Lines changed: 4 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ import Foundation
66

77
// MARK: - ChatMessage
88

9-
public struct ChatMessage: Codable, Identifiable, Equatable {
9+
public struct AssistantChatMessage: Codable, Identifiable, Equatable {
1010
public let id: Int
1111
public let sessionId: Int
1212
public let role: String
@@ -36,7 +36,7 @@ public struct ChatMessage: Codable, Identifiable, Equatable {
3636

3737
// MARK: - ChatMessageRequest
3838

39-
public struct ChatMessageRequest: Encodable {
39+
public struct AssistantChatMessageRequest: Encodable {
4040
public let sessionId: Int
4141
public let role: String
4242
public let content: String
@@ -69,20 +69,6 @@ public struct ChatMessageRequest: Encodable {
6969
}
7070
}
7171

72-
// MARK: - Conversation
73-
74-
public struct Conversation: Codable, Identifiable, Equatable {
75-
public let id: Int
76-
public let title: String?
77-
public let timestamp: Int
78-
79-
enum CodingKeys: String, CodingKey {
80-
case id
81-
case title
82-
case timestamp
83-
}
84-
}
85-
8672
// MARK: - Session
8773

8874
public struct AssistantConversation: Codable, Equatable, Hashable {
@@ -114,7 +100,7 @@ public struct AssistantConversation: Codable, Equatable, Hashable {
114100

115101
// MARK: - CreateConversation
116102

117-
public struct CreateConversation: Codable, Equatable {
103+
public struct AssistantCreatedConversation: Codable, Equatable {
118104
public let conversation: AssistantConversation
119105

120106
enum CodingKeys: String, CodingKey {
@@ -142,7 +128,7 @@ public struct AssistantSession: Codable, Equatable {
142128

143129
// MARK: - SessionTask
144130

145-
public struct SessionTask: Codable, Equatable {
131+
public struct AssistantSessionTask: Codable, Equatable {
146132
public let taskId: Int
147133

148134
enum CodingKeys: String, CodingKey {

Sources/NextcloudKit/NextcloudKit+AssistantV2.swift

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -282,7 +282,7 @@ public extension NextcloudKit {
282282
taskHandler: @escaping (_ task: URLSessionTask) -> Void = { _ in }
283283
) async -> (
284284
account: String,
285-
chatMessages: [ChatMessage]?,
285+
chatMessages: [AssistantChatMessage]?,
286286
responseData: AFDataResponse<Data>?,
287287
error: NKError
288288
) {
@@ -308,7 +308,7 @@ public extension NextcloudKit {
308308
}
309309
case .success(let data):
310310
let decoder = JSONDecoder()
311-
let result = try? decoder.decode([ChatMessage].self, from: data)
311+
let result = try? decoder.decode([AssistantChatMessage].self, from: data)
312312
options.queue.async {
313313
continuation.resume(returning: (account: account, chatMessages: result, responseData: response, error: .success))
314314
}
@@ -325,13 +325,13 @@ public extension NextcloudKit {
325325
/// - options: Optional HTTP request configuration.
326326
/// - taskHandler: Optional closure to access the underlying URLSessionTask.
327327
/// - Returns: A tuple with named values for account, created message, response, and error.
328-
func createAssistantChatMessage(messageRequest: ChatMessageRequest,
328+
func createAssistantChatMessage(messageRequest: AssistantChatMessageRequest,
329329
account: String,
330330
options: NKRequestOptions = NKRequestOptions(),
331331
taskHandler: @escaping (_ task: URLSessionTask) -> Void = { _ in }
332332
) async -> (
333333
account: String,
334-
chatMessage: ChatMessage?,
334+
chatMessage: AssistantChatMessage?,
335335
responseData: AFDataResponse<Data>?,
336336
error: NKError
337337
) {
@@ -357,7 +357,7 @@ public extension NextcloudKit {
357357
}
358358
case .success(let data):
359359
let decoder = JSONDecoder()
360-
let result = try? decoder.decode(ChatMessage.self, from: data)
360+
let result = try? decoder.decode(AssistantChatMessage.self, from: data)
361361
options.queue.async {
362362
continuation.resume(returning: (account: account, chatMessage: result, responseData: response, error: .success))
363363
}
@@ -382,7 +382,7 @@ public extension NextcloudKit {
382382
taskHandler: @escaping (_ task: URLSessionTask) -> Void = { _ in }
383383
) async -> (
384384
account: String,
385-
conversation: CreateConversation?,
385+
conversation: AssistantCreatedConversation?,
386386
responseData: AFDataResponse<Data>?,
387387
error: NKError
388388
) {
@@ -413,7 +413,7 @@ public extension NextcloudKit {
413413
}
414414
case .success(let data):
415415
let decoder = JSONDecoder()
416-
let result = try? decoder.decode(CreateConversation.self, from: data)
416+
let result = try? decoder.decode(AssistantCreatedConversation.self, from: data)
417417
options.queue.async {
418418
continuation.resume(returning: (account: account, conversation: result, responseData: response, error: .success))
419419
}
@@ -438,7 +438,7 @@ public extension NextcloudKit {
438438
taskHandler: @escaping (_ task: URLSessionTask) -> Void = { _ in }
439439
) async -> (
440440
account: String,
441-
chatMessage: ChatMessage?,
441+
chatMessage: AssistantChatMessage?,
442442
responseData: AFDataResponse<Data>?,
443443
error: NKError
444444
) {
@@ -466,7 +466,7 @@ public extension NextcloudKit {
466466
}
467467
case .success(let data):
468468
let decoder = JSONDecoder()
469-
let result = try? decoder.decode(ChatMessage.self, from: data)
469+
let result = try? decoder.decode(AssistantChatMessage.self, from: data)
470470
options.queue.async {
471471
continuation.resume(returning: (account: account, chatMessage: result, responseData: response, error: .success))
472472
}
@@ -489,7 +489,7 @@ public extension NextcloudKit {
489489
taskHandler: @escaping (_ task: URLSessionTask) -> Void = { _ in }
490490
) async -> (
491491
account: String,
492-
sessionTask: SessionTask?,
492+
sessionTask: AssistantSessionTask?,
493493
responseData: AFDataResponse<Data>?,
494494
error: NKError
495495
) {
@@ -517,7 +517,7 @@ public extension NextcloudKit {
517517
}
518518
case .success(let data):
519519
let decoder = JSONDecoder()
520-
let result = try? decoder.decode(SessionTask.self, from: data)
520+
let result = try? decoder.decode(AssistantSessionTask.self, from: data)
521521
options.queue.async {
522522
continuation.resume(returning: (account: account, sessionTask: result, responseData: response, error: .success))
523523
}

0 commit comments

Comments
 (0)