Merge pull request #5 from Olemi-llm-apprentice/feature/file-data-blob

refactor: update Excel cell editor and writer tools to use blob data …

Author: Olemi-llm-apprentice

.gitignore

@@ -166,3 +166,6 @@ cython_debug/
 #  and can be added to the global gitignore or merged into this file.  For a more nuclear
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 #.idea/
+
+samples/
+

tools/excel-cell-editor.py

@@ -15,6 +15,14 @@ def get_url_from_file_data(file_data: Any) -> str:
         return file_data['url']
     return ''
 
+def get_blob_from_file_data(file_data: Any) -> bytes:
+    """Difyのファイルデータからblobを抽出する"""
+    if hasattr(file_data, 'blob'):
+        return file_data.blob
+    elif isinstance(file_data, dict) and 'blob' in file_data:
+        return file_data['blob']
+    return None
+
 class ExcelCellEditorTool(Tool):
     def _invoke(self, tool_parameters: dict[str, Any]) -> Generator[ToolInvokeMessage, None, None]:
         try:
@@ -27,35 +35,19 @@ def _invoke(self, tool_parameters: dict[str, Any]) -> Generator[ToolInvokeMessag
                 )
                 return
 
-            # 操作モードの取得
-            # operation = tool_parameters.get("operation", "read")
-            
-            # ファイルのURLを取得
-            file_url = get_url_from_file_data(excel_file)
+            # ファイルのblobデータを取得
+            file_blob = get_blob_from_file_data(excel_file)
 
-            if not file_url:
-                yield ToolInvokeMessage(
-                    type="text",
-                    message={"text": "ファイルのURLが見つかりません。"}
-                )
-                return
-
-            # ファイルをダウンロード
-            try:
-                response = requests.get(file_url)
-                response.raise_for_status()
-                file_bytes = response.content
-                
-            except Exception as e:
+            if not file_blob:
                 yield ToolInvokeMessage(
                     type="text",
-                    message={"text": f"ファイルのダウンロードに失敗しました: {str(e)}"}
+                    message={"text": "ファイルのblobデータが見つかりません。"}
                 )
                 return
 
             # Excelファイルの読み込み
             try:
-                wb = openpyxl.load_workbook(BytesIO(file_bytes))
+                wb = openpyxl.load_workbook(BytesIO(file_blob))
                 ws = wb.active
             except Exception as e:
                 yield ToolInvokeMessage(
@@ -64,8 +56,7 @@ def _invoke(self, tool_parameters: dict[str, Any]) -> Generator[ToolInvokeMessag
                 )
                 return
 
-            # if operation == "read":
-            #     # セル内容の読み取り
+            # セル内容の読み取り
             cell_data = {}
             for row in ws.iter_rows():
                 for cell in row:

tools/excel-cell-writer.py

@@ -15,6 +15,14 @@ def get_url_from_file_data(file_data: Any) -> str:
         return file_data['url']
     return ''
 
+def get_blob_from_file_data(file_data: Any) -> bytes:
+    """Difyのファイルデータからblobを抽出する"""
+    if hasattr(file_data, 'blob'):
+        return file_data.blob
+    elif isinstance(file_data, dict) and 'blob' in file_data:
+        return file_data['blob']
+    return None
+
 class ExcelCellWriterTool(Tool):
     def _invoke(self, tool_parameters: dict[str, Any]) -> Generator[ToolInvokeMessage, None, None]:
         try:
@@ -36,32 +44,19 @@ def _invoke(self, tool_parameters: dict[str, Any]) -> Generator[ToolInvokeMessag
                 )
                 return
 
-            # ファイルのURLを取得
-            file_url = get_url_from_file_data(excel_file)
+            # ファイルのblobデータを取得
+            file_blob = get_blob_from_file_data(excel_file)
 
-            if not file_url:
-                yield ToolInvokeMessage(
-                    type="text",
-                    message={"text": "ファイルのURLが見つかりません。"}
-                )
-                return
-
-            # ファイルをダウンロード
-            try:
-                response = requests.get(file_url)
-                response.raise_for_status()
-                file_bytes = response.content
-                
-            except Exception as e:
+            if not file_blob:
                 yield ToolInvokeMessage(
                     type="text",
-                    message={"text": f"ファイルのダウンロードに失敗しました: {str(e)}"}
+                    message={"text": "ファイルのblobデータが見つかりません。"}
                 )
                 return
 
             # Excelファイルの読み込み
             try:
-                wb = openpyxl.load_workbook(BytesIO(file_bytes))
+                wb = openpyxl.load_workbook(BytesIO(file_blob))
                 ws = wb.active
             except Exception as e:
                 yield ToolInvokeMessage(