feat: complete assignment

functions/write_file.py

@@ -0,0 +1,47 @@
+import os
+
+from google.genai import types
+
+
+def write_file(working_directory: str, file_path: str, content: 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 write to "{file_path}" as it is outside the permitted working directory'
+
+        if os.path.isdir(target):
+            return f'Error: Cannot write to "{file_path}" as it is a directory'
+
+        os.makedirs("/".join(target.split("/")[:-1]), exist_ok=True)
+
+        with open(target, "w") as f:
+            f.write(content)
+
+        return (
+            f'Successfully wrote to "{file_path}" ({len(content)} characters written)'
+        )
+    except Exception as e:
+        return f"Error: {e}"
+
+
+schema_write_file = types.FunctionDeclaration(
+    name="write_file",
+    description="",
+    parameters=types.Schema(
+        type=types.Type.OBJECT,
+        properties={
+            "file_path": types.Schema(
+                type=types.Type.STRING,
+                description="",
+            ),
+            "content": types.Schema(
+                type=types.Type.STRING,
+                description="",
+            ),
+        },
+        required=["file_path", "content"],
+    ),
+)