Fix Two Bugs:

1. [bedrock retrieve plugin]x-amz-bedrock-kb-source-uri is no longer exists, use s3Location instead
2. [s3 plugin] set content_type for s3 put_object, if not it will be set as binary type by default

Author: Yuanbo Li

plugins/aws_tools.difypkg

@@ -1,4 +1,4 @@
-version: 0.0.18
+version: 0.0.20
 type: plugin
 author: langgenius
 name: aws_tools

plugins/aws_tools/manifest.yaml

@@ -17,7 +17,7 @@ def convert_to_dify_kb_format(self, kb_repsonse):
         result_array = []
         for idx, item in enumerate(kb_repsonse['retrievalResults']):
             # 提取基础字段
-            source_uri = item['metadata']['x-amz-bedrock-kb-source-uri']
+            source_uri = item['location']['s3Location']['uri']
             page_number = item['metadata'].get('x-amz-bedrock-kb-document-page-number', 0)
             data_source_id = item['metadata'].get('x-amz-bedrock-kb-data-source-id', '')
             chunk_id = item['metadata'].get('x-amz-bedrock-kb-chunk-id','')

plugins/aws_tools/tools/bedrock_retrieve.py

@@ -1,5 +1,6 @@
 from typing import Any, Union
 from urllib.parse import urlparse
+import mimetypes
 
 import boto3
 
@@ -52,8 +53,18 @@ def _invoke(
                 if not text_content:
                     yield self.create_text_message("text_content parameter is required for write operation")
 
+                # Infer content type from file extension
+                content_type, _ = mimetypes.guess_type(key)
+                if not content_type:
+                    content_type = "text/plain; charset=utf-8"
+
                 # Write content to S3
-                self.s3_client.put_object(Bucket=bucket, Key=key, Body=text_content.encode("utf-8"))
+                self.s3_client.put_object(
+                    Bucket=bucket, 
+                    Key=key, 
+                    Body=text_content.encode("utf-8"),
+                    ContentType=content_type
+                )
                 result = f"s3://{bucket}/{key}"
 
                 # Generate presigned URL for the written object if requested