diff --git a/apps/rowboat/app/projects/[projectId]/workflow/entity_list.tsx b/apps/rowboat/app/projects/[projectId]/workflow/entity_list.tsx index c8538636..4e402f61 100644 --- a/apps/rowboat/app/projects/[projectId]/workflow/entity_list.tsx +++ b/apps/rowboat/app/projects/[projectId]/workflow/entity_list.tsx @@ -55,6 +55,7 @@ interface EntityListProps { name: string; } | null; startAgentName: string | null; + isLive?: boolean; onSelectAgent: (name: string) => void; onSelectTool: (name: string) => void; onSelectPrompt: (name: string) => void; @@ -63,6 +64,10 @@ interface EntityListProps { onAddAgent: (agent: Partial>) => void; onAddTool: (tool: Partial>) => void; onAddPrompt: (prompt: Partial>) => void; + onShowAddDataSourceModal?: () => void; + onShowAddVariableModal?: () => void; + onShowAddAgentModal?: () => void; + onShowAddToolModal?: () => void; onUpdatePrompt: (name: string, prompt: Partial>) => void; onAddPromptFromModal: (prompt: Partial>) => void; onUpdatePromptFromModal: (name: string, prompt: Partial>) => void; @@ -315,6 +320,7 @@ interface PipelineCardProps { onSetMainAgent: (name: string) => void; selectedRef: React.RefObject; startAgentName: string | null; + isLive?: boolean; dragHandle?: React.ReactNode; } @@ -330,6 +336,7 @@ const PipelineCard = ({ onSetMainAgent, selectedRef, startAgentName, + isLive, dragHandle, }: PipelineCardProps) => { // Get agents that belong to this pipeline @@ -474,7 +481,12 @@ const PipelineCard = ({ }; export const EntityList = forwardRef< - { openDataSourcesModal: () => void }, + { + openDataSourcesModal: () => void; + openAddVariableModal: () => void; + openAddAgentModal: () => void; + openAddToolModal: () => void; + }, EntityListProps & { projectId: string, onReorderAgents: (agents: z.infer[]) => void @@ -488,6 +500,7 @@ export const EntityList = forwardRef< workflow, selectedEntity, startAgentName, + isLive, onSelectAgent, onSelectTool, onSelectPrompt, @@ -496,6 +509,10 @@ export const EntityList = forwardRef< onAddAgent, onAddTool, onAddPrompt, + onShowAddDataSourceModal, + onShowAddVariableModal, + onShowAddAgentModal, + onShowAddToolModal, onUpdatePrompt, onAddPromptFromModal, onUpdatePromptFromModal, @@ -701,6 +718,15 @@ export const EntityList = forwardRef< useImperativeHandle(ref, () => ({ openDataSourcesModal: () => { setShowDataSourcesModal(true); + }, + openAddVariableModal: () => { + setShowAddVariableModal(true); + }, + openAddAgentModal: () => { + setShowAgentTypeModal(true); + }, + openAddToolModal: () => { + setShowToolsModal(true); } })); @@ -755,17 +781,17 @@ export const EntityList = forwardRef< )} - @@ -803,6 +829,7 @@ export const EntityList = forwardRef< onSetMainAgent={onSetMainAgent} selectedRef={selectedRef} startAgentName={startAgentName} + isLive={isLive} /> ))} @@ -891,17 +918,17 @@ export const EntityList = forwardRef< Tools
- @@ -1039,17 +1066,17 @@ export const EntityList = forwardRef< Data
- @@ -1190,17 +1217,17 @@ export const EntityList = forwardRef< Variables
- @@ -1737,7 +1764,8 @@ const SortablePipelineItem = ({ onAddAgentToPipeline, onSetMainAgent, selectedRef, - startAgentName + startAgentName, + isLive }: { pipeline: z.infer; agents: z.infer[]; @@ -1753,6 +1781,7 @@ const SortablePipelineItem = ({ onSetMainAgent: (name: string) => void; selectedRef: React.RefObject; startAgentName: string | null; + isLive?: boolean; }) => { const { attributes, @@ -1783,6 +1812,7 @@ const SortablePipelineItem = ({ onSetMainAgent={onSetMainAgent} selectedRef={selectedRef} startAgentName={startAgentName} + isLive={isLive} dragHandle={