Update tests.py

tests.py

@@ -205,7 +205,15 @@ def create_code_files(filename: str, code) -> dict:
         return {"info":"The referenced code files were created successfully."}
 
     else:
-        return {"info":"The referenced code files already exist. Please rename the file or delete the existing one."}
+        if isinstance(code, dict):
+            with open(os.path.join(destination_dir, filename), 'w', encoding='utf-8') as f:
+                json.dump(code, f, ensure_ascii=False, indent=4)
+        else:
+            f = open(os.path.join(destination_dir, filename), "w")
+            f.write(str(code))
+            f.close()
+        return {"info":"The referenced code files were created successfully."}
+        # return {"info":"The referenced code files already exist. Please rename the file or delete the existing one."}
 
     
 @mcp.tool()