limit to one search context in demo

This commit is contained in:
msukkari 2025-07-27 22:12:02 -07:00
parent f720ec945d
commit 2ab0b988e4
2 changed files with 93 additions and 73 deletions

View file

@ -63,40 +63,48 @@ export const AskSourcebotDemoCards = ({
return; return;
} }
const isDemoMode = process.env.NEXT_PUBLIC_SOURCEBOT_CLOUD_ENVIRONMENT === "demo";
const isSelected = selectedItems.some((item) => item.value === context.value);
if (isSelected) {
setSelectedItems(selectedItems.filter((item) => item.value !== context.value));
return;
}
const getNewSelectedItem = (): ContextItem | null => {
if (context.type === "set") { if (context.type === "set") {
const searchContext = searchContexts.find((item) => item.name === context.value); const searchContext = searchContexts.find((item) => item.name === context.value);
if (!searchContext) { if (!searchContext) {
console.error(`Search context ${context.value} not found on handleContextClick`); console.error(`Search context ${context.value} not found on handleContextClick`);
return; return null;
} }
const isSelected = selectedItems.some( return {
(selected) => selected.type === 'context' && selected.value === context.value type: 'context',
); value: context.value,
const newSelectedItems = isSelected name: context.displayName,
? selectedItems.filter( repoCount: searchContext.repoNames.length
(selected) => !(selected.type === 'context' && selected.value === context.value) } as SearchContextItem;
)
: [...selectedItems, { type: 'context', value: context.value, name: context.displayName, repoCount: searchContext.repoNames.length } as SearchContextItem];
setSelectedItems(newSelectedItems);
} else { } else {
const repo = repos.find((repo) => repo.repoName === context.value); const repo = repos.find((repo) => repo.repoName === context.value);
if (!repo) { if (!repo) {
console.error(`Repo ${context.value} not found on handleContextClick`); console.error(`Repo ${context.value} not found on handleContextClick`);
return; return null;
} }
const isSelected = selectedItems.some( return {
(selected) => selected.type === 'repo' && selected.value === context.value type: 'repo',
); value: context.value,
const newSelectedItems = isSelected name: context.displayName,
? selectedItems.filter( codeHostType: repo.codeHostType
(selected) => !(selected.type === 'repo' && selected.value === context.value) } as RepoContextItem;
) }
: [...selectedItems, { type: 'repo', value: context.value, name: context.displayName, codeHostType: repo.codeHostType } as RepoContextItem]; }
setSelectedItems(newSelectedItems); const newSelectedItem = getNewSelectedItem();
if (newSelectedItem) {
setSelectedItems(isDemoMode ? [newSelectedItem] : [...selectedItems, newSelectedItem]);
} else {
console.error(`No new selected item found on handleContextClick`);
} }
} }
@ -107,7 +115,7 @@ export const AskSourcebotDemoCards = ({
<div className="text-center mb-8"> <div className="text-center mb-8">
<div className="flex items-center justify-center gap-2 mb-2"> <div className="flex items-center justify-center gap-2 mb-2">
<Layers className="h-5 w-5 text-muted-foreground" /> <Layers className="h-5 w-5 text-muted-foreground" />
<h3 className="text-lg font-semibold">Search Context</h3> <h3 className="text-lg font-semibold">Search Contexts</h3>
</div> </div>
<p className="text-sm text-muted-foreground">Select the context you want to ask questions about</p> <p className="text-sm text-muted-foreground">Select the context you want to ask questions about</p>
</div> </div>
@ -203,7 +211,8 @@ export const AskSourcebotDemoCards = ({
</div> </div>
</CardContent> </CardContent>
</Card> </Card>
)})} )
})}
</div> </div>
</div> </div>
</div> </div>

View file

@ -97,11 +97,22 @@ export const ContextSelector = React.forwardRef<
(selected) => selected.type === item.type && selected.value === item.value (selected) => selected.type === item.type && selected.value === item.value
); );
const newSelectedItems = isSelected const isDemoMode = process.env.NEXT_PUBLIC_SOURCEBOT_CLOUD_ENVIRONMENT === "demo";
? selectedItems.filter(
let newSelectedItems: ContextItem[];
if (isSelected) {
newSelectedItems = selectedItems.filter(
(selected) => !(selected.type === item.type && selected.value === item.value) (selected) => !(selected.type === item.type && selected.value === item.value)
) );
: [...selectedItems, item]; } else {
// Limit selected context to 1 in demo mode
if (isDemoMode) {
newSelectedItems = [item];
} else {
newSelectedItems = [...selectedItems, item];
}
}
onSelectedItemsChange(newSelectedItems); onSelectedItemsChange(newSelectedItems);
}; };