Ling0925
/
buster
mirror of https://github.com/buster-so/buster.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1126 from buster-so/wells-bus-1900-braintrust-metadata-for-generate-suggested-prompts 

add metadata to generate suggested prompts
This commit is contained in:
wellsbunk5 2025-09-24 15:20:04 -06:00 committed by GitHub
parent e46da44445 9a046f7911
commit 3a9a8c0b2a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
packages/ai/src/tasks/suggested-prompts/suggested-prompts-task.ts
View File

@ -1,6 +1,7 @@
import { getUser } from '@buster/database/queries';
import { DEFAULT_USER_SUGGESTED_PROMPTS } from '@buster/database/schema-types';
import { generateObject } from 'ai';
import { wrapTraced } from 'braintrust';
import { currentSpan, wrapTraced } from 'braintrust';
import { z } from 'zod';
import { GPT5Nano } from '../../llm';
import SUGGESTED_PROMPTS_SYSTEM_PROMPT from './suggested-prompts-system-prompt.txt';
@ -46,8 +47,19 @@ ${chatHistoryText}
Generate suggestions that are relevant to the conversation context and available data.`;
const userInfo = await getUser({ id: userId });
const tracedGeneration = wrapTraced(
async () => {
currentSpan().log({
metadata: {
userName: userInfo.name || 'Unknown',
userEmail: userInfo.email,
userId: userId,
chatHistoryTextLength: chatHistoryText.length,
},
});
const result = await generateObject({
model: GPT5Nano,
prompt: userMessage,
@ -74,10 +86,6 @@ Generate suggestions that are relevant to the conversation context and available
},
{
name: 'Generate Suggested Prompts',
spanAttributes: {
chatHistoryLength: chatHistoryText.length,
userId: userId,
},
}
);