ref: sidebar experience improvement with new bundles (#8824)

* cohere sidebar

* side bar embeddings

* move memories

* sort style utils

* [autofix.ci] apply automated fixes

* serach components add

* Update __init__.py

* Update styleUtils.ts

* [autofix.ci] apply automated fixes

* Update model_input_constants.py

* fix lint

* update to wikipedia

* ✨ (tableInputComponent.spec.ts): Refactor test to use for loops for better readability and efficiency
✨ (filterEdge-shard-1.spec.ts): Update model names for better clarity and consistency
🐛 (duckduckgo.spec.ts): Update selector for search button to match changes in the frontend
🐛 (generalBugs-shard-11.spec.ts): Update selector for search button to match changes in the frontend

* [autofix.ci] apply automated fixes

* update tests path

* update tests

* Update test_arxiv_component.py

* [autofix.ci] apply automated fixes

* make embeddings functions

---------

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
Co-authored-by: cristhianzl <cristhian.lousa@gmail.com>
This commit is contained in:
Edwin Jose 2025-07-02 15:11:59 -05:00 • committed by GitHub
commit e28093234b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
84 changed files with 264 additions and 238 deletions

View file

@ -228,69 +228,87 @@ export const SIDEBAR_CATEGORIES = [
];
export const SIDEBAR_BUNDLES = [
{
display_name: "Language Models",
name: "languagemodels",
icon: "BrainCircuit",
},
{ display_name: "Embeddings", name: "embeddings", icon: "Binary" },
{ display_name: "Memories", name: "memories", icon: "Cpu" },
{ display_name: "AI/ML", name: "aiml", icon: "AI/ML" },
{ display_name: "Anthropic", name: "anthropic", icon: "Anthropic" },
{ display_name: "Amazon", name: "amazon", icon: "Amazon" },
// Add apify
{ display_name: "Apify", name: "apify", icon: "Apify" },
{ display_name: "LangChain", name: "langchain_utilities", icon: "LangChain" },
{ display_name: "AgentQL", name: "agentql", icon: "AgentQL" },
{ display_name: "Amazon", name: "amazon", icon: "Amazon" },
{ display_name: "Anthropic", name: "anthropic", icon: "Anthropic" },
{ display_name: "Apify", name: "apify", icon: "Apify" },
{ display_name: "Arxiv", name: "arxiv", icon: "arXiv" },
{ display_name: "AssemblyAI", name: "assemblyai", icon: "AssemblyAI" },
{ display_name: "Azure", name: "azure", icon: "Azure" },
{
display_name: "DataStax",
name: "datastax",
icon: "AstraDB",
},
{ display_name: "Baidu", name: "baidu", icon: "BaiduQianfan" },
{ display_name: "Bing", name: "bing", icon: "Bing" },
{ display_name: "Cleanlab", name: "cleanlab", icon: "Cleanlab" },
{ display_name: "Cloudflare", name: "cloudflare", icon: "Cloudflare" },
{ display_name: "Cohere", name: "cohere", icon: "Cohere" },
{ display_name: "Composio", name: "composio", icon: "Composio" },
{ display_name: "Confluence", name: "confluence", icon: "Confluence" },
{ display_name: "CrewAI", name: "crewai", icon: "CrewAI" },
{ display_name: "DataStax", name: "datastax", icon: "AstraDB" },
{ display_name: "DeepSeek", name: "deepseek", icon: "DeepSeek" },
{ display_name: "Docling", name: "docling", icon: "Docling" },
{ display_name: "Gmail", name: "gmail", icon: "Gmail" },
{ display_name: "DuckDuckGo", name: "duckduckgo", icon: "DuckDuckGo" },
{ display_name: "Embeddings", name: "embeddings", icon: "Binary" },
{ display_name: "Exa", name: "exa", icon: "Exa" },
{ display_name: "Firecrawl", name: "firecrawl", icon: "FirecrawlCrawlApi" },
{ display_name: "Git", name: "git", icon: "GitLoader" },
{ display_name: "GitHub", name: "github", icon: "Github" },
{ display_name: "Glean", name: "glean", icon: "Glean" },
{ display_name: "Gmail", name: "gmail", icon: "Gmail" },
{ display_name: "Google", name: "google", icon: "Google" },
{
display_name: "Googlecalendar",
name: "googlecalendar",
icon: "Googlecalendar",
},
{ display_name: "HuggingFace", name: "huggingface", icon: "HuggingFace" },
{ display_name: "IBM", name: "ibm", icon: "WatsonxAI" },
{ display_name: "LangWatch", name: "langwatch", icon: "Langwatch" },
{ display_name: "Olivya", name: "olivya", icon: "Olivya" },
{ display_name: "Outlook", name: "outlook", icon: "Outlook" },
{ display_name: "OpenAI", name: "openai", icon: "OpenAI" },
{ display_name: "Notion", name: "Notion", icon: "Notion" },
{ display_name: "Needle", name: "needle", icon: "Needle" },
{ display_name: "NVIDIA", name: "nvidia", icon: "NVIDIA" },
{ display_name: "Vectara", name: "vectara", icon: "Vectara" },
{ display_name: "Icosa Computing", name: "icosacomputing", icon: "Icosa" },
{ display_name: "Google", name: "google", icon: "Google" },
{ display_name: "CrewAI", name: "crewai", icon: "CrewAI" },
{ display_name: "NotDiamond", name: "notdiamond", icon: "NotDiamond" },
{ display_name: "Composio", name: "composio", icon: "Composio" },
{ display_name: "Cohere", name: "cohere", icon: "Cohere" },
{ display_name: "Firecrawl", name: "firecrawl", icon: "FirecrawlCrawlApi" },
{ display_name: "Unstructured", name: "unstructured", icon: "Unstructured" },
{ display_name: "Git", name: "git", icon: "GitLoader" },
{ display_name: "Confluence", name: "confluence", icon: "Confluence" },
{ display_name: "Mem0", name: "mem0", icon: "Mem0" },
{ display_name: "Youtube", name: "youtube", icon: "YouTube" },
{ display_name: "ScrapeGraph AI", name: "scrapegraph", icon: "ScrapeGraph" },
{ display_name: "Twelve Labs", name: "twelvelabs", icon: "TwelveLabs" },
{ display_name: "Groq", name: "groq", icon: "Groq" },
{
display_name: "Home Assistant",
name: "homeassistant",
icon: "HomeAssistant",
},
{ display_name: "Cleanlab", name: "cleanlab", icon: "Cleanlab" },
{ display_name: "Search", name: "search", icon: "Search" },
{ display_name: "Tavily", name: "tavily", icon: "TavilyIcon" },
{ display_name: "HuggingFace", name: "huggingface", icon: "HuggingFace" },
{ display_name: "IBM", name: "ibm", icon: "WatsonxAI" },
{ display_name: "Icosa Computing", name: "icosacomputing", icon: "Icosa" },
{ display_name: "LangChain", name: "langchain_utilities", icon: "LangChain" },
{ display_name: "LangWatch", name: "langwatch", icon: "Langwatch" },
{ display_name: "LMStudio", name: "lmstudio", icon: "LMStudio" },
{ display_name: "MariTalk", name: "maritalk", icon: "MariTalk" },
{ display_name: "Mem0", name: "mem0", icon: "Mem0" },
{ display_name: "Memories", name: "memories", icon: "Cpu" },
{ display_name: "MistralAI", name: "mistral", icon: "MistralAI" },
{ display_name: "Needle", name: "needle", icon: "Needle" },
{ display_name: "NotDiamond", name: "notdiamond", icon: "NotDiamond" },
{ display_name: "Notion", name: "Notion", icon: "Notion" },
{ display_name: "Novita", name: "novita", icon: "Novita" },
{ display_name: "NVIDIA", name: "nvidia", icon: "NVIDIA" },
{ display_name: "Olivya", name: "olivya", icon: "Olivya" },
{ display_name: "Ollama", name: "ollama", icon: "Ollama" },
{ display_name: "OpenAI", name: "openai", icon: "OpenAI" },
{ display_name: "OpenRouter", name: "openrouter", icon: "OpenRouter" },
{ display_name: "Outlook", name: "outlook", icon: "Outlook" },
{ display_name: "Perplexity", name: "perplexity", icon: "Perplexity" },
{ display_name: "Redis", name: "redis", icon: "Redis" },
{ display_name: "SambaNova", name: "sambanova", icon: "SambaNova" },
{ display_name: "ScrapeGraph AI", name: "scrapegraph", icon: "ScrapeGraph" },
{ display_name: "SearchAPI", name: "searchapi", icon: "SearchAPI" },
{ display_name: "SerpApi", name: "serpapi", icon: "SerpSearch" },
{ display_name: "Tavily", name: "tavily", icon: "TavilyIcon" },
{ display_name: "Twelve Labs", name: "twelvelabs", icon: "TwelveLabs" },
{ display_name: "Unstructured", name: "unstructured", icon: "Unstructured" },
{ display_name: "Vectara", name: "vectara", icon: "Vectara" },
{ display_name: "VertexAI", name: "vertexai", icon: "VertexAI" },
{ display_name: "Wikipedia", name: "wikipedia", icon: "Wikipedia" },
{
display_name: "WolframAlpha",
name: "wolframalpha",
icon: "WolframAlphaAPI",
},
{ display_name: "xAI", name: "xai", icon: "xAI" },
{ display_name: "YahooSearch", name: "yahoosearch", icon: "trending-up" },
{ display_name: "Youtube", name: "youtube", icon: "YouTube" },
{ display_name: "Zep", name: "zep", icon: "ZepMemory" },
];
export const categoryIcons: Record<string, string> = {

View file

@ -177,13 +177,14 @@ class CustomComponent(Component):
const visibleTexts = ["Alpha", "Bravo", "Charlie", "Delta", "Echo"];
const notVisibleTexts = ["X1", "thirdRandomText"];
await Promise.all(
visibleTexts.map((text) => expect(page.getByText(text)).toBeVisible()),
);
await Promise.all(
notVisibleTexts.map((text) =>
expect(page.getByText(text)).not.toBeVisible(),
),
);
for (const text of visibleTexts) {
const count = await page.getByText(text).count();
expect(count).toBeGreaterThanOrEqual(1);
}
for (const text of notVisibleTexts) {
const count = await page.getByText(text).count();
expect(count).toBe(0);
}
},
);

View file

@ -75,7 +75,7 @@ test(
"vectorstoresAstra DB",
"langchain_utilitiesTool Calling Agent",
"langchain_utilitiesConversationChain",
"memoriesMem0 Chat Memory",
"mem0Mem0 Chat Memory",
"logicCondition",
"langchain_utilitiesSelf Query Retriever",
"langchain_utilitiesCharacterTextSplitter",
@ -101,15 +101,15 @@ test(
await page.getByTestId("sidebar-search-input").click();
const visibleModelSpecsTestIds = [
"languagemodelsCohere",
"languagemodelsGroq",
"languagemodelsLM Studio",
"languagemodelsMaritalk",
"languagemodelsMistralAI",
"languagemodelsPerplexity",
"languagemodelsQianfan",
"languagemodelsSambaNova",
"languagemodelsxAI",
"cohereCohere Language Models",
"groqGroq",
"lmstudioLM Studio",
"maritalkMaritalk",
"mistralMistralAI",
"perplexityPerplexity",
"baiduQianfan",
"sambanovaSambaNova",
"xaixAI",
];
await Promise.all(

View file

@ -11,12 +11,15 @@ test(
await page.getByTestId("sidebar-search-input").click();
await page.getByTestId("sidebar-search-input").fill("duck");
await page.waitForSelector('[data-testid="searchDuckDuckGo Search"]', {
timeout: 3000,
});
await page.waitForSelector(
'[data-testid="disclosure-bundles-duckduckgo"]',
{
timeout: 3000,
},
);
await page
.getByTestId("searchDuckDuckGo Search")
.getByTestId("duckduckgoDuckDuckGo Search")
.hover()
.then(async () => {
await page

View file

@ -49,14 +49,14 @@ test(
await page.getByTestId("sidebar-search-input").click();
await page.getByTestId("sidebar-search-input").fill("search api");
await page.waitForSelector('[data-testid="searchSearch API"]', {
await page.waitForSelector('[data-testid="searchapiSearch API"]', {
timeout: 1000,
});
await zoomOut(page, 3);
await page
.getByTestId("searchSearch API")
.getByTestId("searchapiSearch API")
.dragTo(page.locator('//*[@id="react-flow-id"]'), {
targetPosition: { x: 100, y: 100 },
});