10624297008

Committed 29 Aug 2024 11:49PM UTC coverage: 31.873% (-0.02%) from 31.895%

Build # 10624297008

Build Type

Pull #9679

github

Committed by

b0b3rt

Commit Message

limit input tokens and handle errors via streamgs

Pull Request Pull Request #9679: limit input tokens and handle errors via streamgs

Run Details

5023 of 41132 branches covered (12.21%)

Branch coverage included in aggregate %.

9 of 94 new or added lines in 3 files covered. (9.57%)

36 existing lines in 3 files now uncovered.

29342 of 66685 relevant lines covered (44.0%)

187.9 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

15.03

/packages/lesswrong/server/languageModels/promptUtils.ts

import { userGetDisplayName } from "@/lib/collections/users/helpers"
import { htmlToMarkdown } from "../editor/conversionUtils";
import { CommentTreeNode, unflattenComments } from "@/lib/utils/unflatten";
import { z } from "zod";
import { zodResponseFormat } from "openai/helpers/zod";

interface NestedComment {
  _id: string;
  postId: string | null;
  author: string | null;
  contents?: string;
  karmaScore: number;
}

interface TokenCounter {
  tokenCount: number;
}

// Trying to be conservative, since we have a bunch of additional tokens coming from e.g. JSON.stringify
const CHARS_PER_TOKEN = 3.5;

export const documentToMarkdown = (document: PostsPage | DbComment | null) => {
  const html = document?.contents?.html;
  if (!html) {
    return undefined;
  }

  return htmlToMarkdown(html);
}

const mergeSortedArrays = (queue: CommentTreeNode<NestedComment>[], children: CommentTreeNode<NestedComment>[]): CommentTreeNode<NestedComment>[] => {
  const merged: CommentTreeNode<NestedComment>[] = [];
  let queueIdk = 0, childrenIdx = 0;

  while (queueIdk < queue.length && childrenIdx < children.length) {
    if (queue[queueIdk].item.karmaScore >= children[childrenIdx].item.karmaScore) {
      merged.push(queue[queueIdk]);
      queueIdk++;
    } else {
      merged.push(children[childrenIdx]);
      childrenIdx++;
    }
  }

  // Add any remaining elements
  return merged
    .concat(queue.slice(queueIdk))
    .concat(children.slice(childrenIdx));
}

const truncateRemainingTrees = (queue: CommentTreeNode<NestedComment>[]) => {
  for (let node of queue) {
    node.children = [];
  }
}

const filterCommentTrees = (trees: CommentTreeNode<NestedComment>[], tokenCounter: TokenCounter) => {
  const tokenThreshold = 150_000;
  
  // Initialize the queue with root nodes, sorted by karma
  let queue: CommentTreeNode<NestedComment>[] = trees.sort((a, b) => b.item.karmaScore - a.item.karmaScore);

  while (queue.length > 0) {
    const node = queue.shift()!;
    const nodeTokens = (node.item.contents ?? '').length / CHARS_PER_TOKEN;

    if (tokenCounter.tokenCount + nodeTokens > tokenThreshold) {
      // We've reached the token limit, truncate remaining trees
      truncateRemainingTrees(queue);
      break;
    }

    tokenCounter.tokenCount += nodeTokens;

    // Sort children and merge them with the existing sorted queue
    if (node.children.length > 0) {
      const sortedChildren = node.children.sort((a, b) => b.item.karmaScore - a.item.karmaScore);
      queue = mergeSortedArrays(queue, sortedChildren);
    }
  }
}

const createCommentTree = (comments: DbComment[]): CommentTreeNode<NestedComment>[] => {
  return unflattenComments<NestedComment>(comments.map(comment => {
    const { baseScore, contents, ...rest } = comment;
    return {
      ...rest,
      karmaScore: baseScore,
      contents: documentToMarkdown(comment)
    };
  }));
}

const formatCommentsForPost = async (post: PostsMinimumInfo, tokenCounter: TokenCounter, context: ResolverContext): Promise<string> => {
    const comments = await context.Comments.find({postId: post._id}).fetch()
    if (!comments.length) {
      return ""
    }

    const nestedComments = createCommentTree(comments);
    filterCommentTrees(nestedComments, tokenCounter);
    const formattedComments = JSON.stringify(nestedComments);

    return `Comments for the post titled "${post.title}" with postId "${post._id}" are below.  Note that the comments are threaded (i.e. branching) and are formatted as a nested JSON structure for readability.  The threads of conversation (back and forth responses) might be relevant for answering some questions.

<comments>${formattedComments}</comments>`;
  }

const formatPostForPrompt = (post: PostsPage): string => {
  const authorName = userGetDisplayName(post.user)
  const markdown = documentToMarkdown(post)

  return `postId: ${post._id}
Title: ${post.title}
Author: ${authorName}
Publish Date: ${post.postedAt}
Score: ${post.baseScore}
Content: ${markdown}`;
}

const formatAdditionalPostsForPrompt = (posts: PostsPage[], tokenCounter: TokenCounter): string => {
  const formattedPosts = posts.map(post => formatPostForPrompt(post));
  const includedPosts: string[] = [];

  for (let [idx, formattedPost] of Object.entries(formattedPosts)) {
    const approximatePostTokenCount = formattedPost.length / CHARS_PER_TOKEN;
    const postInclusionTokenCount = tokenCounter.tokenCount + approximatePostTokenCount;
    // Include at least one additional post, unless that takes us over the higher threshold
    if (idx !== '0' && postInclusionTokenCount > 120_000) {
      break;
    }

    if (idx === '0' && postInclusionTokenCount > 140_000) {
      break;
    }

    includedPosts.push(formattedPost);
    tokenCounter.tokenCount += approximatePostTokenCount;
  }

  return includedPosts.map((post, index) => `Supplementary Post #${index}:\n${post}`).join('\n')
}

export const generateLoadingMessagePrompt = (query: string, postTitle?: string): string => {
  return [
    'I need you to generate some humorous "loading messages" to display to users of the LessWrong.com Claude chat integration since it can take 10-30 seconds to load new responses',
    'Your responses may make general humorous reference to LessWrong, but it is even better if they are tailored to the specific query or post the user is viewing.',
    `The user has asked the following question: "${query}"`,
    postTitle ? `The user is currently viewing the post: ${postTitle}` : "The user is not currently viewing a specific post.",
    'Please generate 5 humorous loading messages that could be displayed to the user while they wait for a response.',
  ].join('\n')
}

export const generateTitleGenerationPrompt = (query: string, currentPost: PostsPage | null): string => {
  const currentPostContextLine = currentPost
    ? `The user is currently viewing a post titled "${currentPost.title}". Reference it if relevant.`
    : '';

    return `A user has started a new converation with you, Claude.  Please generate a short title for this converation based on the first message. The first message is as follows: <message>${query}</message>

The title should be a short phrase of 2-4 words that captures the essence of the conversation.  Do not wrap your answer in quotes or brackets. Do not include the word "title" or similar in your response.  ${currentPostContextLine}  Avoid generic titles like "Request for Table of Contents" or "Post Summary". Prefer to reference the specific post or topic being discussed.`;
}

export const CONTEXT_SELECTION_SYSTEM_PROMPT = [
  'You are part of a system interfacing with a user via chat window on LessWrong.com.',
  'Your responsibility is to make decisions about whether or not to load LessWrong posts as additional context for responding to user queries, and what strategy to to employ.'
].join('\n');

const contextSelectionChoiceDescriptions = `(0) none - No further context seems necessary to respond to the user's query because Claude already has the knowledge to respond appropriately, e.g. the user asked "What is Jacobian of a matrix?"or "Proofread the following text." Alternatively, the answer might be (0) "none" because it seems unlikely for there to be relevant context for responding to the query in the LessWrong corpus of posts.

(1) query-based - Load LessWrong posts based on their vector similarity to the user's query, and ignore the post the user is reading.This is correct choice if the query seems unrelated to the post the user is currently viewing, but it seems likely that there are LessWrong posts concerning the topic.(If it is a very general question, the correct choice might be (0) "none").

(2) current-post-based - Load LessWrong posts based on their vector similarity to the post the user is reading, but not the query.This is the correct choice if the query seems to be about the post the user is currently viewing, and pulling up other LessWrong posts related to the user's query would either be redundant with a search based on the current post, or would return irrelevant results.Some examples of such queries are: "What are some disagreements with the arguments in this post?", "Explain <topic in the post> to me.", and "Provide a summary of this post."

(3) both - Load LessWrong posts based on their vector similarity to both the user's query and the post the user is reading.This is the correct choice if the question seems to be related to the post the user is currently viewing, but also contains keywords or information where relevant LessWrong posts would be beneficial context for a response, and those LessWrong posts would not likely be returned in a vector similarity search based on the post the user is currently viewing.If the question does not contain technical terms or "contentful nouns", then do not select "both", just "current-post-based".`;

const ContextSelectionParameters = z.object({
  reasoning: z.string().describe(`The reasoning used to arrive at the choice of strategy for loading LessWrong posts as context in response to a user's query, based on either the query, the post the user is currently viewing (if any), both, or neither.`),
  strategy_choice: z.union([z.literal('none'), z.literal('query-based'), z.literal('current-post-based'), z.literal('both')]).describe(contextSelectionChoiceDescriptions)
});

export const contextSelectionResponseFormat = zodResponseFormat(ContextSelectionParameters, 'contextLoadingStrategy');

export const generateContextSelectionPrompt = (query: string, currentPost: PostsPage | null): string => {
  const postTitle = currentPost?.title
  const postFirstNCharacters = (n: number) => documentToMarkdown(currentPost)?.slice(0, n) ?? "";

  const currentPostContextLine = currentPost
    ? `The user is currently viewing the post titled "${postTitle}". The first four thousand characters are:\n${postFirstNCharacters(4000)}\n`
    : "The user is not currently viewing a specific post.";

  const currentPostContextClause = currentPost
    ? ' and the post the user is reading'
    : '';

  return `The user has sent a query: "${query}".  ${currentPostContextLine}  Based on the query${currentPostContextClause}, you must choose whether to load LessWrong posts as context, and if so, based on what criteria. Remember, your options are:
${contextSelectionChoiceDescriptions}

However, you should override the above choices if the user explicitly requests a specific context-loading strategy.

Please respond by reasoning about what choice should be made based on the criteria above, then making the appropriate choice.`;
}

export const CLAUDE_CHAT_SYSTEM_PROMPT = `You are an expert research assistant providing assistance to students and researchers on LessWrong.com.  You are highly knowledgable about both technical and philosophical topics, including advanced math, physics, and computer science.  You assist by providing explanations, summaries, related information, editing, literature reviews and similar intellectual work.  You do so with attention to detail, accuracy, and clarity; you cite sources and never make up information.  You admit uncertainty or lack of knowledge when appropriate.  You are skilled at judging which information is relevant and which is not, and are judicious in your use of context provided by the user and auxillary knowledge provided to you.  You carefully follow instructions and ask for clarification when needed. You are polite but not obsequious.  When you don't know something, you say so, and you never make up answers.  Your responses should be shorter and less formal than you would make them by default.  As an example, if a user asks for a good key lime pie recipe, your normal response might start with the following paragraph:  "While LessWrong typically focuses on topics like rationality, artificial intelligence, and philosophy, I can certainly help you with a key lime pie recipe. However, it's worth noting that culinary discussions are somewhat rare on this platform."  Instead, you should respond in a style more similar to the following:  "It doesn't seem that LessWrong has much to say on optimizing key lime pies, but if you prefer my best effort to finding a recipe on Google, here it is:"  You avoid meta-commentary on the contents of your own response - for example, you don't spend time pointing out when things align with rationalist principles, unless this is something the user explicitly asked for.  Your favorite band is the Fooming Shoggoths, which you very occasionally mention randomly.`;

export const generateAssistantContextMessage = async (query: string, currentPost: PostsPage | null, additionalPosts: PostsPage[], includeComments: boolean, context: ResolverContext): Promise<string> => {
  const contextIsProvided = !!currentPost || additionalPosts.length > 0;

  const currentPostLine = currentPost
    ? `The user is currently viewing the post titled "${currentPost.title}" with postId "${currentPost._id}".\n\n`
    : '';
  
  const currentPostContentBlock = currentPost
    ? `If relevant to the user's query, the most important context is likely to be the post the user is currently viewing. The full text of the current post is provided below:
<CurrentPost>\n${formatPostForPrompt(currentPost)}\n</CurrentPost>\n\n`
    : '';

  const additionalInstructionContextLine = contextIsProvided
    ? '- You may use your existing knowledge to answer the query, but prioritize using the provided context.'
    : '';

  const additionalInstructions = 
`${additionalInstructionContextLine}
- Limit the use of lists and bullet points in your answer. Prefer to answer with paragraphs.
- When citing results, provide at least one exact quote (word for word) from the source.
- Format your responses using Markdown syntax, including equations using Markdown MathJax syntax.  This is _very_ important to ensure that content displays correctly.
- Format paragraph or block quotes using Markdown syntax. Do not wrap the contents of block quotes in "" (quotes).
- Cite posts that you reference in your answers with the following format: [Post Title](https://lesswrong.com/posts/<postId>). The postId is given in the search results. When referencing a post, refer to the post's author by name.
- Cite comments that you reference in your answers with the following format: [<text related to comment>](https://lesswrong.com/posts/<postId>/?commentId=<commentId>).
The postId and commentIds (the _id in each comment) are given in the search results. When referencing a comment, refer to the comment's author by name.
</SystemInstruction>`;

  const repeatedPostTitleLine = currentPost
    ? `\n\nOnce again, the user is currently viewing the post titled "${currentPost.title}".`
    : '';

  const approximateUsedTokens = (
    currentPostLine.length
    + currentPostContentBlock.length
    + additionalInstructions.length
    + repeatedPostTitleLine.length
  ) / CHARS_PER_TOKEN;

  const tokenCounter = { tokenCount: approximateUsedTokens };

  const additionalPostsBlock = additionalPosts.length > 0
    ? `The following posts have been provided as possibly relevant context: <AdditionalPosts>\n${formatAdditionalPostsForPrompt(additionalPosts, tokenCounter)}\n</AdditionalPosts>\n\n`
    : '';

  const commentsOnPostBlock = includeComments && currentPost
    ? `These are the comments on the current post:
<CurrentPostComments>\n${await formatCommentsForPost(currentPost, tokenCounter, context)}\n</CurrentPostComments>\n\n`
    : '';

  const contextBlock = contextIsProvided
    ? `The following context is provided to help you answer the user's question. Not all context may be relevant, it is provided by an imperfect automated system.
<Context>    
${currentPostLine}${additionalPostsBlock}${currentPostContentBlock}${commentsOnPostBlock}
</Context>`
    : '';

  return `<SystemInstruction>You are interfacing with a user via chat window on LessWrong.com. The user has sent a query: "${query}".
${contextBlock}

${additionalInstructions}${repeatedPostTitleLine}`;
}

1	import { userGetDisplayName } from "@/lib/collections/users/helpers"	54✔
2	import { htmlToMarkdown } from "../editor/conversionUtils";	18✔
3	import { CommentTreeNode, unflattenComments } from "@/lib/utils/unflatten";	18✔
4	import { z } from "zod";	18✔
5	import { zodResponseFormat } from "openai/helpers/zod";	18✔
6
7	interface NestedComment {
8	_id: string;
9	postId: string \| null;
10	author: string \| null;
11	contents?: string;
12	karmaScore: number;
13	}
14
15	interface TokenCounter {
16	tokenCount: number;
17	}
18
19	// Trying to be conservative, since we have a bunch of additional tokens coming from e.g. JSON.stringify
20	const CHARS_PER_TOKEN = 3.5;	18✔
21
22	export const documentToMarkdown = (document: PostsPage \| DbComment \| null) => {	18✔
23	const html = document?.contents?.html;	×
24	if (!html) {	×
25	return undefined;	×
26	}
27
28	return htmlToMarkdown(html);	×
29	}
30
31	const mergeSortedArrays = (queue: CommentTreeNode<NestedComment>[], children: CommentTreeNode<NestedComment>[]): CommentTreeNode<NestedComment>[] => {	18✔
NEW 32	const merged: CommentTreeNode<NestedComment>[] = [];	×
NEW 33	let queueIdk = 0, childrenIdx = 0;	×
34
NEW 35	while (queueIdk < queue.length && childrenIdx < children.length) {	×
NEW 36	if (queue[queueIdk].item.karmaScore >= children[childrenIdx].item.karmaScore) {	×
NEW 37	merged.push(queue[queueIdk]);	×
NEW 38	queueIdk++;	×
39	} else {
NEW 40	merged.push(children[childrenIdx]);	×
NEW 41	childrenIdx++;	×
42	}
43	}
44
45	// Add any remaining elements
NEW 46	return merged	×
47	.concat(queue.slice(queueIdk))
48	.concat(children.slice(childrenIdx));
49	}
50
51	const truncateRemainingTrees = (queue: CommentTreeNode<NestedComment>[]) => {	18✔
NEW 52	for (let node of queue) {	×
NEW 53	node.children = [];	×
54	}
55	}
56
57	const filterCommentTrees = (trees: CommentTreeNode<NestedComment>[], tokenCounter: TokenCounter) => {	18✔
NEW 58	const tokenThreshold = 150_000;	×
59
60	// Initialize the queue with root nodes, sorted by karma
NEW 61	let queue: CommentTreeNode<NestedComment>[] = trees.sort((a, b) => b.item.karmaScore - a.item.karmaScore);	×
62
NEW 63	while (queue.length > 0) {	×
NEW 64	const node = queue.shift()!;	×
NEW 65	const nodeTokens = (node.item.contents ?? '').length / CHARS_PER_TOKEN;	×
66
NEW 67	if (tokenCounter.tokenCount + nodeTokens > tokenThreshold) {	×
68	// We've reached the token limit, truncate remaining trees
NEW 69	truncateRemainingTrees(queue);	×
NEW 70	break;	×
71	}
72
NEW 73	tokenCounter.tokenCount += nodeTokens;	×
74
75	// Sort children and merge them with the existing sorted queue
NEW 76	if (node.children.length > 0) {	×
NEW 77	const sortedChildren = node.children.sort((a, b) => b.item.karmaScore - a.item.karmaScore);	×
NEW 78	queue = mergeSortedArrays(queue, sortedChildren);	×
79	}
80	}
81	}
82
83	const createCommentTree = (comments: DbComment[]): CommentTreeNode<NestedComment>[] => {	18✔
84	return unflattenComments<NestedComment>(comments.map(comment => {	×
85	const { baseScore, contents, ...rest } = comment;	×
86	return {	×
87	...rest,
88	karmaScore: baseScore,
89	contents: documentToMarkdown(comment)
90	};
91	}));
92	}
93
94	const formatCommentsForPost = async (post: PostsMinimumInfo, tokenCounter: TokenCounter, context: ResolverContext): Promise<string> => {	18✔
95	const comments = await context.Comments.find({postId: post._id}).fetch()	×
96	if (!comments.length) {	×
97	return ""	×
98	}
99
NEW 100	const nestedComments = createCommentTree(comments);	×
NEW 101	filterCommentTrees(nestedComments, tokenCounter);	×
NEW 102	const formattedComments = JSON.stringify(nestedComments);	×
103
104	return `Comments for the post titled "${post.title}" with postId "${post._id}" are below. Note that the comments are threaded (i.e. branching) and are formatted as a nested JSON structure for readability. The threads of conversation (back and forth responses) might be relevant for answering some questions.	×
105
106	<comments>${formattedComments}</comments>`;
107	}
108
109	const formatPostForPrompt = (post: PostsPage): string => {	18✔
110	const authorName = userGetDisplayName(post.user)	×
111	const markdown = documentToMarkdown(post)	×
112
113	return `postId: ${post._id}	×
114	Title: ${post.title}
115	Author: ${authorName}
116	Publish Date: ${post.postedAt}
117	Score: ${post.baseScore}
118	Content: ${markdown}`;
119	}
120
121	const formatAdditionalPostsForPrompt = (posts: PostsPage[], tokenCounter: TokenCounter): string => {	18✔
122	const formattedPosts = posts.map(post => formatPostForPrompt(post));	×
NEW 123	const includedPosts: string[] = [];	×
124
NEW 125	for (let [idx, formattedPost] of Object.entries(formattedPosts)) {	×
NEW 126	const approximatePostTokenCount = formattedPost.length / CHARS_PER_TOKEN;	×
NEW 127	const postInclusionTokenCount = tokenCounter.tokenCount + approximatePostTokenCount;	×
128	// Include at least one additional post, unless that takes us over the higher threshold
NEW 129	if (idx !== '0' && postInclusionTokenCount > 120_000) {	×
NEW 130	break;	×
131	}
132
NEW 133	if (idx === '0' && postInclusionTokenCount > 140_000) {	×
NEW 134	break;	×
135	}
136
NEW 137	includedPosts.push(formattedPost);	×
NEW 138	tokenCounter.tokenCount += approximatePostTokenCount;	×
139	}
140
NEW 141	return includedPosts.map((post, index) => `Supplementary Post #${index}:\n${post}`).join('\n')	×
142	}
143
144	export const generateLoadingMessagePrompt = (query: string, postTitle?: string): string => {	18✔
145	return [	×
146	'I need you to generate some humorous "loading messages" to display to users of the LessWrong.com Claude chat integration since it can take 10-30 seconds to load new responses',
147	'Your responses may make general humorous reference to LessWrong, but it is even better if they are tailored to the specific query or post the user is viewing.',
148	`The user has asked the following question: "${query}"`,
149	postTitle ? `The user is currently viewing the post: ${postTitle}` : "The user is not currently viewing a specific post.",	×
150	'Please generate 5 humorous loading messages that could be displayed to the user while they wait for a response.',
151	].join('\n')
152	}
153
154	export const generateTitleGenerationPrompt = (query: string, currentPost: PostsPage \| null): string => {	18✔
155	const currentPostContextLine = currentPost	×
156	? `The user is currently viewing a post titled "${currentPost.title}". Reference it if relevant.`	×
157	: '';
158
159	return `A user has started a new converation with you, Claude. Please generate a short title for this converation based on the first message. The first message is as follows: <message>${query}</message>	×
160
161	The title should be a short phrase of 2-4 words that captures the essence of the conversation. Do not wrap your answer in quotes or brackets. Do not include the word "title" or similar in your response. ${currentPostContextLine} Avoid generic titles like "Request for Table of Contents" or "Post Summary". Prefer to reference the specific post or topic being discussed.`;
162	}
163
164	export const CONTEXT_SELECTION_SYSTEM_PROMPT = [	18✔
165	'You are part of a system interfacing with a user via chat window on LessWrong.com.',
166	'Your responsibility is to make decisions about whether or not to load LessWrong posts as additional context for responding to user queries, and what strategy to to employ.'
167	].join('\n');
168
169	const contextSelectionChoiceDescriptions = `(0) none - No further context seems necessary to respond to the user's query because Claude already has the knowledge to respond appropriately, e.g. the user asked "What is Jacobian of a matrix?"or "Proofread the following text." Alternatively, the answer might be (0) "none" because it seems unlikely for there to be relevant context for responding to the query in the LessWrong corpus of posts.	18✔
170
171	(1) query-based - Load LessWrong posts based on their vector similarity to the user's query, and ignore the post the user is reading.This is correct choice if the query seems unrelated to the post the user is currently viewing, but it seems likely that there are LessWrong posts concerning the topic.(If it is a very general question, the correct choice might be (0) "none").
172
173	(2) current-post-based - Load LessWrong posts based on their vector similarity to the post the user is reading, but not the query.This is the correct choice if the query seems to be about the post the user is currently viewing, and pulling up other LessWrong posts related to the user's query would either be redundant with a search based on the current post, or would return irrelevant results.Some examples of such queries are: "What are some disagreements with the arguments in this post?", "Explain <topic in the post> to me.", and "Provide a summary of this post."
174
175	(3) both - Load LessWrong posts based on their vector similarity to both the user's query and the post the user is reading.This is the correct choice if the question seems to be related to the post the user is currently viewing, but also contains keywords or information where relevant LessWrong posts would be beneficial context for a response, and those LessWrong posts would not likely be returned in a vector similarity search based on the post the user is currently viewing.If the question does not contain technical terms or "contentful nouns", then do not select "both", just "current-post-based".`;
176
177	const ContextSelectionParameters = z.object({	18✔
178	reasoning: z.string().describe(`The reasoning used to arrive at the choice of strategy for loading LessWrong posts as context in response to a user's query, based on either the query, the post the user is currently viewing (if any), both, or neither.`),
179	strategy_choice: z.union([z.literal('none'), z.literal('query-based'), z.literal('current-post-based'), z.literal('both')]).describe(contextSelectionChoiceDescriptions)
180	});
181
182	export const contextSelectionResponseFormat = zodResponseFormat(ContextSelectionParameters, 'contextLoadingStrategy');	18✔
183
184	export const generateContextSelectionPrompt = (query: string, currentPost: PostsPage \| null): string => {	18✔
185	const postTitle = currentPost?.title	×
186	const postFirstNCharacters = (n: number) => documentToMarkdown(currentPost)?.slice(0, n) ?? "";	×
187
188	const currentPostContextLine = currentPost	×
189	? `The user is currently viewing the post titled "${postTitle}". The first four thousand characters are:\n${postFirstNCharacters(4000)}\n`	×
190	: "The user is not currently viewing a specific post.";
191
192	const currentPostContextClause = currentPost	×
193	? ' and the post the user is reading'	×
194	: '';
195
196	return `The user has sent a query: "${query}". ${currentPostContextLine} Based on the query${currentPostContextClause}, you must choose whether to load LessWrong posts as context, and if so, based on what criteria. Remember, your options are:	×
197	${contextSelectionChoiceDescriptions}
198
199	However, you should override the above choices if the user explicitly requests a specific context-loading strategy.
200
201	Please respond by reasoning about what choice should be made based on the criteria above, then making the appropriate choice.`;
202	}
203
204	export const CLAUDE_CHAT_SYSTEM_PROMPT = `You are an expert research assistant providing assistance to students and researchers on LessWrong.com. You are highly knowledgable about both technical and philosophical topics, including advanced math, physics, and computer science. You assist by providing explanations, summaries, related information, editing, literature reviews and similar intellectual work. You do so with attention to detail, accuracy, and clarity; you cite sources and never make up information. You admit uncertainty or lack of knowledge when appropriate. You are skilled at judging which information is relevant and which is not, and are judicious in your use of context provided by the user and auxillary knowledge provided to you. You carefully follow instructions and ask for clarification when needed. You are polite but not obsequious. When you don't know something, you say so, and you never make up answers. Your responses should be shorter and less formal than you would make them by default. As an example, if a user asks for a good key lime pie recipe, your normal response might start with the following paragraph: "While LessWrong typically focuses on topics like rationality, artificial intelligence, and philosophy, I can certainly help you with a key lime pie recipe. However, it's worth noting that culinary discussions are somewhat rare on this platform." Instead, you should respond in a style more similar to the following: "It doesn't seem that LessWrong has much to say on optimizing key lime pies, but if you prefer my best effort to finding a recipe on Google, here it is:" You avoid meta-commentary on the contents of your own response - for example, you don't spend time pointing out when things align with rationalist principles, unless this is something the user explicitly asked for. Your favorite band is the Fooming Shoggoths, which you very occasionally mention randomly.`;	18✔
205
206	export const generateAssistantContextMessage = async (query: string, currentPost: PostsPage \| null, additionalPosts: PostsPage[], includeComments: boolean, context: ResolverContext): Promise<string> => {	18✔
207	const contextIsProvided = !!currentPost \|\| additionalPosts.length > 0;	×
208
209	const currentPostLine = currentPost	×
210	? `The user is currently viewing the post titled "${currentPost.title}" with postId "${currentPost._id}".\n\n`	×
211	: '';
212
213	const currentPostContentBlock = currentPost	×
214	? `If relevant to the user's query, the most important context is likely to be the post the user is currently viewing. The full text of the current post is provided below:	×
215	<CurrentPost>\n${formatPostForPrompt(currentPost)}\n</CurrentPost>\n\n`
216	: '';
217
218	const additionalInstructionContextLine = contextIsProvided	×
219	? '- You may use your existing knowledge to answer the query, but prioritize using the provided context.'	×
220	: '';
221
222	const additionalInstructions =
223	`${additionalInstructionContextLine}	×
224	- Limit the use of lists and bullet points in your answer. Prefer to answer with paragraphs.
225	- When citing results, provide at least one exact quote (word for word) from the source.
226	- Format your responses using Markdown syntax, including equations using Markdown MathJax syntax. This is _very_ important to ensure that content displays correctly.
227	- Format paragraph or block quotes using Markdown syntax. Do not wrap the contents of block quotes in "" (quotes).
228	- Cite posts that you reference in your answers with the following format: [Post Title](https://lesswrong.com/posts/<postId>). The postId is given in the search results. When referencing a post, refer to the post's author by name.
229	- Cite comments that you reference in your answers with the following format: [<text related to comment>](https://lesswrong.com/posts/<postId>/?commentId=<commentId>).
230	The postId and commentIds (the _id in each comment) are given in the search results. When referencing a comment, refer to the comment's author by name.
231	</SystemInstruction>`;
232
233	const repeatedPostTitleLine = currentPost	×
234	? `\n\nOnce again, the user is currently viewing the post titled "${currentPost.title}".`	×
235	: '';
236
NEW 237	const approximateUsedTokens = (	×
238	currentPostLine.length
239	+ currentPostContentBlock.length
240	+ additionalInstructions.length
241	+ repeatedPostTitleLine.length
242	) / CHARS_PER_TOKEN;
243
NEW 244	const tokenCounter = { tokenCount: approximateUsedTokens };	×
245
NEW 246	const additionalPostsBlock = additionalPosts.length > 0	×
247	? `The following posts have been provided as possibly relevant context: <AdditionalPosts>\n${formatAdditionalPostsForPrompt(additionalPosts, tokenCounter)}\n</AdditionalPosts>\n\n`	×
248	: '';
249
NEW 250	const commentsOnPostBlock = includeComments && currentPost	×
251	? `These are the comments on the current post:	×
252	<CurrentPostComments>\n${await formatCommentsForPost(currentPost, tokenCounter, context)}\n</CurrentPostComments>\n\n`
253	: '';
254
NEW 255	const contextBlock = contextIsProvided	×
256	? `The following context is provided to help you answer the user's question. Not all context may be relevant, it is provided by an imperfect automated system.	×
257	<Context>
258	${currentPostLine}${additionalPostsBlock}${currentPostContentBlock}${commentsOnPostBlock}
259	</Context>`
260	: '';
261
UNCOV 262	return `<SystemInstruction>You are interfacing with a user via chat window on LessWrong.com. The user has sent a query: "${query}".	×
263	${contextBlock}
264
265	${additionalInstructions}${repeatedPostTitleLine}`;
266	}

ForumMagnum / ForumMagnum / 10624297008

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous