Merge pull request #815 from bjwswang/main

fix: set MaxNumberOfConcurrent to 1 if not provided
kubeagi · Mar 8, 2024 · d2a7365 · d2a7365
2 parents a42c127 + 0692bb3
commit d2a7365
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/config/samples/app_llmchain_abstract.yaml b/config/samples/app_llmchain_abstract.yaml
@@ -91,6 +91,7 @@ metadata:
 spec:
   displayName: "llm chain"
   description: "llm chain"
+  maxNumberOfConccurent: 1
   maxTokens: 20480
   maxLength: 20480
   model: "6ac7baa2-71e7-4ffc-bd49-9356e743ecbb"

diff --git a/pkg/appruntime/chain/mpchain.go b/pkg/appruntime/chain/mpchain.go
@@ -102,7 +102,7 @@ func (l *MapReduceChain) Run(ctx context.Context, _ client.Client, args map[stri
 	v2, ok := args["max_number_of_conccurent"]
 	if ok {
 		maxNumberOfConcurrent, ok := v2.(int)
-		if ok {
+		if ok && maxNumberOfConcurrent > 0 {
 			l.MapReduceDocuments.MaxNumberOfConcurrent = maxNumberOfConcurrent
 		}
 	}