fix: when no related doc is found, should return retriever.spec.docNu…

…llReturn Signed-off-by: Abirdcfly <fp544037857@gmail.com>
Lanture1064 · Jan 19, 2024 · 360066b · 360066b
1 parent 36f6e18
commit 360066b
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 3 deletions.
diff --git a/.github/workflows/data_processing_image_build_test.yml b/.github/workflows/data_processing_image_build_test.yml
@@ -3,6 +3,8 @@ name: Build KubeAGI Data Processing images
 on:
   pull_request:
     branches: [main]
+    paths:
+      - 'pypi/data-processing'
   workflow_dispatch:
 
 jobs:
@@ -27,4 +29,4 @@ jobs:
           context: ./pypi/data-processing
           file: ./pypi/data-processing/Dockerfile
           platforms: linux/amd64,linux/arm64
-          push: false
+          push: false
diff --git a/pkg/appruntime/retriever/knowledgebaseretriever.go b/pkg/appruntime/retriever/knowledgebaseretriever.go
@@ -168,8 +168,15 @@ type KnowledgeBaseStuffDocuments struct {
 	References []Reference
 }
 
-var _ chains.Chain = &KnowledgeBaseStuffDocuments{}
-var _ callbacks.Handler = &KnowledgeBaseStuffDocuments{}
+func (c *KnowledgeBaseStuffDocuments) GetCallbackHandler() callbacks.Handler {
+	return c
+}
+
+var (
+	_ chains.Chain           = &KnowledgeBaseStuffDocuments{}
+	_ callbacks.Handler      = &KnowledgeBaseStuffDocuments{}
+	_ callbacks.HandlerHaver = &KnowledgeBaseStuffDocuments{}
+)
 
 func (c *KnowledgeBaseStuffDocuments) joinDocuments(ctx context.Context, docs []langchaingoschema.Document) string {
 	logger := klog.FromContext(ctx)

diff --git a/tests/example-test.sh b/tests/example-test.sh
@@ -365,12 +365,26 @@ kubectl apply -f config/samples/app_retrievalqachain_knowledgebase.yaml
 waitCRDStatusReady "Application" "arcadia" "base-chat-with-knowledgebase"
 sleep 3
 getRespInAppChat "base-chat-with-knowledgebase" "arcadia" "旷工最小计算单位为多少天？" "" "true"
+info "8.2.1.2 When no related doc is found, return retriever.spec.docNullReturn info"
+getRespInAppChat "base-chat-with-knowledgebase" "arcadia" "飞天的主演是谁？" "" "false"
+expected=$(kubectl get knowledgebaseretrievers -n arcadia base-chat-with-knowledgebase -o json | jq -r .spec.docNullReturn)
+if [[ $ai_data != $expected ]]; then
+	echo "when no related doc is found, return retriever.spec.docNullReturn info should be:"$expected ", but resp:"$resp
+	exit 1
+fi
 
 info "8.2.2 QA app using knowledgebase base on pgvector"
 kubectl apply -f config/samples/app_retrievalqachain_knowledgebase_pgvector.yaml
 waitCRDStatusReady "Application" "arcadia" "base-chat-with-knowledgebase-pgvector"
 sleep 3
 getRespInAppChat "base-chat-with-knowledgebase" "arcadia" "旷工最小计算单位为多少天？" "" "true"
+info "8.2.2.2 When no related doc is found, return retriever.spec.docNullReturn info"
+getRespInAppChat "base-chat-with-knowledgebase" "arcadia" "飞天的主演是谁？" "" "false"
+expected=$(kubectl get knowledgebaseretrievers -n arcadia base-chat-with-knowledgebase -o json | jq -r .spec.docNullReturn)
+if [[ $ai_data != $expected ]]; then
+	echo "when no related doc is found, return retriever.spec.docNullReturn info should be:"$expected ", but resp:"$resp
+	exit 1
+fi
 
 info "8.3 conversation chat app"
 kubectl apply -f config/samples/app_llmchain_chat_with_bot.yaml