phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-13 04:22:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge 679e2d09c1 into 63422e5b36

Browse source
This commit is contained in:
Francisco Arceo 2025-10-27 10:47:57 -07:00 • committed by GitHub
commit 1665c6c4be
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
470 changed files with 17725 additions and 2810 deletions
Download patch file Download diff file Expand all files Collapse all files

8
llama_stack/ui/app/chat-playground/page.test.tsx
View file

@ -144,7 +144,7 @@ const mockModels = [
const mockToolgroups = [
{
identifier: "builtin::rag",
identifier: "builtin::file_search",
provider_id: "test-provider",
type: "tool_group",
provider_resource_id: "test-resource",
@ -171,7 +171,7 @@ describe("ChatPlaygroundPage", () => {
mockClient.agents.retrieve.mockResolvedValue({
agent_id: "test-agent",
agent_config: {
toolgroups: ["builtin::rag"],
toolgroups: ["builtin::file_search"],
instructions: "Test instructions",
model: "test-model",
},
@ -629,7 +629,7 @@ describe("ChatPlaygroundPage", () => {
agent_config: {
toolgroups: [
{
name: "builtin::rag/knowledge_search",
name: "builtin::file_search/knowledge_search",
args: { vector_db_ids: ["test-vector-db"] },
},
],
@ -664,7 +664,7 @@ describe("ChatPlaygroundPage", () => {
agent_config: {
toolgroups: [
{
name: "builtin::rag/knowledge_search",
name: "builtin::file_search/knowledge_search",
args: { vector_db_ids: ["test-vector-db"] },
},
],

13
llama_stack/ui/app/chat-playground/page.tsx
View file

@ -433,7 +433,7 @@ export default function ChatPlaygroundPage() {
) => {
try {
const processedToolgroups = toolgroups.map(toolgroup => {
if (toolgroup === "builtin::rag" && vectorDBs.length > 0) {
if (toolgroup === "builtin::file_search" && vectorDBs.length > 0) {
return {
name: "builtin::rag/knowledge_search",
args: {
@ -1167,7 +1167,10 @@ export default function ChatPlaygroundPage() {
// find RAG toolgroups that have vector_db_ids configured
const ragToolgroups = selectedAgentConfig.toolgroups.filter(toolgroup => {
if (typeof toolgroup === "object" && toolgroup.name?.includes("rag")) {
if (
typeof toolgroup === "object" &&
toolgroup.name?.includes("file_search")
) {
return toolgroup.args && "vector_db_ids" in toolgroup.args;
}
return false;
@ -1505,7 +1508,7 @@ export default function ChatPlaygroundPage() {
const toolArgs =
typeof toolgroup === "object" ? toolgroup.args : null;
const isRAGTool = toolName.includes("rag");
const isRAGTool = toolName.includes("file_search");
const displayName = isRAGTool ? "RAG Search" : toolName;
const displayIcon = isRAGTool
? "🔍"
@ -1761,7 +1764,7 @@ export default function ChatPlaygroundPage() {
</div>
{/* Vector DB Configuration for RAG */}
{selectedToolgroups.includes("builtin::rag") && (
{selectedToolgroups.includes("builtin::file_search") && (
<div>
<label className="text-sm font-medium block mb-2">
Vector Databases for RAG
@ -1825,7 +1828,7 @@ export default function ChatPlaygroundPage() {
)}
</div>
{selectedVectorDBs.length === 0 &&
selectedToolgroups.includes("builtin::rag") && (
selectedToolgroups.includes("builtin::file_search") && (
<p className="text-xs text-muted-foreground mt-1">
⚠️ RAG tool selected but no vector databases chosen.
Create or select a vector database.