Enhance file reading with error handling

- Wrapped the file reading functionality inside a `_run` method.
- Added error handling to return a descriptive error message if an exception occurs during file reading.
This commit is contained in:
Victor C Tavernari
2024-03-31 23:32:40 +01:00
parent 80f9613959
commit 92abe0b726

View File

@@ -2,13 +2,19 @@ from typing import Optional, Type, Any
from pydantic.v1 import BaseModel, Field from pydantic.v1 import BaseModel, Field
from ..base_tool import BaseTool from ..base_tool import BaseTool
class FixedFileReadToolSchema(BaseModel): class FixedFileReadToolSchema(BaseModel):
"""Input for FileReadTool.""" """Input for FileReadTool."""
pass pass
class FileReadToolSchema(FixedFileReadToolSchema): class FileReadToolSchema(FixedFileReadToolSchema):
"""Input for FileReadTool.""" """Input for FileReadTool."""
file_path: str = Field(..., description="Mandatory file full path to read the file") file_path: str = Field(
...,
description="Mandatory file full path to read the file"
)
class FileReadTool(BaseTool): class FileReadTool(BaseTool):
name: str = "Read a file's content" name: str = "Read a file's content"
@@ -16,7 +22,11 @@ class FileReadTool(BaseTool):
args_schema: Type[BaseModel] = FileReadToolSchema args_schema: Type[BaseModel] = FileReadToolSchema
file_path: Optional[str] = None file_path: Optional[str] = None
def __init__(self, file_path: Optional[str] = None, **kwargs): def __init__(
self,
file_path: Optional[str] = None,
**kwargs
):
super().__init__(**kwargs) super().__init__(**kwargs)
if file_path is not None: if file_path is not None:
self.file_path = file_path self.file_path = file_path
@@ -28,6 +38,9 @@ class FileReadTool(BaseTool):
self, self,
**kwargs: Any, **kwargs: Any,
) -> Any: ) -> Any:
try:
file_path = kwargs.get('file_path', self.file_path) file_path = kwargs.get('file_path', self.file_path)
with open(file_path, 'r') as file: with open(file_path, 'r') as file:
return file.read() return file.read()
except Exception as e:
return f"Fail to read the file {file_path}. Error: {e}"