Merge pull request #36 from ssdeanx/develop

feat: enhance agent components and workflows

Author: ssdeanx

app/chat/helpers/tool-part-transform.ts

@@ -128,7 +128,7 @@ export function mapDataToolPartToDynamicToolPart(part: any): DynamicToolUIPart |
     (inner?.state ?? inner?.status ?? payload?.state ?? payload?.status ?? "").toString() ?? "";
 
   // Convert a free-form status to tool states defined by the SDK
-  function mapToToolState(s: string): DynamicToolUIPart["state"] {
+  const mapToToolState = (s: string): DynamicToolUIPart["state"] => {
     const st = String(s ?? "").toLowerCase();
 
     if (!st) {
@@ -164,7 +164,7 @@ export function mapDataToolPartToDynamicToolPart(part: any): DynamicToolUIPart |
     // fallback: if an output is present consider finished, otherwise input available
     if (output !== undefined && output !== null) {return "output-available";}
     return "input-available";
-  }
+  };
 
   const state = mapToToolState(rawState);
 

app/dashboard/agents/_components/agent-details.tsx

@@ -42,7 +42,13 @@ export function AgentDetails({ agentId }: AgentDetailsProps) {
         )}
         <div className="flex gap-2">
           <Badge variant="outline">{agent.id}</Badge>
-          {agent.model && <Badge variant="secondary">{agent.model}</Badge>}
+          {agent.model && (
+            <Badge variant="secondary">
+              {typeof agent.model === "string"
+                ? agent.model
+                : `${agent.model.provider}${agent.model.name ? ` / ${agent.model.name}` : ""}`}
+            </Badge>
+          )}
         </div>
       </div>
 
@@ -76,7 +82,7 @@ export function AgentDetails({ agentId }: AgentDetailsProps) {
           {agent.model && (
             <div>
               <h4 className="text-sm font-medium mb-2">Model</h4>
-              <p className="text-sm text-muted-foreground">{agent.model}</p>
+              {typeof agent.model === "string" ? agent.model : `${agent.model.provider}${agent.model.name ? ` / ${agent.model.name}` : ""}`}
             </div>
           )}
 

app/dashboard/agents/_components/agent-tab.tsx

@@ -62,13 +62,16 @@ export function AgentTab({ agentId }: AgentTabProps) {
         </div>
       </div>
 
-      {agent.tools && agent.tools.length > 0 && (
+      {Array.isArray(agent.tools) && agent.tools.length > 0 && (
         <div>
           <h4 className="text-sm font-medium mb-3">Tools</h4>
           <div className="flex flex-wrap gap-2">
-            {agent.tools.map((tool, index) => (
-              <Badge key={index} variant="secondary">{tool.name}</Badge>
-            ))}
+            {agent.tools.map((tool, index) => {
+              const label = typeof tool === "string" ? tool : (tool.name ?? tool.id)
+              return (
+                <Badge key={index} variant="secondary">{label}</Badge>
+              )
+            })}
           </div>
         </div>
       )}

app/dashboard/agents/_components/agent-tools-tab.tsx

@@ -10,7 +10,11 @@ interface AgentToolsTabProps {
 }
 
 export function AgentToolsTab({ agent }: AgentToolsTabProps) {
-  const tools = agent.tools ?? []
+  const tools = Array.isArray(agent.tools)
+    ? agent.tools
+    : agent.tools
+    ? [agent.tools]
+    : []
 
   if (tools.length === 0) {
     return (
@@ -25,16 +29,21 @@ export function AgentToolsTab({ agent }: AgentToolsTabProps) {
   return (
     <div className="space-y-3">
       {tools.map((tool, index) => {
-        const toolId = typeof tool === "string" ? tool : tool.id
-        const toolName = typeof tool === "string" ? tool : tool.name || tool.id
+        const toolId = typeof tool === "string" ? tool : (tool as any).id
+        const toolName = typeof tool === "string" ? tool : tool.name || (tool as any).id
         return (
           <div
-            key={`${toolId}-${index}`}
+            key={`${toolName}-${toolId}-${index}`}
             className="flex items-center justify-between p-3 rounded-md border"
+            data-testid={`agent-tool-${toolId}-${toolName}`}
           >
             <div className="flex items-center gap-3">
               <Wrench className="h-4 w-4 text-muted-foreground" />
-              <span className="font-medium">{toolName}</span>
+              <span className="text-sm font-medium">
+                {typeof tool === "string"
+                  ? tool
+                  : String(tool.name ?? (tool as any).id ?? JSON.stringify(tool))}
+              </span>
             </div>
             <Badge variant="secondary">Tool</Badge>
           </div>

app/dashboard/agents/_components/index.ts

@@ -2,4 +2,4 @@ export { AgentList } from "./agent-list"
 export { AgentListItem } from "./agent-list-item"
 export { AgentDetails } from "./agent-details"
 export { AgentToolsTab } from "./agent-tools-tab"
-export { AgentEvalsTab } from "./agent-tab"
+export { AgentTab } from "./agent-tab"

src/mastra/agents/sql.ts

@@ -89,6 +89,7 @@ import { weatherWorkflow } from './workflows/weather-workflow';
 import { repoIngestionWorkflow } from './workflows/repo-ingestion-workflow';
 import { specGenerationWorkflow } from './workflows/spec-generation-workflow';
 import { ResearchRuntimeContext } from './agents/index';
+import { governedRagIndex } from "./workflows/governed-rag-index.workflow";
 
 const ml = process.env.MLFLOW_EXPERIMENT_ID
 
@@ -106,6 +107,8 @@ export const mastra = new Mastra({
     telephoneGameWorkflow,
     repoIngestionWorkflow,
     specGenerationWorkflow,
+    governedRagIndex,
+    
   },
   agents: {
     // Core Agents

src/mastra/index.ts

@@ -279,7 +279,7 @@ const stepA2 = createStep({
 });
 
 const changelogWorkflow = createWorkflow({
-  id: 'changelog',
+  id: 'changelogWorkflow',
   inputSchema: z.object({
     channelId: z.string(),
   }),

src/mastra/workflows/changelog.ts

@@ -700,7 +700,7 @@ const finalizeContentStep = createStep({
 });
 
 export const contentReviewWorkflow = createWorkflow({
-  id: 'content-review-workflow',
+  id: 'contentReviewWorkflow',
   description: 'Content creation workflow with iterative quality review using .dowhile() loop',
   inputSchema: contentInputSchema,
   outputSchema: finalOutputSchema,

src/mastra/workflows/content-review-workflow.ts

@@ -415,7 +415,7 @@ Script: ${refinedScript}`;
 // --- Workflow ---
 
 export const contentStudioWorkflow = createWorkflow({
-  id: 'content-studio-workflow',
+  id: 'contentStudioWorkflow',
   inputSchema: z.object({
     topic: z.string(),
   }),