end of project

functions/write_file.py

@@ -1,4 +1,5 @@
 import os
+from google.genai import types
 
 def write_file(working_directory, file_path, content):
     abs_working_dir = os.path.abspath(working_directory)
@@ -15,3 +16,25 @@ def write_file(working_directory, file_path, content):
             return f'Successfully wrote to "{file_path}" ({len(content)} characters written)'
     except Exception as e:
         print(f"Error: {e}")
+
+schema_write_file = types.FunctionDeclaration(
+    name="write_file",
+    description="Write file.",
+    parameters=types.Schema(
+        type=types.Type.OBJECT,
+        properties={
+            "working_directory": types.Schema(
+                type=types.Type.STRING,
+                description="The directory to read the file from. If not provided, lists files in the working directory itself.",
+            ),
+            "file_path": types.Schema(
+                type=types.Type.STRING,
+                description="The file to write the content to. If not provided, the function will not work."
+            ),
+            "content": types.Schema(
+                type=types.Type.STRING,
+                description="Content to write to the file."
+            )
+        },
+    ),
+)