Skip to content

Commit

Permalink
refactor: improve title context for openai (#10)
Browse files Browse the repository at this point in the history
  • Loading branch information
@tim-smart
tim-smart authored May 1, 2023
1 parent f447cea commit b14bc72
Show file tree
Hide file tree
Showing 4 changed files with 209 additions and 175 deletions.
332 changes: 170 additions & 162 deletions src/AutoThreads.ts
View file
Original file line number Diff line number Diff line change
@@ -1,7 +1,9 @@
import { ChannelsCache, ChannelsCacheLive } from "bot/ChannelsCache"
import { OpenAI, OpenAIError } from "bot/OpenAI"
import * as Str from "bot/utils/String"
import {
Cause,
Config,
Data,
Duration,
Effect,
Expand Down Expand Up @@ -30,188 +32,194 @@ export class NotValidMessageError extends Data.TaggedClass(
readonly reason: "non-default" | "from-bot" | "non-text-channel" | "disabled"
}> {}

const truncate = (str: string, len: number) =>
str.length > len ? str.substring(0, len - 3) + "..." : str
export interface AutoThreadsOptions {
readonly topicKeyword: string
}

const make = Effect.gen(function* ($) {
const log = yield* $(Log.Log)
const openai = yield* $(OpenAI)
const gateway = yield* $(DiscordGateway)
const rest = yield* $(DiscordREST)
const channels = yield* $(ChannelsCache)
const make = ({ topicKeyword }: AutoThreadsOptions) =>
Effect.gen(function* ($) {
const log = yield* $(Log.Log)
const openai = yield* $(OpenAI)
const gateway = yield* $(DiscordGateway)
const rest = yield* $(DiscordREST)
const channels = yield* $(ChannelsCache)

const handleMessages = gateway.handleDispatch("MESSAGE_CREATE", message =>
pipe(
Effect.allPar({
message: Effect.cond(
() => message.type === Discord.MessageType.DEFAULT,
() => message,
() => new NotValidMessageError({ reason: "non-default" }),
const handleMessages = gateway.handleDispatch("MESSAGE_CREATE", message =>
pipe(
Effect.allPar({
message: Effect.cond(
() => message.type === Discord.MessageType.DEFAULT,
() => message,
() => new NotValidMessageError({ reason: "non-default" }),
),
channel: channels.get(message.guild_id!, message.channel_id),
}),
Effect.filterOrFail(
() => message.author.bot !== true,
() => new NotValidMessageError({ reason: "from-bot" }),
),
channel: channels.get(message.guild_id!, message.channel_id),
}),
Effect.filterOrFail(
() => message.author.bot !== true,
() => new NotValidMessageError({ reason: "from-bot" }),
),
Effect.filterOrFail(
({ channel }) => channel.type === Discord.ChannelType.GUILD_TEXT,
() => new NotValidMessageError({ reason: "non-text-channel" }),
),
Effect.filterOrFail(
({ channel }) => channel.topic?.includes("[threads]") === true,
() => new NotValidMessageError({ reason: "disabled" }),
),
Effect.bind("title", () =>
pipe(
Option.fromNullable(message.content),
Option.filter(_ => _.trim().length > 0),
Effect.flatMap(content =>
pipe(
openai.generateTitle(content),
Effect.retry(retryPolicy),
Effect.tapError(_ => log.info(_)),
Effect.filterOrFail(
({ channel }) => channel.type === Discord.ChannelType.GUILD_TEXT,
() => new NotValidMessageError({ reason: "non-text-channel" }),
),
Effect.filterOrFail(
({ channel }) => channel.topic?.includes(topicKeyword) === true,
() => new NotValidMessageError({ reason: "disabled" }),
),
Effect.bind("title", () =>
pipe(
Str.nonEmpty(message.content),
Effect.flatMap(content =>
pipe(
openai.generateTitle(content),
Effect.retry(retryPolicy),
Effect.tapError(_ => log.info(_)),
),
),
),
Effect.orElseSucceed(() =>
pipe(
Option.fromNullable(message.member?.nick),
Option.getOrElse(() => message.author.username),
_ => `${_}'s thread`,
Effect.orElseSucceed(() =>
pipe(
Option.fromNullable(message.member?.nick),
Option.getOrElse(() => message.author.username),
_ => `${_}'s thread`,
),
),
),
),
),
Effect.flatMap(({ channel, title }) =>
rest.startThreadFromMessage(channel.id, message.id, {
name: truncate(title, 100),
}),
),
Effect.flatMap(_ => _.json),
Effect.flatMap(thread =>
rest.createMessage(thread.id, {
components: UI.grid([
[
UI.button({
custom_id: `edit_${message.author.id}`,
label: "Edit title",
}),
UI.button({
custom_id: `archive_${message.author.id}`,
label: "Archive",
style: Discord.ButtonStyle.SECONDARY,
}),
],
]),
Effect.flatMap(({ channel, title }) =>
rest.startThreadFromMessage(channel.id, message.id, {
name: Str.truncate(title, 100),
}),
),
Effect.flatMap(_ => _.json),
Effect.flatMap(thread =>
rest.createMessage(thread.id, {
components: UI.grid([
[
UI.button({
custom_id: `edit_${message.author.id}`,
label: "Edit title",
}),
UI.button({
custom_id: `archive_${message.author.id}`,
label: "Archive",
style: Discord.ButtonStyle.SECONDARY,
}),
],
]),
}),
),
Effect.catchTags({
NotValidMessageError: () => Effect.unit(),
DiscordRESTError: _ =>
"response" in _.error
? Effect.flatMap(_.error.response.json, _ =>
Effect.logInfo(JSON.stringify(_, null, 2)),
)
: log.info(_.error),
}),
Effect.catchAllCause(Effect.logErrorCause),
),
Effect.catchTags({
NotValidMessageError: () => Effect.unit(),
DiscordRESTError: _ =>
"response" in _.error
? Effect.flatMap(_.error.response.json, _ =>
Effect.logInfo(JSON.stringify(_, null, 2)),
)
: log.info(_.error),
}),
Effect.catchAllCause(Effect.logErrorCause),
),
)
)

const hasManage = Perms.has(Discord.PermissionFlag.MANAGE_CHANNELS)
const hasManage = Perms.has(Discord.PermissionFlag.MANAGE_CHANNELS)

const withEditPermissions = <R, E, A>(self: Effect.Effect<R, E, A>) =>
Effect.gen(function* ($) {
const ix = yield* $(Ix.Interaction)
const ctx = yield* $(Ix.MessageComponentData)
const authorId = ctx.custom_id.split("_")[1]
const canEdit =
authorId === ix.member?.user?.id || hasManage(ix.member!.permissions!)
const withEditPermissions = <R, E, A>(self: Effect.Effect<R, E, A>) =>
Effect.gen(function* ($) {
const ix = yield* $(Ix.Interaction)
const ctx = yield* $(Ix.MessageComponentData)
const authorId = ctx.custom_id.split("_")[1]
const canEdit =
authorId === ix.member?.user?.id || hasManage(ix.member!.permissions!)

if (!canEdit) {
return Ix.response({
type: Discord.InteractionCallbackType.CHANNEL_MESSAGE_WITH_SOURCE,
data: {
flags: Discord.MessageFlag.EPHEMERAL,
content: "You don't have permissions to edit this thread",
},
})
}
if (!canEdit) {
return Ix.response({
type: Discord.InteractionCallbackType.CHANNEL_MESSAGE_WITH_SOURCE,
data: {
flags: Discord.MessageFlag.EPHEMERAL,
content: "You don't have permissions to edit this thread",
},
})
}

return yield* $(self)
})
return yield* $(self)
})

const edit = Ix.messageComponent(
Ix.idStartsWith("edit_"),
pipe(
Ix.Interaction,
Effect.flatMap(ix => channels.get(ix.guild_id!, ix.channel_id!)),
Effect.map(channel =>
Ix.response({
type: Discord.InteractionCallbackType.MODAL,
data: {
custom_id: "edit",
title: "Edit title",
components: UI.singleColumn([
UI.textInput({
custom_id: "title",
label: "New title",
max_length: 100,
value: channel.name!,
}),
]),
},
}),
),
withEditPermissions,
),
)

const editModal = Ix.modalSubmit(
Ix.id("edit"),
pipe(
Effect.allPar({
title: Ix.modalValue("title"),
context: Ix.Interaction,
}),
Effect.tap(({ title, context }) =>
pipe(
rest.modifyChannel(context.channel_id!, { name: title }),
Effect.flatMap(_ => _.json),
Effect.tap(channels.put),
const edit = Ix.messageComponent(
Ix.idStartsWith("edit_"),
pipe(
Ix.Interaction,
Effect.flatMap(ix => channels.get(ix.guild_id!, ix.channel_id!)),
Effect.map(channel =>
Ix.response({
type: Discord.InteractionCallbackType.MODAL,
data: {
custom_id: "edit",
title: "Edit title",
components: UI.singleColumn([
UI.textInput({
custom_id: "title",
label: "New title",
max_length: 100,
value: channel.name!,
}),
]),
},
}),
),
withEditPermissions,
),
Effect.as(
Ix.response({
type: Discord.InteractionCallbackType.DEFERRED_UPDATE_MESSAGE,
)

const editModal = Ix.modalSubmit(
Ix.id("edit"),
pipe(
Effect.allPar({
title: Ix.modalValue("title"),
context: Ix.Interaction,
}),
Effect.tap(({ title, context }) =>
pipe(
rest.modifyChannel(context.channel_id!, { name: title }),
Effect.flatMap(_ => _.json),
Effect.tap(channels.put),
),
),
Effect.as(
Ix.response({
type: Discord.InteractionCallbackType.DEFERRED_UPDATE_MESSAGE,
}),
),
),
),
)
)

const archive = Ix.messageComponent(
Ix.idStartsWith("archive_"),
pipe(
Ix.Interaction,
Effect.tap(ix => rest.modifyChannel(ix.channel_id!, { archived: true })),
Effect.as(
Ix.response({
type: Discord.InteractionCallbackType.DEFERRED_UPDATE_MESSAGE,
}),
const archive = Ix.messageComponent(
Ix.idStartsWith("archive_"),
pipe(
Ix.Interaction,
Effect.tap(ix =>
rest.modifyChannel(ix.channel_id!, { archived: true }),
),
Effect.as(
Ix.response({
type: Discord.InteractionCallbackType.DEFERRED_UPDATE_MESSAGE,
}),
),
withEditPermissions,
),
withEditPermissions,
),
)
)

const runInteractions = pipe(
Ix.builder.add(archive).add(edit).add(editModal),
runIx(Effect.catchAllCause(Effect.logErrorCause), { sync: false }),
)
const runInteractions = pipe(
Ix.builder.add(archive).add(edit).add(editModal),
runIx(Effect.catchAllCause(Effect.logErrorCause), { sync: false }),
)

yield* $(Effect.allPar(runInteractions, handleMessages))
})
yield* $(Effect.allPar(runInteractions, handleMessages))
})

export const AutoThreadsLive = Layer.provide(
ChannelsCacheLive,
Layer.effectDiscard(make),
)
export const makeLayer = (config: Config.Config.Wrap<AutoThreadsOptions>) =>
Layer.provide(
ChannelsCacheLive,
Layer.effectDiscard(
Effect.flatMap(Effect.config(Config.unwrap(config)), make),
),
)
Loading

0 comments on commit b14bc72

Please sign in to comment.