Skip to content

feat(compass-assistant): show related resources below messages if it exists COMPASS-9838 #7323

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 5 commits into from
Sep 16, 2025
Merged

feat(compass-assistant): show related resources below messages if it exists COMPASS-9838 #7323

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
2a9422c
Show related resources below messages if it exists
lerouxb Sep 12, 2025
3f38951
tests
lerouxb Sep 16, 2025
ab74a2d
Update packages/compass-assistant/src/assistant-chat.tsx
lerouxb Sep 16, 2025
f660823
Merge branch 'main' into display-chat-sources
lerouxb Sep 16, 2025
2ce150f
skip in electron
lerouxb Sep 16, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 18 additions & 4 deletions packages/compass-assistant/src/assistant-chat.tsx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,6 @@ import {
LgChatChatWindow,
LgChatLeafygreenChatProvider,
LgChatMessage,
LgChatMessageActions,
LgChatInputBar,
spacing,
css,
Expand All @@ -26,7 +25,6 @@ const { DisclaimerText } = LgChatChatDisclaimer;
const { ChatWindow } = LgChatChatWindow;
const { LeafyGreenChatProvider, Variant } = LgChatLeafygreenChatProvider;
const { Message } = LgChatMessage;
const { MessageActions } = LgChatMessageActions;
const { InputBar } = LgChatInputBar;

const GEN_AI_FAQ_LINK = 'https://www.mongodb.com/docs/generative-ai-faq/';
Expand Down Expand Up @@ -141,6 +139,12 @@ const errorBannerWrapperStyles = css({
margin: spacing[400],
});

const messagesWrapStyles = css({
display: 'flex',
flexDirection: 'column',
gap: spacing[400],
});

export const AssistantChat: React.FunctionComponent<AssistantChatProps> = ({
chat,
hasNonGenuineConnections,
Expand Down Expand Up @@ -185,6 +189,13 @@ export const AssistantChat: React.FunctionComponent<AssistantChatProps> = ({
.map((part) => part.text)
.join(''),
isSender: message.role === 'user',
sources: message.parts
.filter((part) => part.type === 'source-url')
.map((part) => ({
children: part.title || part.url,
href: part.url,
variant: 'Docs',
})),
}));

const handleMessageSend = useCallback(
Expand Down Expand Up @@ -247,7 +258,7 @@ export const AssistantChat: React.FunctionComponent<AssistantChatProps> = ({
data-testid="assistant-chat-messages"
className={messageFeedFixesStyles}
>
<div>
<div className={messagesWrapStyles}>
Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

drive-by: I broke the spacing between messages in the recent PR where I fixed the scrolling in a way where it didn't in turn break text selection across messages.

{lgMessages.map((messageFields) => (
<Message
key={messageFields.id}
Expand All @@ -256,11 +267,14 @@ export const AssistantChat: React.FunctionComponent<AssistantChatProps> = ({
data-testid={`assistant-message-${messageFields.id}`}
>
{messageFields.isSender === false && (
<MessageActions
<Message.Actions
Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

drive-by: MessageActions is deprecated. Just like MessageLinks. These are now on Message like Message.Actions and Message.Links.

onRatingChange={handleFeedback}
onSubmitFeedback={handleFeedback}
/>
)}
{messageFields.sources.length > 0 && (
<Message.Links links={messageFields.sources} />
)}
</Message>
))}
</div>
Expand Down
2 changes: 1 addition & 1 deletion packages/compass-assistant/src/docs-provider-transport.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@ export class DocsProviderTransport implements ChatTransport<UIMessage> {
},
});

return Promise.resolve(result.toUIMessageStream());
return Promise.resolve(result.toUIMessageStream({ sendSources: true }));
}

reconnectToStream(): Promise<ReadableStream<UIMessageChunk> | null> {
Expand Down
2 changes: 2 additions & 0 deletions packages/compass-assistant/src/prompts.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,8 @@ You CANNOT:
2. Query MongoDB directly or execute code.
3. Access the current state of the UI
</inabilities>

Always call the 'search_content' tool when asked a technical question that would benefit from getting relevant info from the documentation.
Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The Jedi mind trick of getting this thing to use the sources and therefore reliably return some. https://mongodb.github.io/chatbot/server/context-engineering#elicit-internal-retrieval-augmented-generation-tool-calls

`;
};

Expand Down
1 change: 0 additions & 1 deletion packages/compass-assistant/test/assistant.eval.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -178,7 +178,6 @@ async function makeAssistantCall(

const sources = resolvedSources
.map((source) => {
console.log(source);
return source.url;
})
.filter((url) => !!url);
Expand Down
Loading