Candy/Nahi
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

cleanup

Browse Source
This commit is contained in:
Candygoblen123
2025-08-20 13:17:13 -04:00
parent ac6e6ae73d
commit 47b3c75740
1 changed files with 44 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

55
Sources/nahi/nahi.swift
View File

@@ -23,8 +23,16 @@ struct Nahi {
let bot = await BotGatewayManager(
token: discordToken,
presence: .init(activities: [.init(name: "Vibing", type: .competing)], status: .online, afk: false),
intents: [.guildMessages, .messageContent, .directMessages]
presence: .init(
activities: [.init(name: "Vibing", type: .competing)],
status: .online,
afk: false
),
intents: [
.guildMessages,
.messageContent,
//.directMessages
]
)
let gemini = GeminiKit(
@@ -59,7 +67,13 @@ struct Nahi {
taskGroup.addTask {
for await event in await bot.events {
await EventHandler(event: event, client: bot.client, gemini: gemini, log: log, aiChat: aiChat).handleAsync()
await EventHandler(
event: event,
client: bot.client,
gemini: gemini,
log: log,
aiChat: aiChat
).handleAsync()
}
}
}
@@ -74,17 +88,21 @@ struct EventHandler: GatewayEventHandler {
let aiChat: Chat
func onMessageCreate(_ payload: Gateway.MessageCreate) async throws {
let channel = try? await client.getChannel(id: payload.channel_id)
guard
payload.author!.id.rawValue != client.appId!.rawValue,
payload.mentions.contains(where: { mention in mention.id.rawValue == client.appId!.rawValue }) || (try? channel?.decode().type ?? .guildText) == .dm
payload.mentions.contains(where: { mention in
mention.id.rawValue == client.appId!.rawValue
})
else { return }
_ = try await client.triggerTypingIndicator(channelId: payload.channel_id)
let prompt = "\(payload.author!.global_name!): \(payload.content.replacingOccurrences(of: "<@\(client.appId!.rawValue)>", with: "@Nahida"))"
let author = payload.author?.global_name ?? payload.author?.username ?? "Unknown"
let message = payload.content
.replacingOccurrences(of: "<@\(client.appId!.rawValue)>", with: "@Nahida")
let prompt = "\(author): \(message)"
let aiRes = await Task {
//return try await gemini.generateContent(model: .gemini20Flash, prompt: prompt, systemInstruction: "You are a discord chat bot pretending to be Nahida, the dendro archon from the game Genshin Impact. Act like Nahida in your responces.")
return try await aiChat.sendMessage(prompt)
}.result
@@ -98,19 +116,34 @@ struct EventHandler: GatewayEventHandler {
for text in resStr {
let messageRes = try await client.createMessage(
channelId: payload.channel_id,
payload: .init(content: text, message_reference: .init(message_id: payload.id, channel_id: payload.channel_id, guild_id: payload.guild_id, fail_if_not_exists: false))
payload: .init(
content: text,
message_reference: .init(
message_id: payload.id,
channel_id: payload.channel_id,
guild_id: payload.guild_id,
fail_if_not_exists: false
)
)
)
try messageRes.guardSuccess()
}
case .failure(let err):
let errMessage = "Someone tell <@259709415416922113> there is a problem with my AI: \(String(describing: err))"
try await client.createMessage(
channelId: payload.channel_id,
payload: .init(content: "Someone tell <@259709415416922113> there is a problem with my AI: \(String(describing: err))", message_reference: .init(message_id: payload.id, channel_id: payload.channel_id, guild_id: payload.guild_id, fail_if_not_exists: false))
payload: .init(
content: errMessage,
message_reference: .init(
message_id: payload.id,
channel_id: payload.channel_id,
guild_id: payload.guild_id,
fail_if_not_exists: false
)
)
).guardSuccess()
log.error("\(err)")
}
}
}