fix: bedrock knowledge base null exception (#2165)

* fix: null exception

* refactor: 使用 Dify 插件标准日志系统替换文件日志

- 移除文件写入方式 (/tmp/bedrock.log)
- 使用 plugin_logger_handler 实现标准日志输出
- 日志现在输出到 plugin-daemon 容器，便于 Kubernetes 环境下的日志收集和监控
- 保持原有 log() 函数接口不变，确保代码兼容性

* Update extensions/aws_bedrock_knowledge_base/endpoints/aws_bedrock_knowledge_base.py

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>

* fix: code review

* fix: upgrade version 0.0.4

---------

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>

Author: zhangzhenhuajack

extensions/aws_bedrock_knowledge_base/endpoints/aws_bedrock_knowledge_base.py

@@ -4,16 +4,48 @@
 from werkzeug.wrappers import Request, Response
 from dify_plugin import Endpoint
 from typing import Mapping
+import logging
+from dify_plugin.config.logger_format import plugin_logger_handler
+
+logger = logging.getLogger(__name__)
+logger.setLevel(logging.INFO)
+logger.addHandler(plugin_logger_handler)
+
+
+def log(msg):
+    logger.info(msg)
 
 
 class Knowledgebaseretrieval(Endpoint):
     def _invoke(self, r: Request, values: Mapping, settings: Mapping) -> Response:
-        body = r.get_json()
+        log("Knowledge base retrieval invoked.22")
+        
+        try:
+            body = r.get_json()
+        except Exception as e:
+            log(f"Failed to parse JSON: {e}")
+            return Response(
+                response=json.dumps({"records": []}),
+                status=200,
+                content_type="application/json"
+            )
+        
+        log(f"Request: method={r.method}, url={r.url}, headers={dict(r.headers)}, data={body}")
 
         retrieval_setting = body.get('retrieval_setting')
         query = body.get('query')
         knowledge_id = body.get('knowledge_id')
 
+        log(f"Received request - knowledge_id: {knowledge_id}, query: {query}, retrieval_setting: {retrieval_setting}")
+
+        if not knowledge_id:
+            log("knowledge_id is empty, returning empty records")
+            return Response(
+                response=json.dumps({"records": []}),
+                status=200,
+                content_type="application/json"
+            )
+
         client = boto3.client(
             "bedrock-agent-runtime",
             aws_secret_access_key=settings.get("aws_secret_access_key"),
@@ -22,6 +54,7 @@ def _invoke(self, r: Request, values: Mapping, settings: Mapping) -> Response:
         )
 
         try:
+            log(f"Calling bedrock-agent-runtime retrieve API with knowledgeBaseId: {knowledge_id}")
             response = client.retrieve(
                 knowledgeBaseId=knowledge_id,
                 retrievalConfiguration={
@@ -31,10 +64,13 @@ def _invoke(self, r: Request, values: Mapping, settings: Mapping) -> Response:
                 retrievalQuery={"text": query},
             )
 
+            log(f"API response status: {response.get('ResponseMetadata', {}).get('HTTPStatusCode')}")
+
             results = []
             if response.get("ResponseMetadata") and response.get("ResponseMetadata").get("HTTPStatusCode") == 200:
                 if response.get("retrievalResults"):
                     retrieval_results = response.get("retrievalResults")
+                    log(f"Retrieved {len(retrieval_results)} results from knowledge base")
                     for retrieval_result in retrieval_results:
                         # filter out results with score less than threshold
                         if retrieval_result.get("score") < retrieval_setting.get("score_threshold", .0):
@@ -47,6 +83,7 @@ def _invoke(self, r: Request, values: Mapping, settings: Mapping) -> Response:
                         }
                         results.append(result)
 
+            log(f"Returning {len(results)} results after filtering")
             return Response(
                 response=json.dumps({"records": results}),
                 status=200,
@@ -55,6 +92,7 @@ def _invoke(self, r: Request, values: Mapping, settings: Mapping) -> Response:
 
         except botocore.exceptions.ClientError as error:
             error_code = error.response['Error']['Code']
+            log(f"AWS ClientError: {error_code} - {error.response['Error'].get('Message')}")
             if error_code == "InvalidSignatureException":
                 return Response(
                     response=json.dumps({"error_code": 1002, "error_msg": "Wrong AWS secret access key"}),

extensions/aws_bedrock_knowledge_base/manifest.yaml

@@ -1,4 +1,4 @@
-version: 0.0.3
+version: 0.0.4
 type: plugin
 author: langgenius
 name: aws_bedrock_knowledge_base