Sborole-Final-Assignment

Runtime error

Sborole-Final-Assignment / tools /ReadImageTool.py

Update tools/ReadImageTool.py

7dab6f9 verified 12 months ago

513 Bytes

import base64
from smolagents.tools import Tool
class ReadImageTool(Tool):
    name = "read_image"
    description = "Reads an image file and returns its base64 string for Gemini."
    inputs = {
        "file_path": {
            "type": "string",
            "description": "The full path of the text file to read"
        }
    }
    output_type = "string"

    def forward(self, file_path: str) -> str:
        with open(file_path, "rb") as img:
            return base64.b64encode(img.read()).decode("utf-8")

	import base64
	from smolagents.tools import Tool
	class ReadImageTool(Tool):
	name = "read_image"
	description = "Reads an image file and returns its base64 string for Gemini."
	inputs = {
	"file_path": {
	"type": "string",
	"description": "The full path of the text file to read"
	}
	}
	output_type = "string"

	def forward(self, file_path: str) -> str:
	with open(file_path, "rb") as img:
	return base64.b64encode(img.read()).decode("utf-8")