fix: zhipu embedding error with empty string

Signed-off-by: Abirdcfly <fp544037857@gmail.com>
Abirdcfly · Mar 22, 2024 · 0d329e3 · 0d329e3
1 parent 8d3e355
commit 0d329e3
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/apiserver/pkg/chat/chat_server.go b/apiserver/pkg/chat/chat_server.go
@@ -289,7 +289,7 @@ func (cs *ChatServer) ListPromptStarters(ctx context.Context, req APPMetadata, l
 	content := bytes.Buffer{}
 	// if there is a knowledgebase, use it to generate prompt starter
 	if kb != nil {
-		outArg, finish, err := retriever.GenerateKnowledgebaseRetriever(ctx, c, kb.Name, kb.Namespace, apiretriever.CommonRetrieverConfig{NumDocuments: limit}, map[string]any{"question": "开始"})
+		outArg, finish, err := retriever.GenerateKnowledgebaseRetriever(ctx, c, kb.Name, kb.Namespace, apiretriever.CommonRetrieverConfig{NumDocuments: limit * 2}, map[string]any{"question": "开始"})
 		if err != nil {
 			return nil, err
 		}
@@ -314,6 +314,9 @@ func (cs *ChatServer) ListPromptStarters(ctx context.Context, req APPMetadata, l
 							question := strings.TrimSuffix(d.PageContent, "\na: "+answer)
 							promptStarters = append(promptStarters, strings.TrimPrefix(question, "q: "))
 							hasAnswer = true
+							if len(promptStarters) == limit {
+								break
+							}
 						}
 					}
 					if !hasAnswer {