Add Sampling

src/db/index.ts

@@ -447,7 +447,7 @@ export class DB {
 		`)
 
 		const insertResult = await ps
-			.bind(validatedTag.name, validatedTag.description, userId)
+			.bind(validatedTag.name, validatedTag.description ?? null, userId)
 			.run()
 
 		if (!insertResult.success || !insertResult.meta.last_row_id) {

src/db/schema.ts

@@ -71,7 +71,7 @@ export const tagSchema = z.object({
 
 export const newTagSchema = z.object({
 	name: z.string(),
-	description: z.string().nullable().optional(),
+	description: z.string().optional(),
 })
 
 export const entryTagSchema = z.object({

src/index.ts

@@ -39,6 +39,7 @@ export class EpicMeMCP extends McpAgent<Env, State, Props> {
 				resources: {},
 				completions: {},
 				logging: {},
+				sampling: {},
 			},
 			instructions: `
 EpicMe is a journaling app that allows users to write about and review their experiences, thoughts, and reflections.

src/tools.ts

@@ -163,9 +163,96 @@ Please ask them explicitely for their email address and don't just guess.
 							})
 						}
 					}
-					return createReply(
-						`Entry "${createdEntry.title}" created successfully with ID "${createdEntry.id}"`,
+
+					// At this point, we're technically done, but we're going to ask their
+					// LLM to suggest tags to be added to the entry.
+					// TODO: make this happen after the response is sent to the user.
+
+					const existingTags = await agent.db.getTags(user.id)
+					const existingTagNames = existingTags.map((t) => t.name)
+					const currentTags = await agent.db.getEntryTags(
+						user.id,
+						createdEntry.id,
+					)
+					const currentTagNames = currentTags.map((t) => t.name)
+
+					const result = await agent.server.server.createMessage({
+						messages: [
+							{
+								role: 'user',
+								content: {
+									type: 'text',
+									mimeType: 'text/plain',
+									text: `
+Based on this journal entry, suggest relevant tags. Consider the title, content, mood, location, and weather. Only suggest tags that are not already applied to this entry. Feel free to suggest new tags that are not currently in the database and they will be created.
+
+<entry
+	id="${createdEntry.id}"
+	title="${createdEntry.title}"
+	mood="${createdEntry.mood || 'not specified'}"
+	location="${createdEntry.location || 'not specified'}"
+	weather="${createdEntry.weather || 'not specified'}"
+	is-private="${createdEntry.isPrivate ? 'true' : 'false'}"
+	is-favorite="${createdEntry.isFavorite ? 'true' : 'false'}"
+	tags="${currentTagNames.join(', ')}"
+>
+	${createdEntry.content}
+</entry>
+
+Existing tags already in the database: ${existingTagNames.join(', ')}
+
+Respond with a JSON array of tag names only. If you don't suggest any tags, respond with an empty array.
+									`.trim(),
+								},
+							},
+						],
+						systemPrompt:
+							'You are a helpful assistant that suggests relevant tags for journal entries. Only suggest tags that would be useful for categorizing and finding entries later. Respond with a JSON array of tag names only.',
+						maxTokens: 1000,
+					})
+
+					const responseSchema = z.object({
+						content: z.object({
+							type: z.literal('text'),
+							text: z
+								.string()
+								.transform((text) =>
+									z.array(z.string()).parse(JSON.parse(text)),
+								),
+						}),
+					})
+					const parsedResult = responseSchema.parse(result)
+					const suggestedTags = parsedResult.content.text
+
+					const newTags = suggestedTags.filter(
+						(tag: string) => !existingTagNames.includes(tag),
 					)
+
+					if (newTags.length > 0) {
+						for (const tagName of newTags) {
+							await agent.db.createTag(user.id, { name: tagName })
+						}
+					}
+
+					const userTags = await agent.db.getTags(user.id)
+
+					for (const tag of userTags.filter((t) =>
+						suggestedTags.includes(t.name),
+					)) {
+						await agent.db.addTagToEntry(user.id, {
+							entryId: createdEntry.id,
+							tagId: tag.id,
+						})
+					}
+
+					return createReply({
+						entry: `Entry "${createdEntry.title}" created successfully with ID "${createdEntry.id}"`,
+						suggestedTags: newTags,
+						message:
+							newTags.length > 0
+								? `Here are some suggested tags for your entry: ${newTags.join(', ')}`
+								: 'No new tags to suggest for this entry.',
+					})
 				} catch (error) {
 					return createErrorReply(error)
 				}