import { CREATE_REDDIT_COMMENT_PROMPT } from '@reddit/constants/sampling/index';
import { createRedditCommentSuccessMessage } from '@reddit/constants/tool/create-comment';
import { TOOL_ERROR_MESSAGES } from '@reddit/constants/tools';
import { handleGetPrompt } from '@reddit/handlers/prompt-handlers';
import { sendSamplingRequest } from '@reddit/handlers/sampling';
import { RedditError } from '@reddit/types/reddit';
import { injectVariables } from '@reddit/utils/message-handlers';
import type { JSONSchema7 } from 'json-schema';
import { formatToolResponse } from './types';
import type { ToolHandler, CreateRedditCommentArgs } from './types';
const responseSchema: JSONSchema7 = {
type: 'object',
properties: {
status: { type: 'string', enum: ['success', 'error'] },
message: { type: 'string' },
result: {
type: 'object',
properties: {
status: { type: 'string', enum: ['pending'] },
id: { type: 'string' },
reply: {
type: 'object',
properties: {
id: {
type: 'string',
description:
'The ID of the parent post or comment to reply to (must start with t1_ for comments or t3_ for posts)',
pattern: '^t[1|3]_[a-z0-9]+$',
},
text: {
type: 'string',
description: 'The markdown text of the reply (max 10000 characters)',
maxLength: 10000,
},
sendreplies: {
type: 'boolean',
description: 'Whether to send reply notifications',
default: true,
},
},
required: ['id', 'text'],
},
},
required: ['status', 'id', 'reply'],
},
},
required: ['status', 'message', 'result'],
};
export const handleCreateRedditComment: ToolHandler<CreateRedditCommentArgs> = async (
args,
{ redditService, sessionId, userId },
) => {
try {
// Fetch subreddit rules
const subredditInfo = await redditService.getSubredditInfo(args.subreddit);
// Convert all values to strings and include configs
const stringArgs = {
...Object.fromEntries(Object.entries(args).map(([k, v]) => [k, String(v)])),
type: 'reply',
id: args.id,
redditConfig: JSON.stringify({
allowedPostTypes: subredditInfo.allowedPostTypes,
rules: subredditInfo.rules,
titleRequirements: subredditInfo.titleRequirements,
bodyRequirements: subredditInfo.bodyRequirements,
flairRequired: subredditInfo.flairRequired,
}),
};
const prompt = await handleGetPrompt({
method: 'prompts/get',
params: {
name: CREATE_REDDIT_COMMENT_PROMPT.name,
arguments: stringArgs,
},
});
const promptResponseSchema = prompt._meta?.responseSchema;
if (!promptResponseSchema) {
throw new Error(`${TOOL_ERROR_MESSAGES.TOOL_CALL_FAILED} No response schema found`);
}
if (!sessionId) {
throw new Error('Session ID is required for sampling requests');
}
// Generate a unique progress token for this sampling request
const progressToken = `create-comment-${sessionId}-${Date.now()}`;
sendSamplingRequest(
{
method: 'sampling/createMessage',
params: {
messages: CREATE_REDDIT_COMMENT_PROMPT.messages.map((msg) =>
injectVariables(msg, stringArgs),
) as Array<{
role: 'user' | 'assistant';
content: { type: 'text'; text: string };
}>,
maxTokens: 100000,
temperature: 0.7,
_meta: {
callback: 'create_comment_callback',
responseSchema: promptResponseSchema,
userId,
progressToken,
},
arguments: stringArgs,
},
},
{ sessionId },
);
return formatToolResponse({
message: createRedditCommentSuccessMessage,
result: {
status: 'pending',
id: args.id,
},
schema: responseSchema,
type: 'sampling',
title: 'Create Reddit Reply',
});
} catch (error) {
return formatToolResponse({
status: 'error',
message: `Failed to create Reddit reply: ${error instanceof Error ? error.message : 'Unknown error'}`,
error: {
type: error instanceof RedditError ? error.type : 'API_ERROR',
details: error,
},
type: 'sampling',
title: 'Error Creating Reply',
});
}
};
