feat: complete assignment

2026-06-29 07:14:19 -05:00
parent 28229d39aa
commit 865cba9797
17 changed files with 618 additions and 2 deletions
@@ -0,0 +1,47 @@
+import os
+
+from google.genai import types
+
+
+def get_file_content(working_directory: str, file_path: str) -> str:
+    try:
+        wdp = os.path.abspath(working_directory)
+        target = os.path.normpath(os.path.join(wdp, file_path))
+        valid_target_dir = os.path.commonpath([wdp, target]) == wdp
+
+        if not valid_target_dir:
+            return f'Error: Cannot read "{file_path}" as it is outside the permitted working directory'
+
+        if not os.path.isfile(target):
+            return f'Error: File not found or is not a regular file: "{file_path}"'
+
+        MAX_CHARS = 10000
+        content = ""
+
+        with open(target, "r") as f:
+            content += f.read(MAX_CHARS)
+
+            if f.read(1):
+                content += (
+                    f'[...File "{file_path}" truncated at {MAX_CHARS} characters]'
+                )
+
+        return content
+    except Exception as e:
+        return f"Error: {e}"
+
+
+schema_get_file_content = types.FunctionDeclaration(
+    name="get_file_content",
+    description="",
+    parameters=types.Schema(
+        type=types.Type.OBJECT,
+        properties={
+            "file_path": types.Schema(
+                type=types.Type.STRING,
+                description="",
+            ),
+        },
+        required=["file_path"],
+    ),
+)