Skip to content

Commit 2847a3f

Browse files
authored
Fix: make sure task can use agent's knowledge (#1870)
Signed-off-by: Daishan Peng <[email protected]>
1 parent d96c83e commit 2847a3f

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

pkg/api/handlers/tasks.go

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -727,6 +727,12 @@ func (t *TaskHandler) CreateFromScope(req api.Context) error {
727727
if thread.Spec.AgentName != "" {
728728
additionalCredentialContexts = append(additionalCredentialContexts, thread.Spec.AgentName)
729729
}
730+
731+
var agent v1.Agent
732+
if err := req.Get(&agent, thread.Spec.AgentName); err != nil {
733+
return err
734+
}
735+
730736
workflow := v1.Workflow{
731737
ObjectMeta: metav1.ObjectMeta{
732738
GenerateName: system.WorkflowPrefix,
@@ -735,7 +741,7 @@ func (t *TaskHandler) CreateFromScope(req api.Context) error {
735741
Spec: v1.WorkflowSpec{
736742
ThreadName: thread.Name,
737743
Manifest: workflowManifest,
738-
KnowledgeSetNames: thread.Status.KnowledgeSetNames,
744+
KnowledgeSetNames: append(agent.Status.KnowledgeSetNames, thread.Status.KnowledgeSetNames...),
739745
WorkspaceName: workspace.Name,
740746
CredentialContextID: thread.Name,
741747
AdditionalCredentialContexts: additionalCredentialContexts,

0 commit comments

Comments
 (0)